o g@sHddlZddlZddlZddlZddlZddlmZmZddlmZm Z ddl m Z ddl m Z ddlmZddlmZddlmZmZmZdd lmZmZdd lmZdd lmZmZmZmZm Z m!Z!dd l"m#Z#m$Z$m%Z%m&Z&dd l'm(Z(zddl)Z)Wn e*yddl+Z)Ynwej,e efZ-dZ.Gddde/Z0ddZ1ddZ2dS)N)datetime timedelta)UUIDuuid4)tzutc) string_types)Consumer)ExceptionCapture)exc_info_from_errorexceptions_from_error_tuple handle_in_app)InconclusiveMatchErrormatch_feature_flag_properties)Poller) DEFAULT_HOSTAPIError batch_postdecidedetermine_server_hostget)SizeLimitedDictcleanguess_timezoneremove_trailing_slash)VERSIONiPc@seZdZdZedZ                    dKd dZdLddZ dMddZ dMddZ dMddZ dMddZ         dNddZ dLddZdLddZ        dOdd ZdLd!d"Z dPd#d$Z       dPd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Ziiidd3d4d5Ziiidddd6d7d8Ziiidddd6d9d:Zdiiidddd;dd?Ziiiddd@dAdBZiiiddd@dCdDZ iiidd3dEdFZ!dGdHZ"dIdJZ#dS)QClientzCreate a new PostHog client.posthogNF'Td?c Cst||_|p ||_td|jt||_||_||_| |_|p!t |_ t ||_ | |_ | |_d|_d|_d|_d|_||_||_d|_ttt|_||_||_||_||_||_||_d|_ ||_!|durwzt"#}Wn t$yvd}Ynw||_%||_&|rt'(|j)*t'j+n|j)*t'j,|jrt-||jd|_ | rd|_.dS|rt/0|j1t2| D]"}g|_.t3|j|j|j |||| | | |d }|j.4||r|5qdS)Napi_key) integrations)hoston_errorflush_atflush_intervalgzipretriestimeouthistorical_migration)6queueQueuer$requirerr'debugsend sync_moderraw_hostrr&r*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_autocapture"exception_autocapture_integrationsexception_capture privacy_modeosgetcwd Exception project_rootpersonal_api_keylogging basicConfiglogsetLevelDEBUGWARNINGr consumersatexitregisterjoinrangerappendstart)selfr$r&r1max_queue_sizer2r'r(r)r* max_retriesr3r,threadr9rJproject_api_keyr?r@r-r:rArBrCrIrEnconsumerr_E/var/www/visachat/venv/lib/python3.10/site-packages/posthog/client.py__init__$s           zClient.__init__cCF|pi}|pi}td|ttd|t||||d|d}|||S)N distinct_id propertiesz $identify timestampcontextrc$seteventuuidr0ID_TYPESdict_enqueuerXrcrdrgrfrjr@msgr_r_r`identify   zClient.identifycC||||||}|dS)N featureFlags get_deciderXrcgroupsperson_propertiesgroup_propertiesr@ resp_datar_r_r`get_feature_variantszClient.get_feature_variantscCrs)NfeatureFlagPayloadsrurwr_r_r`get_feature_payloadsr}zClient.get_feature_payloadscCs$||||||}|d|ddS)Nrtr~rtr~rurwr_r_r`get_feature_flags_and_payloadssz%Client.get_feature_flags_and_payloadscCs`td|t|dur |j}|rtd|tni}|||||d}t|j|jfd|ji|}|S)Nrcrx)rcrxryrzr@r,)r0rlr@rmrr$r&r:)rXrcrxryrzr@ request_datar{r_r_r`rvs zClient.get_decidec  Cs8|pi}|pi}td|ttd|ttd|t||||||d} |r1td|t|| dd<i} i} |r]z |j||| d} Wn)ty\} z|jd| WYd} ~ nd} ~ ww|jrk|j ||pfi| d d } | D] \}}|| d |<qod d | D}|r|| d<| ri| | d| d<| | | S)Nrcrdri)rdrfrgrcrirjrxz$groups)r@z0[FEATURE FLAGS] Unable to get feature variants: T)rxr@only_evaluate_locally $feature/cSsg|] \}}|dur|qS)Fr_).0keyvaluer_r_r` sz"Client.capture..z$active_feature_flags) r0rlrmrr|rHrM exceptionr5 get_all_flagsitemsrn)rXrcrirdrgrfrjrxsend_feature_flagsr@rpextra_propertiesfeature_variantsefeaturevariantactive_feature_flagsr_r_r`capturesH        zClient.capturecCrb)Nrcrdrhrerkror_r_r`r= rrz Client.setcCrb)Nrcrd $set_once)rfrgrcrrirjrkror_r_r`set_oncerrzClient.set_oncec Csx|pi}|pi}td|ttd|ttd|t|r#td|tnd||}d|||d||||d} || |S) N group_type group_keyrdrcz${}_{}z$groupidentify)z $group_type $group_keyz $group_set)rirdrcrfrgrj)r0rlrmformatrn) rXrrrdrgrfrjr@rcrpr_r_r`group_identify,s&      zClient.group_identifycCsB|pi}td|ttd|t||d||d|d}|||S)N previous_idrc)rcaliasz $create_alias)rdrfrgrirc)r0rlrn)rXrrcrgrfrjr@rpr_r_r`rQs   z Client.aliasc CsZ|pi}|pi}td|ttd|ttd|t||d<d|||||d}|||S)Nrcrdurlz $current_urlz $pageview)rirdrfrgrcrj)r0rlrmrrn) rXrcrrdrgrfrjr@rpr_r_r`pageds    z Client.pagec Cs,zy|pi}|durd|d<t}td|ttd|t|dur%t|}nt}|dus1|dkr:|jdWdSt |} t dd| ii|j d } | dd} | d  d | d  d | t |jd |jd|d|}||d|||||WSty} z|jd| WYd} ~ dSd} ~ ww)NFz$process_person_profilercrd)NNNz"No exception information availablervalues)rIrtyperz /project/z/person/)z$exception_typez$exception_messagez$exception_listz$exception_personURLz $exceptionzFailed to capture exception: )rr0rlrmr sysexc_inforMwarningr r rIrrr4r$rrHr) rXrrcrdrgrfrjrxrall_exceptions_with_traceri$all_exceptions_with_trace_and_in_apprr_r_r`capture_exception{sD         zClient.capture_exceptioncCs|jrdS|d}|durtjtd}td|ttd|dtt|}||d<d|vr=|d}|r=t ||d<| dsFi|d<d|dd <t |dd <|durY|j }|rad |dd <|j roi|d|j |d<t | d d|d <t|}|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)`)Fr?rfNtzrgrjrdzposthog-pythonz$libz $lib_versionTz$geoip_disablercz queueing: %szenqueued with blocking %s.ri)r*r,batchr-F)blockz enqueued %s.zanalytics-python queue is full)r?rnowrr0rmr isoformatpop stringify_idrrr@rArrMr1r2r3rr$r&r*r,r-r.putFullr)rXrpr@rfrjr_r_r`rns^            zClient._enqueuecCs(|j}|}||jd|dS)z4Forces a flush from the internal queue to the serverz$successfully flushed about %s items.N)r.qsizerTrMr1)rXr.sizer_r_r`flushsz Client.flushc CsJ|jD]}|z|WqtyYqw|jr#|jdSdS)zbEnds the consumer thread once the queue is empty. Blocks execution until finished N)rQpauserT RuntimeErrorr;stop)rXr^r_r_r`rTs   z Client.joincCs(|||jr|jdSdS)z2Flush all messages and cleanly shutdown the clientN)rrTrDcloserXr_r_r`shutdowns zClient.shutdownc Csz/t|jd|jd|jdd}|dpg|_dd|jD|_|dp%i|_|d p,i|_WnPty_}z$|j d krL|j d |j rKtd d d 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,flagscSs$i|]}|ddur|d|qS)rNr)rflagr_r_r` "sz.Client._load_feature_flags..r7r8iz[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)statusmessagez-[FEATURE FLAGS] Error loading feature flags: z`[FEATURE FLAGS] Fetching feature flags failed with following error. We will retry in %s seconds.r)rrJr$r&r5r6r7r8rrrMerrorr1rHrr9rrr_last_feature_flag_poll)rXresponserr_r_r`_load_feature_flagssH  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) rJrMrr5rr;is_aliverrr9rWrr_r_r`load_feature_flags?s zClient.load_feature_flagsrxryrzwarn_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) rr r7strrMrr1rr8) rX feature_flagrcrxryrzr flag_filtersr group_namefocused_group_propertiesr_r_r`_compute_flag_locallyJs4   zClient._compute_flag_locallyrxryrzrsend_feature_flag_eventsr@c Cs.|j||||||||d} | durdSt| S)Nr)get_feature_flagbool) rXrrcrxryrzrrr@rr_r_r`feature_enabledxs  zClient.feature_enabledc Cstd|ttd|ttd|t|jrdS|||||\}}|jdur-|jr-|d} |jr|jD]Y} | d|krz|j | ||||d} |j d|d| Wq5t ys} z|j d|d| WYd} ~ q5d} ~ wt y} z|j d | WYd} ~ q5d} ~ wwq5| du} | s|sz#|j|||||d } | |} | durd } |j d |d | Wnt y} z|j d| WYd} ~ nd} ~ ww|dt| }||j|vr|r|j|dd|d| d| d|| i||d|j||| S)Nrrcrxrxryrzz$Successfully computed flag locally: z -> zFailed to compute flag z locally: z7[FEATURE FLAGS] Error while computing variant locally: rxryrzr@Fz&Successfully computed flag remotely: #z -> #z-[FEATURE FLAGS] Unable to get flag remotely: _$feature_flag_called $feature_flag$feature_flag_responselocally_evaluatedrrxr@)r0rrlrmr?&_add_local_person_and_group_propertiesr5rJrrrMr1r rHrr|rrr>radd)rXrrcrxryrzrrr@rrrflag_was_locally_evaluatedr5feature_flag_reported_keyr_r_r`rs         zClient.get_feature_flag) match_valuerxryrzrrr@c Cs4|jrdS|dur|j|||||dd| d}d} d} |dur$|||} | du} | sg|sgz|||||| } | d|d} | dt|d} Wntyf}z|j d|WYd}~nd}~ww|dt| }||j |vr|r|j |dd |d | d | d | d || i|| d|j | || S)NFT)rxryrzrrr@rtr~:[FEATURE FLAGS] Unable to get feature flags and payloads: rrrrz$feature_flag_payloadrrr) r?r_compute_payload_locallyrrrlowerrHrMrr>rr)rXrrcrrxryrzrrr@rpayloadrresponses_and_payloadsrrr_r_r`get_feature_flag_payloads^      zClient.get_feature_flag_payloadcCsVd}|jdur |S|j|pi}|dpi}|dpi}|t|d}|S)Nrpayloads)r6rrr)rXrrrflag_definitionr flag_payloadsr_r_r`r-s zClient._compute_payload_locallyrxryrzrr@cCs|j||||||d}|dS)Nrrt)get_all_flags_and_payloads)rXrcrxryrzrr@rr_r_r`r9s zClient.get_all_flagsc Cs|jrdddS|||||\}}|j||||d\}}} ||d} | rS|sSz|j|||||d} | } W| StyR} z|jd| WYd} ~ | Sd} ~ ww| S)Nrrrr)r?r#_get_all_flags_and_payloads_locallyrvrHrMr) rXrcrxryrzrr@rrfallback_to_deciderflags_and_payloadsrr_r_r`rMs4  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}|||fS)NrcrxFrrTz;[FEATURE FLAGS] Error while computing variant and payload: ) r0rlrmr5rJrrrr rHrMr) rXrcrxryrzrrrrrmatched_payloadrr_r_r`rrs@      z*Client._get_all_flags_and_payloads_locallycCs|jSN)r5rr_r_r`feature_flag_definitionsszClient.feature_flag_definitionscCsHd|i|pi}i}|r |D]}d||i||pi||<q||fS)Nrcrr)rXrcrxryrzall_person_propertiesall_group_propertiesrr_r_r`rs  z-Client._add_local_person_and_group_properties)NNFrTNrrFr Fr!r"r#NNFTFr NFNNF)NNNNNN)NNNN) NNNNNNNFN)NNNNNNNN)NNNNNNN)$__name__ __module__ __qualname____doc__rK getLoggerrMrarqr|rrrvrr=rrrrrrnrrTrrrrrrrrrrrrrr_r_r_r`rs  q    9  %  ;C' 3   Y F  & & rcCs$t||sd|||}t|dS)z8Require that the named `field` has the right `data_type`z{0} must have {1}, got: {2}N) isinstancerAssertionError)namefield data_typerpr_r_r`r0s r0cCs"|durdSt|tr |St|Sr)rrr)valr_r_r`rs  r)3rRrKnumbersrFrrrrjrr dateutil.tzrsixrposthog.consumerrposthog.exception_capturer posthog.exception_utilsr r r posthog.feature_flagsr rposthog.pollerrposthog.requestrrrrrr posthog.utilsrrrrposthog.versionrr. ImportErrorr/Numberrlr<objectrr0rr_r_r_r`sD