o \h=@sddlZddlZddlZddlZddlZddlZddlZddlmZmZddl m Z m Z m Z ddl mZmZddlZddlmZddlmZddlmZddlmZdd lmZmZmZmZmZdd lm Z m!Z!dd l"m#Z#dd l$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-m.Z.m/Z/ddl0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:ddl;mZ>m?Z?ddl@mAZAzddlBZBWn eCyddlDZBYnwejEeefZFdZGddZHdeIeJe ffddZKGdddeLZMddZNddZOdS)N)datetime timedelta)AnyOptionalUnion)UUIDuuid4)tzutc) string_types)Consumer)ExceptionCapture)exc_info_from_errorexceptions_from_error_tuple handle_in_appexception_is_already_capturedmark_exception_as_captured)InconclusiveMatchErrormatch_feature_flag_properties)Poller) DEFAULT_HOSTAPIError batch_postdetermine_server_hostflagsget remote_config)_get_current_contextget_context_distinct_idget_context_session_id) FeatureFlagFeatureFlagResult FlagMetadataFlagsAndPayloads FlagsResponse FlagValuenormalize_flags_responseto_flags_and_payloads to_payloads to_values)SizeLimitedDictcleanguess_timezoneremove_trailing_slash)VERSIONiPcCsd}d}tj}|dr!d}ttdrtd}|r|}||fS|dkr:d}ttdr6td}|r6|}||fS|d rQd }t}|d rM|d }||fS|d red }ttdrat}||fS|}ttdrpt}||fS)zr Returns standardized OS name and version information. Similar to how user agent parsing works in JS. winWindows win32_verrdarwinzMac OS Xmac_verlinuxLinuxversionfreebsdFreeBSDrelease) sysplatform startswithhasattrr1r3distroinfor9)os_name os_version platform_name win_version mac_version linux_inforFQ/home/air/segue/gemini/backup/venv/lib/python3.10/site-packages/posthog/client.py get_os_infoBs@         rHreturncCs*t\}}tdtjdd||dS)Nz%s.%s.%s)z$python_runtimez$python_versionz$osz $os_version)rHr;python_implementationr: version_info)r@rArFrFrGsystem_contextms rMc@s eZdZdZedZ                      did dZeddZ e j ddZ      djddZ    dkde e eee fffddZ    dkde e e ffddZ    dkdefddZ    dkdefddZ         dlddZ      djdd Z      djd!d"Z        dmd#d$Z      djd%d&Z       dnd'd(Z       dnd)d*Zd+d,Zd-d.Zd/d0Zd1d2Z d3d4Z!d5d6Z"iiidd7de#fd8d9Z$iiidddd:d;d<Z%diiidddd=d>e&e#de&e'fd?d@Z(iiidddd:de&e'fdAdBZ)iiidddd:de&e#fdCdDZ*dEe dFe dGe e e fdHe e e fdIe e e fde&e#f dJdKZ+diiiddddLdMe&e#fdNdOZ,dEe dFe dGe e e fdHe e e fdIe e e fdPe&ede-e&e.e&e ffdQdRZ/dFe dEe dSe&e#dTe&e dUedGe e e fdPe&edVe&e dWe&e.fdXdYZ0dEe fdZd[Z1dEe dMe#de&e fd\d]Z2iiiddd^de&e e eee fffd_d`Z3iiiddd^defdadbZ4iiidd7de-eeffdcddZ5dedfZ6dgdhZ7dS)oClientzCreate a new PostHog client.posthogNF'Td?rJc Cst||_|p ||_td|jt||_||_||_| |_|p!t |_ t ||_ | |_ | |_d|_d|_d|_d|_||_||_d|_ttt|_||_||_||_||_||_||_||_ d|_!||_"|durzzt#$}Wn t%yyd}Ynw||_&||_'|rt()|j*+t(j,n|j*+t(j-|durt.|r||_/n |j*0dd|_/nd|_/|jrt1||j d|_!| rd|_2dS|rt34|j5t6| D]"}g|_2t7|j|j|j |||| | | |d }|j28||r|9qdS)Napi_keyz/before_send is not callable, it will be ignored) integrations)hoston_errorflush_atflush_intervalgzipretriestimeouthistorical_migration):queueQueuerVrequirer rYdebugsend sync_moderraw_hostrrXr\r^_feature_flagsfeature_flags_by_keygroup_type_mappingcohorts poll_interval%feature_flags_request_timeout_secondspollerr) MAX_DICT_SIZEset#distinct_ids_feature_flags_reporteddisabled disable_geoipr_super_propertiesenable_exception_autocapturelog_captured_exceptions"exception_autocapture_integrationsexception_capture privacy_modeosgetcwd Exception project_rootpersonal_api_keylogging basicConfiglogsetLevelDEBUGWARNINGcallable before_sendwarningr consumersatexitregisterjoinranger appendstart)selfrVrXrcmax_queue_sizerdrYrZr[r\ max_retriesrer^threadrkr}project_api_keyrqrrr_rlrsrtrurvr|rxrnconsumerrFrFrG__init__}s            zClient.__init__cC|jS)z9 Get the local evaluation feature flags. )rgrrFrFrG feature_flagsszClient.feature_flagscCs2|pg|_dd|jD|_|jdusJddS)z9 Set the local evaluation feature flags. cSs$i|]}|ddur|d|qS)keyNr).0flagrFrFrG s z(Client.feature_flags..NDfeature_flags_by_key should be initialized when feature_flags is set)rgrh)rrrFrFrGrs  cCsz|dur tjdtdd|durt}|pi}td|ttd|td|vr/tr/t|d<|||d|d}|||S) NNThe 'context' parameter is deprecated and will be removed in a future version. stacklevel distinct_id properties $session_idz $identify timestampr$seteventuuid) warningswarnDeprecationWarningrrbID_TYPESdictr_enqueuerrrcontextrrrrmsgrFrFrGidentifys(     zClient.identifyrIcC||||||}t|piS)zP Get feature flag variants for a distinct_id by calling decide. )get_flags_decisionr(rrgroupsperson_propertiesgroup_propertiesrr resp_datarFrFrGget_feature_variants9  zClient.get_feature_variantscCr)zP Get feature flag payloads for a distinct_id by calling decide. )rr'rrFrFrGget_feature_payloadsIrzClient.get_feature_payloadscCs||||||}t|S)zU Get feature flags and payloads for a distinct_id by calling decide. )rr&)rrrrrrrresprFrFrGget_feature_flags_and_payloadsYs z%Client.get_feature_flags_and_payloadscCsr|durt}td|t|dur|j}|rtd|tni}|||||d}t|j|jfd|ji|}t |S)zd Get feature flags decision, using either flags() or decide() API based on rollout. Nrr)rrrr geoip_disabler^) rrbrrrrrrVrXrlr%)rrrrrrr request_datarrFrFrGris.  zClient.get_flags_decisionc  Cs|dur tjdtddi|pit}d|vr!tr!t|d<|dur(t}td|ttd|ttd|t t } | rJ| } | || }|||||d} |r`td |t|| dd <i} i}|rz |j ||| d }Wn-ty}z|jd |WYd}~nd}~ww|jr|d kr|j||pi| dd}|piD] \}}|| d|<qdd|piD}|r|| d<| ri| | d| d<|| | S)Nrrrrrrr)rrrrrrz$groups)rrz0[FEATURE FLAGS] Unable to get feature variants: $feature_flag_calledT)rrronly_evaluate_locally $feature/cSsg|] \}}|dur|qS)FrF)rrvaluerFrFrG s z"Client.capture..z$active_feature_flags)rrrrMrrrbrrr r collect_tagsupdaterr{r exceptionr get_all_flagsitemsr)rrrrrrrrsend_feature_flagsrrcurrent_context context_tagsrextra_propertiesfeature_variantsefeaturevariantactive_feature_flagsrFrFrGcapturest           zClient.capturecCb|dur tjdtdd|durt}|pi}td|ttd|t|||d|d}|||S)NrrrrrrrrrrrrbrrrrrFrFrGro$    z Client.setcCr)Nrrrrr $set_once)rrrrrrrrFrFrGset_once rzClient.set_oncec Cs|dur tjdtdd|pi}td|ttd|ttd|t|r+td|tnd||}d |||d |||d } || |S) Nrrr group_type group_keyrrz${}_{}z$groupidentify)z $group_type $group_keyz $group_set)rrrrr)rrrrbrrformatr) rrrrrrrrrrrrFrFrGgroup_identify+s.      zClient.group_identifycCs^|dur tjdtdd|durt}td|ttd|t||d|d|d}|||S) Nrrr previous_idr)raliasz $create_alias)rrrr)rrrrrbrr)rrrrrrrrrrFrFrGrTs$    z Client.aliasc Cs|dur tjdtdd|durt}|pi}td|ttd|td|vr/tr/t|d<td|t||d<d ||||d }| ||S) Nrrrrrrurlz $current_urlz $pageview)rrrrr) rrrrrbrrrr r) rrrrrrrrrrrFrFrGpagevs,      z Client.pagec Ks|dur tjdtdd|durt}z|pi}|dur)t|r)|jdWdS|dur4d|d<t}td|t td|t |durIt |} nt } | dusU| d kr^|jd WdSt| } td d | ii|jd } | d d } | dd| dd| t|jd|jd|d|}|jr|jj||d||d|||||} |durt|| WSty}z|jd|WYd}~dSd}~ww)Nrrrz$Exception already captured, skippingFz$process_person_profilerr)NNNz"No exception information availablervalues)r|rtyperz /project/z/person/)z$exception_typez$exception_messagez$exception_listz$exception_personURL)extraz $exceptionzFailed to capture exception: )rrrrrrrcrrbrrr r:exc_inforrrr|rr,rfrVrurrrr{)rrrrrrrrkwargsrall_exceptions_with_tracer$all_exceptions_with_trace_and_in_appresrrFrFrGcapture_exceptionsl          zClient.capture_exceptionc Cs|jrdS|d}|durtjtd}td|tt|}||d<d|vr5|d}|r5t||d<| ds>i|d<d|dd<t |dd <|durQ|j }|rYd |dd <|j rgi|d|j |d<t| d d|d <t |}|jrz||}|dur|jd WdS|}Wnty}z|jd|WYd}~nd}~ww|jd||jsd |fS|jr|jd|dt|j|j|j|j|g|jdd |fSz|jj|dd|jd|dd |fWStjy|jdd|fYSw)z8Push a new `msg` onto the queue, return `(success, msg)`)FrqrNtzrrzposthog-pythonz$libz $lib_versionTz$geoip_disablerz%Event dropped by before_send callback)TNzError in before_send callback: z queueing: %szenqueued with blocking %s.r)r\r^batchr_F)blockz enqueued %s.zanalytics-python queue is full)rqrnowr rbr+ isoformatpop stringify_idrr-rrrsr*rrrcr{rrdrerrVrXr\r^r_r`putFullr)rrrrrr modified_msgrrFrFrGrsr              zClient._enqueuecCs(|j}|}||jd|dS)z4Forces a flush from the internal queue to the serverz$successfully flushed about %s items.N)r`qsizerrrc)rr`sizerFrFrGflushDsz Client.flushc CsJ|jD]}|z|WqtyYqw|jr#|jdSdS)zbEnds the consumer thread once the queue is empty. Blocks execution until finished N)rpauser RuntimeErrorrmstop)rrrFrFrGrLs   z Client.joincCs(|||jr|jdSdS)z2Flush all messages and cleanly shutdown the clientN)rrrwcloserrFrFrGshutdown[s zClient.shutdownc Cs>z&t|jd|jd|jdd}|dpg|_|dpi|_|dp#i|_Wnntyt}zB|jdkrC|j d |j rBtdd d n'|jd kra|j d g|_i|_i|_|j r`td dd n |j d|WYd}~n%d}~wt y}z|j d|j|j |WYd}~nd}~wwtjtd|_dS)Nz*/api/feature_flag/local_evaluation/?token=z &send_cohorts r^rrirjiz[FEATURE FLAGS] Error loading feature flags: To use feature flags, please set a valid personal_api_key. More information: https://posthog.com/docs/api/overviewzYou are using a write-only key with feature flags. To use feature flags, please set a personal_api_key More information: https://posthog.com/docs/api/overview)statusmessageiz[FEATURE FLAGS] PostHog feature flags quota limited, resetting feature flag data. Learn more about billing limits at https://posthog.com/docs/billing/limits-alertsz#PostHog feature flags quota limitedz-[FEATURE FLAGS] Error loading feature flags: z`[FEATURE FLAGS] Fetching feature flags failed with following error. We will retry in %s seconds.r)rr}rVrXrrirjrrrerrorrcrr{rkrrr _last_feature_flag_poll)rresponserrFrFrG_load_feature_flagscs\   zClient._load_feature_flagscCs^|js|jdg|_dS||jr|js-tt|j d|jd|_|j dSdS)NzL[FEATURE FLAGS] You have to specify a personal_api_key to use feature flags.)seconds)intervalexecute) r}rrrrrmis_aliverrrkrrrFrFrGload_feature_flagss zClient.load_feature_flagsrrrwarn_on_unknown_groupsc Cs|ddr td|dsdS|dpi}|d}|durk|jt|} | s=|jd|d|d td | |vr_|rQ|jd |d d dS|jd |d d dS|| } t||| | St||||jS) Nensure_experience_continuityFz&Flag has experience continuity enabledactivefiltersaggregation_group_type_indexz)[FEATURE FLAGS] Unknown group type index z for feature flag rz!Flag has unknown group type indexz2[FEATURE FLAGS] Can't compute group feature flag: z without group names passed in) rrristrrrrcrrj) r feature_flagrrrrr flag_filtersr group_namefocused_group_propertiesrFrFrG_compute_flag_locallys<     zClient._compute_flag_locallyrrrrsend_feature_flag_eventsrrc Cs.|j||||||||d} | durdSt| S)Nr)get_feature_flagbool) rrrrrrrrrrrrFrFrGfeature_enableds  zClient.feature_enabledoverride_match_valuerrrrrrrr c CsLtd|ttd|ttd|t|jrdS|||||\}}d} d} d} ||||||} | du}|rL|p9| }|rB|||nd}t |||} n>|sz | |||||| \} } t | |} |j d|d| Wnty}z|j d|WYd}~nd}~ww|r|||| r| nd| r| jnd||| | | | S)Nrrrz&Successfully computed flag remotely: #z -> #z-[FEATURE FLAGS] Unable to get flag remotely: )rbr rrrq&_add_local_person_and_group_properties_locally_evaluate_flag_compute_payload_locallyr from_value_and_payload%_get_feature_flag_details_from_decidefrom_flag_detailsrrcr{r_capture_feature_flag_called get_valuepayload)rrrr rrrrrrr flag_result flag_details request_id flag_valueflag_was_locally_evaluatedlookup_match_valuer)rrFrFrG_get_feature_flag_resultst     zClient._get_feature_flag_resultc Cs|j||||||||dS)aL Get a FeatureFlagResult object which contains the flag result and payload for a key by evaluating locally or remotely depending on whether local evaluation is enabled and the flag can be locally evaluated. This also captures the $feature_flag_called event unless send_feature_flag_events is False. r)r0) rrrrrrrrrrrFrFrGget_feature_flag_resultAszClient.get_feature_flag_resultc Cs*|j||||||||d} | r| SdS)a# Get a feature flag value for a key by evaluating locally or remotely depending on whether local evaluation is enabled and the flag can be locally evaluated. This also captures the $feature_flag_called event unless send_feature_flag_events is False. rN)r1r() rrrrrrrrrrfeature_flag_resultrFrFrGr^s zClient.get_feature_flagrrrrrc Cs|jdur |jr |d}|jrv|jdusJd|j|}|rvz|j|||||d}|jd|d|W|StyZ}z|jd|d|WYd}~|Sd}~wt yu}z|j d|WYd}~|Sd}~ww|S)Nrrrrz$Successfully computed flag locally: z -> zFailed to compute flag z locally: z7[FEATURE FLAGS] Error while computing variant locally: ) rr}r rhrrrrcrr{r) rrrrrrrrrrFrFrGr"}sB   "zClient._locally_evaluate_flag) match_valuerrrrrrrr4c Cs*|j||||||||| d } | r| jSdS)Nr)r0r)) rrrr4rrrrrrrr2rFrFrGget_feature_flag_payloads  zClient.get_feature_flag_payloadrrc Cs@||||||}|d}|d} | r| |nd} | |fS)zK Calls /decide and returns the flag details and request id requestIdrN)rr) rrrrrrrrrr,rr+rFrFrGr%s   z,Client._get_feature_flag_details_from_deciderr)r.r,r+c Cs|d|dur dnt|} | |j|vrod|d|d|d||i} |r)|| d<|r/|| d<t| tr[| jrA| jjrA| jj| d <t| jtr[| jjrQ| jj| d <| jj r[| jj | d <|j |d | ||d |j| | dSdS)N_z::null::z $feature_flagz$feature_flag_responselocally_evaluatedrz$feature_flag_payloadz$feature_flag_request_idz$feature_flag_reasonz$feature_flag_versionz$feature_flag_idr)rrr) rrp isinstancerreason descriptionmetadatar!r6idradd) rrrrr)r.rrrr,r+feature_flag_reported_keyrrFrFrGr'sD       z#Client._capture_feature_flag_calledc Csv|jrdS|jdur|jddSz t|j|j||jdWSty:}z|jd|WYd}~dSd}~ww)Nz`[FEATURE FLAGS] You have to specify a personal_api_key to fetch decrypted feature flag payloads.rz>[FEATURE FLAGS] Unable to get decrypted feature flag payload: ) rqr}rrrrXrlr{r)rrrrFrFrGget_remote_config_payloads( z Client.get_remote_config_payloadcCshd}|jdur |S|j|}|r2|dpi}|dpi}t|tr(|r(dnt|}||d}|S)Nrpayloadstrue)rhrr9rr)rrr4r)flag_definitionr flag_payloads lookup_valuerFrFrGr#s   zClient._compute_payload_locallyrrrrrrcCs|j||||||d}|dS)NrF featureFlags)get_all_flags_and_payloads)rrrrrrrrrrFrFrGr4s  zClient.get_all_flagsc Cs|jrdddS|||||\}}|j||||d\}}|rM|sMz|j|||||d} t| WStyL} z|jd| WYd} ~ |Sd} ~ ww|S)NrGfeatureFlagPayloadsr3)rrrrrz:[FEATURE FLAGS] Unable to get feature flags and payloads: )rqr!#_get_all_flags_and_payloads_locallyrr&r{rr) rrrrrrrrrfallback_to_decidedecide_responserrFrFrGrHIs<   z!Client.get_all_flags_and_payloadsc Cstd|ttd|t|jdur|jr|i}i}d}|jrt|jD]N} z%|j| |||||d|| d<|| d|| d} | rJ| || d<Wq$tyVd}Yq$t yr} z|j d| d}WYd} ~ q$d} ~ wwnd}||d|fS) NrrFrrTz;[FEATURE FLAGS] Error while computing variant and payload: rI) rbrrrr}r rr#rr{rr) rrrrrrrrArLrmatched_payloadrrFrFrGrKtsP    z*Client._get_all_flags_and_payloads_locallycCrN)rrrFrFrGfeature_flag_definitionsszClient.feature_flag_definitionscCsHd|i|pi}i}|r |D]}d||i||pi||<q||fS)Nrrr)rrrrrall_person_propertiesall_group_propertiesrrFrFrGr!s  z-Client._add_local_person_and_group_properties)NNFrPTNrQrRFrJFrSrTrUNNFTFrJNFFNNFN)NNNNNN)NNNN) NNNNNNNFN)NNNNNNNN)NNNNNNN)8__name__ __module__ __qualname____doc__r~ getLoggerrrpropertyrsetterrrrrrrrr"rr#rrrorrrrrrrrrrr r$rrrr r0r1rr"r5tuplerr%r'r@r#rrHrKrPr!rFrFrFrGrNxs     (     + Y # # + $ * XN2 7   T "     +        7   /  3 rNcCs$t||sd|||}t|dS)z8Require that the named `field` has the right `data_type`z{0} must have {1}, got: {2}N)r9rAssertionError)namefield data_typerrFrFrGrbs rbcCs"|durdSt|tr |St|SrO)r9r r)valrFrFrGrs  r)Prr~numbersryr;r:rrrtypingrrrrrrr> dateutil.tzr sixr posthog.consumerr posthog.exception_capturer posthog.exception_utilsr rrrrposthog.feature_flagsrrposthog.pollerrposthog.requestrrrrrrrposthog.scopesrrr posthog.typesrr r!r"r#r$r%r&r'r( posthog.utilsr)r*r+r,posthog.versionr-r` ImportErrorraNumberrrnrHrrrMobjectrNrbrrFrFrFrGsZ     $ 0     + O