o NDi@sddlmZmZddlmZmZmZmZddlZddlmZ ddl m Z ddl m Z mZmZddlmZddlmZdd lmZeeZGd d d eZGd d d eZGdddeZeZGdddeZGdddeZ Gddde Z!GdddeZ"Gddde"Z#Gddde"Z$Gddde"Z%GdddeZ&Gd d!d!eZ'Gd"d#d#e"Z(Gd$d%d%eZ)Gd&d'd'eZ*Gd(d)d)eZ+Gd*d+d+eZGd,d-d-eZGd.d/d/eZ,Gd0d1d1eZ-Gd2d3d3eZ.Gd4d5d5e.Z/Gd6d7d7e.Z0Gd8d9d9eZ1Gd:d;d;eZ2Gdd?d?e2Z4Gd@dAdAe4Z5GdBdCdCe4Z6GdDdEdEe4Z7GdFdGdGeZ8GdHdIdIeZ9dJdKZ:GdLdMdMeZ;dS)N)deepcopycopy)datedatetime timedeltatimezoneN)UUID)util)SimpleDateType _cqltypesUserType)ValidationError)get_total_seconds)Durationc@sTeZdZddZeddZeddZddZd d Zd d Z d dZ ddZ dS)BaseValueManagercCs"||_||_||_d|_d|_dSNF)instancecolumnvalueprevious_valueexplicitselfrrrrT/var/www/Datamplify/venv/lib/python3.10/site-packages/cassandra/cqlengine/columns.py__init__s  zBaseValueManager.__init__cCs$|j|jo|jp|j|j SN)r _val_is_nullrrrrrrrdeleted&s$zBaseValueManager.deletedcCst|jr |j|jkSt|jtr8|j}|j|r(|j|j o'|j|jkS|jdur2|j|kS|j|jkSdS)z\ Indicates whether or not this value has changed. :rtype: boolean NF)rrr isinstancerBaseContainerColumn get_defaultrr default_valuerrrchanged*s       zBaseValueManager.changedcCst|j|_dSr)rrrrrrrreset_previous_value@sz%BaseValueManager.reset_previous_valuecC|jSrrrrrrgetvalCzBaseValueManager.getvalcCs||_d|_dS)NT)rrrvalrrrsetvalFs zBaseValueManager.setvalcCs d|_dSrr(rrrrdelvalJ zBaseValueManager.delvalcsBfdd}fdd}fdd}jjrt|||St||S)NcSr)r)slfrrrNz/BaseValueManager.get_property..cs |Sr)r-)r2r,rrrr3Os cr0r)r.r1rrrr3Pr4)r can_deleteproperty)r_get_set_delrrr get_propertyMs     zBaseValueManager.get_propertyN) __name__ __module__ __qualname__rr6rr%r&r)r-r.r:rrrrrs   rc@sbeZdZdZeZdZdZdZ dZ dZ dZ dZ dZ dZ dZ dZ dZ          d9ddZddZd d Zd d Zd dZddZddZddZddZddZddZeddZeddZedd Z d!d"Z!d#d$Z"d%d&Z#d'd(Z$ed)d*Z%ed+d,Z&ed-d.Z'ed/d0Z(d1d2Z)d3d4Z*ed5d6Z+ed7d8Z,dS):ColumnNrTFc Csl||_|p||_||_| |_||_||_||_||_||_d|_ d|_ | |_ d|_ t j|_t jd7_dS)N) partition_key primary_keyindex custom_indexdb_fielddefaultrequiredclustering_orderdiscriminator_column column_name_partition_key_indexstaticrr>instance_counterposition) rrAr@rBrDrErFrGrHrKrCrrrrs zColumn.__init__cCst|tr |j|jkStSrr r>rMNotImplementedrotherrrr__ne__  z Column.__ne__cCst|tr |j|jkStSrrNrPrrr__eq__rSz Column.__eq__cCst|tr |j|jkStSrrNrPrrr__lt__rSz Column.__lt__cCst|tr |j|jkStSrrNrPrrr__le__rSz Column.__le__cCst|tr |j|jkStSrrNrPrrr__gt__rSz Column.__gt__cCst|tr |j|jkStSrrNrPrrr__ge__rSz Column.__ge__cCst|Sr)idrrrr__hash__szColumn.__hash__cCs(|dur|jrtd|jp|j|S)zn Returns a cleaned and validated value. Raises a ValidationError if there's a problem Nz!{0} - None values are not allowed)rFr formatrIrDrrrrrvalidateszColumn.validatecC|S)z Converts data from the database into python values raises a ValidationError if the value can't be converted rr\rrr to_pythonszColumn.to_pythoncCr^)z; Converts python value into database value rr\rrr to_databaseszColumn.to_databasecCs |jduSr)rErrrr has_default zColumn.has_defaultcCr'rrArrrris_primary_keyzColumn.is_primary_keycCs|j Srrcrrrrr5zColumn.can_deletecCs"|jrt|jr |S|jSdSr)racallablerErrrrr"s  zColumn.get_defaultcCs |jrdnd}d|j|j|S)zF Returns a column definition for CQL table definition rKz {0} {1} {2})rKr[cqldb_type)rrKrrrget_column_def szColumn.get_column_defcCr'rrjrrrrcql_parameterized_typer*zColumn.cql_parameterized_typecCs ||_dS)z Sets the column name during document class construction This value will be ignored if db_field is set in __init__ N)rI)rnamerrrset_column_names zColumn.set_column_namecCs|jdur|jS|jS)z1 Returns the name of the cql name of this column N)rDrIrrrr db_field_nameszColumn.db_field_namecC d|jS)z# Returns the name of the cql index z index_{0}r[rprrrr db_index_name$s zColumn.db_index_namecCs |jp|jSr)rBrCrrrr has_index)s zColumn.has_indexcCs|Sr)get_cqlrrrrri-rfz Column.cqlcCrq)N"{0}"rrrrrrru1s zColumn.get_cqlcCs|duS)zQ determines if the given value equates to a null value for the given column type Nrr+rrrr4rfzColumn._val_is_nullcCsgSrrrrrr sub_types8szColumn.sub_typescCs t|jSr)r rjrrrrcql_type<rbzColumn.cql_type) FFFNNFNFFF)-r;r<r=rjr value_managerrL_python_type_hashablerAr@rBrCrDrErFrGrHrKrrRrTrUrVrWrXrZr]r_r`r6rardr5r"rkrmrorprsrtrirurrwrxrrrrr>Xs             r>c$eZdZdZdZfddZZS)Blobz# Stores a raw binary value blobcs6t|ttfstdt|tt||}t|S)Nzexpecting a binary, got a %s)r bytes bytearray ExceptiontypesuperBytesr`rrr, __class__rrr`GszBlob.to_database)r;r<r=__doc__rjr` __classcell__rrrrr|Ar|c@eZdZdZdZdS)Inetz5 Stores an IP address in IPv4 or IPv6 format inetNr;r<r=rrjrrrrrSrcs2eZdZdZdZdfdd ZfddZZS) Textz' Stores a UTF-8 encoded string textNc s|dur |ddr dn||_||_|jdur |jdkr td|jdur.|jdkr.td|jdurB|jdurB|j|jkrBtdtt|jd i|dS) a :param int min_length: Sets the minimum length of this string, for validation purposes. Defaults to 1 if this is a ``required`` column. Otherwise, None. :param int max_length: Sets the maximum length of this string, for validation purposes. NrFFr?rz-Minimum length is not allowed to be negative.z-Maximum length is not allowed to be negative.z:Maximum length must be greater or equal to minimum length.r)get min_length max_length ValueErrorrrr)rrrkwargsrrrr`s(     z Text.__init__cstt||}t|ttfs|durtd|jt ||j dur6|r6t ||j kr6td|j|j |j rO|j r>|rEt ||j krOtd|j|j |S)Nz{0} {1} is not a stringz!{0} is longer than {1} charactersz"{0} is shorter than {1} characters) rrr]r strrr r[rIrrlenrr\rrrr]}s z Text.validate)NN)r;r<r=rrjrr]rrrrrrZs rcr{)Asciiz, Stores a US-ASCII character string asciicsRtt||}|r't|tfr|ntt|}ttd |s't d ||S)am Only allow ASCII and None values. Check against US-ASCII, a.k.a. 7-bit ASCII, a.k.a. ISO646-US, a.k.a. the Basic Latin block of the Unicode character set. Source: https://github.com/apache/cassandra/blob /3dcbe90e02440e6ee534f643c7603d50ca08482b/src/java/org/apache/cassandra /serializers/AsciiSerializer.java#L29 z{!r} is not an ASCII string.) rrr]r rmapordsetrange issupersetr r[)rrcharsetrrrr]s zAscii.validater;r<r=rrjr]rrrrrrrrc4eZdZdZdZfddZddZddZZS) Integerz. Stores a 32-bit signed integer value intc Ltt||}|durdSzt|WSttfy%td|j|wNz,{0} {1} can't be converted to integral value) rrr]r TypeErrorrr r[rIrrrrr] zInteger.validatecC ||Srr]r\rrrr_r/zInteger.to_pythoncCrrrr\rrrr`r/zInteger.to_database r;r<r=rrjr]r_r`rrrrrrs   rc@r)TinyIntzr Stores an 8-bit signed integer value .. versionadded:: 2.6.0 requires C* 2.2+ and protocol v4+ tinyintNrrrrrrrc@r)SmallIntzr Stores a 16-bit signed integer value .. versionadded:: 2.6.0 requires C* 2.2+ and protocol v4+ smallintNrrrrrrrrc@r)BigIntz. Stores a 64-bit signed integer value bigintNrrrrrrrrcr) VarIntz/ Stores an arbitrary-precision integer varintc rr) rrr]rrrr r[rIrrrrr]s  zVarInt.validatecCrrrr\rrrr_r/zVarInt.to_pythoncCrrrr\rrrr`r/zVarInt.to_databaserrrrrrs   rcseZdZfddZZS)CounterValueManagercs0tt|||||jpd|_|jpd|_dS)Nr)rrrrrrrrrrs zCounterValueManager.__init__)r;r<r=rrrrrrrsrcs0eZdZdZdZeZ   dfdd ZZS)CounterzB Stores a counter that can be incremented and decremented counterFNcs tt|jdd||d|ddS)NFr)rAr@rBrDrErF)rrr)rrBrDrFrrrrs  zCounter.__init__)FNF) r;r<r=rrjrryrrrrrrrsrcs2eZdZdZdZdZ ddZfddZZS)DateTimez! Stores a datetime value timestampFcCsz|durdSt|tr"tjr |j}|dd}|t||dS|St|tr1t|ddStj|t j dj ddS)N) microseconds)tztzinfo) r rrtruncate_microseconds microsecondrr timetuple fromtimestamprutcreplace)rrus truncated_usrrrr_s   zDateTime.to_pythoncstt||}|durdSt|ts+t|tr"t|j|j|j}n t d |j |tddd|j d}|j r?t |j |nd}tt |||dS)Nz"{0} '{1}' is not a datetime objectir?rrr)rrr`r rryearmonthdayr r[rIrr utcoffsetr)rrepochoffsetrrrr`,s  zDateTime.to_database) r;r<r=rrjrr_r`rrrrrr s rc@s$eZdZdZdZddZddZdS)Datez Stores a simple date, with no time-of-day .. versionchanged:: 2.6.0 removed overload of Date and DateTime. DateTime is a drop-in replacement for legacy models requires C* 2.2+ and protocol v4+ rcCs2|durdSt|tjr|nt|}|jtjSr)r r rdays_from_epochr EPOCH_OFFSET_DAYS)rrdrrrr`Gs zDate.to_databasecCs8|durdSt|tjr|St|tr|}t|Sr)r r rrrr\rrrr_Ps   zDate.to_pythonN)r;r<r=rrjr`r_rrrrr;s   rcs0eZdZdZdZfddZfddZZS)Timez Stores a timezone-naive time-of-day, with nanosecond precision .. versionadded:: 2.6.0 requires C* 2.2+ and protocol v4+ timec6tt||}|durdSt|tjr|St|Srrrr`r r r\rrrr`cszTime.to_databasecrrrr\rrrr_js   zTime.to_python)r;r<r=rrjr`r_rrrrrrYs  rcr{)rz} Stores a duration (months, days, nanoseconds) .. versionadded:: 3.10.0 requires C* 3.10+ and protocol v4+ durationcs<tt||}|durdSt|tstd|j||S)Nz {0} {1} is not a valid Duration.)rrr]r _Durationrr[rIrrrrr]|s  zDuration.validaterrrrrrrsrcr) rz# Stores a type 1 or 4 UUID uuidcsdtt||}|durdSt|tr|St|tr)zt|WSty(Ynwtd|j |)Nz{0} {1} is not a valid uuid) rrr]r _UUIDrrr r[rIrrrrr]s    z UUID.validatecCrrrr\rrrr_r/zUUID.to_pythoncCrrrr\rrrr`r/zUUID.to_databaserrrrrrs  rc@r)TimeUUIDz# UUID containing timestamp timeuuidNrrrrrrsrcs,eZdZdZdZfddZddZZS)Booleanz. Stores a boolean True or False value booleancs$tt||}|durt|}|S)z" Always returns a Python boolean. N)rrr]boolr\rrrr]szBoolean.validatecCrrrr\rrrr_r/zBoolean.to_python)r;r<r=rrjr]r_rrrrrrs   rcs,eZdZfddZddZddZZS) BaseFloatc sLtt||}|durdSzt|WSttfy%td|j|w)Nz{0} {1} is not a valid float) rrr]floatrrr r[rIr\rrrr]rzBaseFloat.validatecCrrrr\rrrr_r/zBaseFloat.to_pythoncCrrrr\rrrr`r/zBaseFloat.to_database)r;r<r=r]r_r`rrrrrrs  rc@r)Floatz8 Stores a single-precision floating-point value rNrrrrrrrrc@r)Doublez8 Stores a double-precision floating-point value doubleNrrrrrrrrcr) Decimalz3 Stores a variable precision decimal value decimalcsxddlm}ddlm}tt||}|durdSzt|tr'|t|WS||WS|y;td |j |w)Nr)r)InvalidOperationz%{0} '{1}' can't be coerced to decimal) rrrrr]r rreprr r[rI)rr_Decimalrr,rrrr]s  " zDecimal.validatecCrrrr\rrrr_r/zDecimal.to_pythoncCrrrr\rrrr`r/zDecimal.to_databaserrrrrrs   rcsTeZdZdZfddZfddZddZdd Zed d Z ed d Z Z S)BaseCollectionColumnz} Base Container type for collection-like columns. http://cassandra.apache.org/doc/cql3/CQL-3.0.html#collections c sg}|D];}t|tr tnt}||tstd|f|jdur'td|ft|tr/|n|}t|tr:|||q||_ t t|j di|dS)zJ :param types: a sequence of sub types in this collection z%s is not a column classNz%s is an abstract typer) r r issubclassr>r rjr_freeze_db_typeappendtypesrr)rrr instancestinheritance_comparatorinstrrrrs    zBaseCollectionColumn.__init__cs8tt||}|durt|dkrtd|j|S)Niz3{0} Collection can't have more than 65535 elements.)rrr]rr r[rIr\rrrr] szBaseCollectionColumn.validatecCs| Srrr+rrrrr*z!BaseCollectionColumn._val_is_nullcCs"|jdsd|jf|_dSdS)Nfrozen frozen<%s>)rj startswithrrrrrs z$BaseCollectionColumn._freeze_db_typecCr'r)rrrrrrwrezBaseCollectionColumn.sub_typescCs"t|jjdd|jDS)NcSg|]}|jqSrrx.0crrr !z1BaseCollectionColumn.cql_type..)r rr;lowerapply_parametersrrrrrrxs"zBaseCollectionColumn.cql_type) r;r<r=rrr]rrr6rwrxrrrrrrs   rcs<eZdZdZfddZfddZddZdd ZZS) Tuplez Stores a fixed-length set of positional values http://docs.datastax.com/en/cql/3.1/cql/cql_reference/tupleType.html csF|stdtt|j|fi|dddd|jD|_dS)zJ :param args: column types representing tuple composition z*Tuple must specify at least one inner typez tuple<{0}>, css|]}|jVqdSrrl)rtyprrr 1sz!Tuple.__init__..N)rrrrr[joinrrj)rargsrrrrr*s"zTuple.__init__csjtt||}|durdSt|t|jkr(td|ddd|jDftddt|j|DS)Nz3Value %r has more fields than tuple definition (%s)rcss|]}|VqdSrr)rrrrrr9sz!Tuple.validate..cs|] \}}||VqdSrrrrvrrrr:) rrr]rrr rtupleziprrrrr]3szTuple.validatecCs(|durtStddt|j|DS)Ncsrr)r_rrrrr?rz"Tuple.to_python..rr rr\rrrr_<szTuple.to_pythoncCs&|durdStddt|j|DS)Ncsrr)r`rrrrrDrz$Tuple.to_database..r r\rrrr`AzTuple.to_database) r;r<r=rrr]r_r`rrrrrr$s    rc@s eZdZdS)r!N)r;r<r=rrrrr!Gsr!csFeZdZdZdZdeffdd ZfddZdd Zd d Z Z S) Setz Stores a set of unordered, unique values http://www.datastax.com/documentation/cql/3.1/cql/cql_using/use_set_t.html FTc sT||_tt|j|ffd|i||jd|_|jjs tdd|jj |_ dS)z :param value_type: a column class indicating the types of the value :param strict: sets whether non set values will be coerced to set type on validation, or raise a validation error, defaults to True rErz?Cannot create a Set with unhashable value type (see PYTHON-494)zset<{0}>N) strictrr rr value_colrzr r[rj)r value_typer rErrrrrTs  z Set.__init__cstt|}|durdSjrttjfnttjttf}t ||s7jr.t d j |t d j |d|vrCt d j tfdd|DS)Nz{0} {1} is not a set objectz){0} {1} cannot be coerced to a set objectz{0} None not allowed in a setc3|] }j|VqdSrrr]rrrrrrpzSet.validate..) rr r]r rr SortedSetlistrr r r[rI)rrr,rrrrr]as z Set.validatecs$|durtStfdd|DS)Nc3rrrr_rrrrrurz Set.to_python..rr\rrrr_rsz Set.to_pythoncs"|durdStfdd|DS)Nc3rrrr`rrrrrzrz"Set.to_database..rr\rrrr`wszSet.to_database) r;r<r=rrzrrr]r_r`rrrrrr Ks r cDeZdZdZdZeffdd ZfddZddZd d Z Z S) Listzz Stores a list of ordered values http://www.datastax.com/documentation/cql/3.1/cql/cql_using/use_list_t.html Fc s>tt|j|ffd|i||jd|_d|jj|_dS)zU :param value_type: a column class indicating the types of the value rErz list<{0}>N)rrrrrr[rj)rrrErrrrrs z List.__init__cshtt|}|durdSt|tttfstdj |d|vr+tdj fdd|DS)Nz{0} {1} is not a list objectz!{0} None is not allowed in a listcg|]}j|qSrrrrrrrz!List.validate..) rrr]r rrrr r[rIrrrrr]sz List.validatecs|durgSfdd|DS)Ncrrrrrrrrrz"List.to_python..rr\rrrr_zList.to_pythoncs|durdSfdd|DS)Ncrrrrrrrrrz$List.to_database..rr\rrrr`rzList.to_database) r;r<r=rrzrrr]r_r`rrrrrr}s  rcr) Mapzx Stores a key -> value map (dictionary) https://docs.datastax.com/en/dse/6.7/cql/cql/cql_using/useMap.html Fc sbtt|j||ffd|i||jd|_|jd|_|jjs$tdd|jj |jj |_ dS)z :param key_type: a column class indicating the types of the key :param value_type: a column class indicating the types of the value rErr?z=Cannot create a Map with unhashable key type (see PYTHON-494)z map<{0}, {1}>N) rrrrkey_colrrzr r[rj)rkey_typerrErrrrrs   z Map.__init__csptt|}|durdSt|ttjfstdj |d|vr+tdj tfdd| DS)Nz{0} {1} is not a dict objectz {0} None is not allowed in a mapc3,|]\}}j|j|fVqdSr)rr]rrkrrrrr*zMap.validate..) rrr]r dictr OrderedMapr r[rIitemsrrrrr]sz Map.validatecs2|duriS|durtfdd|DSdS)Nc3r!r)rr_rr"rrrrr$z Map.to_python..r%r'r\rrrr_s z Map.to_pythoncs&|durdStfdd|DS)Nc3r!r)rr`rr"rrrrr$z"Map.to_database..r(r\rrrr`r zMap.to_database) r;r<r=rrzr%rr]r_r`rrrrrrs  rc@s eZdZeddZddZdS)UDTValueManagercCsZ|jr |j|jkS|j}|j|s|j|kS|jdur+|j|j o*|jSdSr)rrrrr"rhas_changed_fieldsr#rrrr%s     zUDTValueManager.changedcCs$|jdur |jt|j|_dSr)rreset_changed_fieldsrrrrrrr&s  z$UDTValueManager.reset_previous_valueN)r;r<r=r6r%r&rrrrr)s  r)csXeZdZdZeZfddZeddZeddZ fdd Z d d Z d d Z Z S)UserDefinedTypea User Defined Type column http://www.datastax.com/documentation/cql/3.1/cql/cql_using/cqlUseUDT.html These columns are represented by a specialization of :class:`cassandra.cqlengine.usertype.UserType`. Please see :ref:`user_types` for examples and discussion. c s.||_d||_tt|jdi|dS)zq :param type user_type: specifies the :class:`~.cqlengine.usertype.UserType` model of the column rNr) user_type type_namerjrr,r)rr-rrrrrszUserDefinedType.__init__cCst|jjSr)rr-_fieldsvaluesrrrrrwszUserDefinedType.sub_typescCs<tjd|jdd|jjDdd|jjDdS)NrhcSrr)rprrrrrrz,UserDefinedType.cql_type..cSrrrrrrrrr)keyspaceudt_name field_names field_types)r make_udt_classr-r.r/r0rrrrrxszUserDefinedType.cql_typecs(tt||}|durdS||Sr)rr,r]rrrrr]s zUserDefinedType.validatecCV|durdSt|}|jjD]\}}||dust|tr(|||||<q|Sr)rr-r/r'r r!r_rr copied_valuernfieldrrrr_ zUserDefinedType.to_pythoncCr6r)rr-r/r'r r!r`r7rrrr`r:zUserDefinedType.to_database)r;r<r=rr)ryrr6rwrxr]r_r`rrrrrr,s     r,cCs4|jD]}t||qt|tr||jdSdSr)rw resolve_udtsr r,rr-)col_defout_listcolrrrr;#s   r;cs,eZdZdZfddZeddZZS)_PartitionKeysTokenzr virtual column representing token of partition columns. Used by filter(pk__token=Token(...)) filters cs&t|j|_tt|jdddS)NT)r@)r_partition_keysr0partition_columnsrr?r)rmodelrrrr0sz_PartitionKeysToken.__init__cCsdddd|jDS)Nz token({0})rcSsg|]}d|jqS)rvrrrrrrr6rz5_PartitionKeysToken.db_field_name..)r[rrArrrrrp4sz!_PartitionKeysToken.db_field_name)r;r<r=rrr6rprrrrrr?*s  r?)r|rrrrrrrrrrrrrrrrrrrrrrr!r rrr)r,r;r?rrrrs\      ;j0  /1#2&/=