o Rh@s"ddlZddlZddlZddlmZddlmZmZddlm Z mZm Z ddl m Z ddl mZmZddlmZddlZddlmZmZed Zd d Zd d ZddZddZddZddZddZddZdedee fddZ!de"fddZ#Gd d!d!eZ$d"d#Z%d$d%Z&d&d'Z'dS)(N) defaultdict)asdict is_dataclass)datedatetimetimezone)Decimal)AnyOptional)UUID)tzlocaltzutcposthogcCs|jdup |j|duS)z1Determines if a given datetime.datetime is naive.N)tzinfo utcoffset)dtrO/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/posthog/utils.pyis_naivesrcCs"|j|j|jddddS)z2Determines total seconds with python < 2.7 compat.ig.A) microsecondssecondsdays)deltarrr total_secondss"rcCs@t|rt|}t|dkr|jtdS|jtdS|S)z:Attempts to convert a naive datetime to an aware datetime.r)rrnowrreplacer r )rrrrrguess_timezones   rcCs|dr |ddS|S)N/)endswith)hostrrrremove_trailing_slash-s  r$c Cst|tr t|St|trt|St|tjttj t t t dfr#|St|t ttfr/t|Szt|dr?t|jr?|}nt|drMt|jrM|}Wntyh}ztd|WYd}~nd}~wwt|trrt|St|rt|t st|St|S)N model_dumpdictz)Could not serialize Pydantic-like model: ) isinstancerfloatr strsix string_typesboolnumbersNumberrrtypesetlisttuple _clean_listhasattrcallabler%r& TypeErrorlogdebug _clean_dictr_clean_dataclass_coerce_unicode)itemerrrclean3s4     r>cCsdd|DS)NcSsg|]}t|qSr)r>).0r<rrr Qsz_clean_list..r)list_rrrr3Psr3c CsRi}t|D]\}}zt|||<Wqty&td||t|Yqw|S)NzXDictionary values must be serializeable to JSON "%s" value %s of type %s is unsupported.)r* iteritemsr>r6r7warningr/)dict_datakvrrrr9Ts r9cCst|}t|}|SN)rr9) dataclass_rErrrr:csr:cmplxreturnc Csd}zt|tr|dd}W|St|tr|}W|SW|Sty?}zdtt|j}t d|WYd}~dSd}~ww)a In theory, this method is only called after many isinstance checks are carried out in `utils.clean`. When we supported Python 2 it was safe to call `decode` on a `str` but in Python 3 that will throw. So, we check if the input is bytes and only call `decode` in that case. Previously we would always call `decode` on the input That would throw an error. Then we would call `decode` on the stringified error That would throw an error. And then we would return `None` To avoid a breaking change, we can maintain the behavior that anything which did not have `decode` in Python 2 returns None. Nzutf-8strict:zError decoding: %s) r'bytesdecoder) Exceptionjoinmapargsr7rC)rJr< exceptionrrrr;is    r;cCs(zt|WdStjyYdSw)NTF)recompileerror)valuerrris_valid_regexs  rYcs(eZdZfddZfddZZS)SizeLimitedDictcstj|i|||_dSrH)super__init__max_size)selfr]rSkwargs __class__rrr\s zSizeLimitedDict.__init__cs(t||jkr |t||dSrH)lenr]clearr[ __setitem__)r^keyrXr`rrrdszSizeLimitedDict.__setitem__)__name__ __module__ __qualname__r\rd __classcell__rrr`rrZs rZcCs|jdur |jtjd}|S)Nr)rrrutc)date_objrrrconvert_to_datetime_awares rlcCst|t|vS)a Check if a string contains another string, ignoring case. Args: source: The string to search within search: The substring to search for Returns: bool: True if search is a substring of source (case-insensitive), False otherwise Examples: >>> str_icontains("Hello World", "WORLD") True >>> str_icontains("Hello World", "python") False r)casefold)sourcesearchrrr str_icontainsrqcCst|t|kS)a Check if a string equals another string, ignoring case. Args: value: The string to compare comparand: The string to compare with Returns: bool: True if value and comparand are equal (case-insensitive), False otherwise Examples: >>> str_iequals("Hello World", "hello world") True >>> str_iequals("Hello World", "hello") False rm)rX comparandrrr str_iequalsrrrt)(loggingr-rU collectionsr dataclassesrrrrrdecimalrtypingr r uuidr r* dateutil.tzr r getLoggerr7rrrr$r>r3r9r:r)r;r,rYrZrlrqrtrrrrs4