o NDi@sddlZddlZddlmZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Zdd lmZdd lmZeeZd d ZGd ddeZGdddeZGdddeZGdddeZdZGdddeZGdddeZGdddeZ GdddeZ!GdddeZ"Gdd d eZ#Gd!d"d"eZ$Gd#d$d$eZ%Gd%d&d&eZ&Gd'd(d(e j'Z(Gd)d*d*eZ)Gd+d,d,eZ*Gd-d.d.e+Z,Gd/d0d0e*e,d1Z-dS)2N)warn)CQLEngineExceptionValidationError)columns connection)query) DoesNotExist)MultipleObjectsReturned) protect_name) OrderedDictcCsDt|j|f|}z|j|_|j|_|j|_W|Sty!Y|SwN)type__name__ __abstract____discriminator_value____default_ttl__AttributeError)modelattrsnew_typerS/var/www/Datamplify/venv/lib/python3.10/site-packages/cassandra/cqlengine/models.py_clone_model_classs  rc@ eZdZdS)ModelExceptionNr __module__ __qualname__rrrrr*rc@r)ModelDefinitionExceptionNrrrrrr .rr c@r)PolymorphicModelExceptionNrrrrrr!2rr!c@r)UndefinedKeyspaceWarningNrrrrrr"6rr"c@(eZdZdZddZddZddZdS) hybrid_classmethodzv Allows a method to behave as both a class method and normal instance method depending on how it's called cCs||_||_dSr ) clsmethod instmethod)selfr%r&rrr__init__As zhybrid_classmethod.__init__cCs$|dur |j||S|j||Sr )r%__get__r&r'instanceownerrrrr)Eszhybrid_classmethod.__get__cOt)z? Just a hint to IDEs that it's ok to call this NotImplementedErrorr'argskwargsrrr__call__Kszhybrid_classmethod.__call__N)rrr__doc__r(r)r3rrrrr$<s  r$c@ eZdZdZddZddZdS)QuerySetDescriptorzc returns a fresh queryset for the given model it's declared on everytime it's accessed cCsX|jrtd||}|jr*|js*|j|j}}|js|jr*|j di||j iS|S)z :rtype: ModelQuerySet z.cannot execute queries against abstract modelsNr) rr __queryset___is_polymorphic_is_polymorphic_base_discriminator_column_name_discriminator_column partition_keyindexfilterr)r'objrquerysetnamecolumnrrrr)Xs   zQuerySetDescriptor.__get__cOr-)z^ Just a hint to IDEs that it's ok to call this :rtype: ModelQuerySet r.r0rrrr3iszQuerySetDescriptor.__call__Nrrrr4r)r3rrrrr6Rs r6c@r5)ConditionalDescriptor( returns a query set descriptor cs0r fdd}|Sfdd}|S)Ncs4t|dkr |d}n jjdi|j}|_S)Nrr)lenobjectsiff _conditional)prepared_conditionalunprepared_conditionals conditionalsr+rrconditional_setterxs  z9ConditionalDescriptor.__get__..conditional_settercsjjdi|j}|_S)Nr)rGrHrI)rKrL)rqsrrrNsr7)r'r+rrNr)r+rrOrr)vs   zConditionalDescriptor.__get__cOr-r r.r0rrrr3zConditionalDescriptor.__call__NrCrrrrrDrs rDc@r5) TTLDescriptorrEc.r fdd}|S||fdd}|S)Nc |_Sr _ttltsrMrr ttl_setterz)TTLDescriptor.__get__..ttl_settercrTr rUrWrOrrrYrZrP)r'r+rrYrr+rOrr)s    zTTLDescriptor.__get__cOr-r r.r0rrrr3rQzTTLDescriptor.__call__NrCrrrrrRs rRc@r5)TimestampDescriptorzC returns a query set descriptor with a timestamp specified csr fdd}|S|jjS)NcrTr ) _timestamprWrMrrtimestamp_setterrZz5TimestampDescriptor.__get__..timestamp_setter)rG timestamp)r'r+rr_rrMrr)s zTimestampDescriptor.__get__cOr-r r.r0rrrr3rQzTimestampDescriptor.__call__NrCrrrrr] r]c@r5)IfNotExistsDescriptorzK return a query set descriptor with a if_not_exists flag specified cr dfdd }|S|jjS)NTcrTr )_if_not_existsiferMrrifnotexists_setterrZz9IfNotExistsDescriptor.__get__..ifnotexists_setterT)rG if_not_exists)r'r+rrgrrMrr)zIfNotExistsDescriptor.__get__cOr-r r.r0rrrr3rQzIfNotExistsDescriptor.__call__NrCrrrrrbrarbc@r5)IfExistsDescriptorzG return a query set descriptor with a if_exists flag specified crc)NTcrTr ) _if_existsrerMrrifexists_setterrZz3IfExistsDescriptor.__get__..ifexists_setterrh)rG if_exists)r'r+rrmrrMrr)rjzIfExistsDescriptor.__get__cOr-r r.r0rrrr3rQzIfExistsDescriptor.__call__NrCrrrrrkrarkc@r5)ConsistencyDescriptorz` returns a query set descriptor if called on Class, instance if it was an instance call crS)NcrTr )__consistency__ consistencyrMrrconsistency_setterrZz9ConsistencyDescriptor.__get__..consistency_settercrTr ) _consistencyrqr[rrrsrZrP)r'r+rrsrr\rr)s    zConsistencyDescriptor.__get__cOr-r r.r0rrrr3rQzConsistencyDescriptor.__call__NrCrrrrros roc@r5)UsingDescriptorzK return a query set descriptor with a connection context specified csr dfdd }|S|jjS)Ncs|r|_Sr ) _connectionrrMrr using_settersz-UsingDescriptor.__get__..using_setterr )rGusing)r'r+rrwrrMrr)szUsingDescriptor.__get__cOr-r r.r0rrrr3rQzUsingDescriptor.__call__NrCrrrrrus ruc@r#) ColumnQueryEvaluatorz Wraps a column and allows it to be used in comparator expressions, returning query operators ie: Model.column == 5 cCs ||_dSr rBr'rBrrrr(s zColumnQueryEvaluator.__init__cCs|jjSr )rB db_field_namer'rrr __unicode__sz ColumnQueryEvaluator.__unicode__cC|jSr rzr}rrr _get_columnsz ColumnQueryEvaluator._get_columnN)rrrr4r(r~rrrrrrys  ryc@s0eZdZdZddZddZddZdd Zd S) ColumnDescriptorz Handles the reading and writing of column values to and from a model instance's value manager, as well as creating comparator queries cCs||_t|j|_dS)zV :param column: :type column: columns.Column :return: N)rBryquery_evaluatorr{rrrr(!szColumnDescriptor.__init__cCs.z |j|jjWSty|jYSw)z Returns either the value or column, depending on if an instance is provided or not :param instance: the model instance :type instance: Model )_valuesrB column_namegetvalrrr*rrrr)*s   zColumnDescriptor.__get__cCs |r |j|jj|Std)z Sets the value on an instance, raises an exception with classes TODO: use None instance to create update statements zcannot reassign column values)rrBrsetvalr)r'r+valuerrr__set__7szColumnDescriptor.__set__cCs8|r|jjr|j|jjdStd|jjdS)z< Sets the column value to None, if possible zcannot delete {0} columnsN)rB can_deleterrdelvalrformat)r'r+rrr __delete__As zColumnDescriptor.__delete__N)rrrr4r(r)rrrrrrrs   rc@s,eZdZdZGdddeZGdddeZeZ e Z e Z eZeZeZeZeZdZdZdZdZdZdZdZej Z!ej"Z#dZ$dZ%dZ&dZ'dZ(dZ)d d Z*d d Z+d dZ,e-ddZ.e-ddZ/e-ddZ0e-ddZ1dWddZ2ddZ3e-ddZ4e-ddZ5e-dd Z6d!d"Z7d#d$Z8e-dXd%d&Z9e-d'd(Z:d)d*Z;d+d,Zd1d2Z?d3d4Z@d5d6ZAd7d8ZBd9d:ZCd;d<ZDe-d=d>ZEe-d?d@ZFe-dAdBZGe-dCdDZHdEdFZIdGdHZJdIdJZKdKdLZLdMdNZMe-dOdPZNdQdRZOePeNeOZQe-dSdTZRdUdVZSePeReSZTdS)Y BaseModelzZ The base model class, don't inherit from this, inherit from Model, defined below c@r)zBaseModel.DoesNotExistNrrrrrr Qrr c@r)z!BaseModel.MultipleObjectsReturnedNrrrrrr Trr NFTcKsd|_d|_d|_d|_tj|_d|_d|_i|_ |j D]3\}}|j r)| nd}|||}|dus;t|tjr@||}||||}||v|_||j |<qdS)NF)rVr^rI_batchrNOT_SET_timeout _is_persistedrvr_columnsitems has_default get_defaultget isinstancerBaseContainerColumn to_python value_managerexplicit)r'valuesrArBcolumn_defaultr value_mngrrrrr(s"    zBaseModel.__init__c*djjdfddjDS)Nz{0}({1}), c3s,|]}|jkrd|t|VqdS)z {0}={1!r}N)r:rgetattr.0kr}rr s z%BaseModel.__repr__..)r __class__rjoin_defined_columnskeysr}rr}r__repr__s zBaseModel.__repr__cr)z@ Pretty printing of models by their primary key z {0} <{1}>rc3s"|] }d|t|VqdS)z{0}={1}N)rrrr}rrrs z$BaseModel.__str__..)rrrr _primary_keysrr}rr}r__str__s zBaseModel.__str__cCs |||Sr )_key_serializer)cls pk_valuesprotocol_versionrrr_routing_key_from_values z"BaseModel._routing_key_from_valuescs(jstdfdddS)NzN_discover_polymorphic_submodels can only be called on polymorphic base classescs6|js|jdur|j|j<|D]}|qdSr )r9r_discriminator_map__subclasses__)klasssubklass _discoverrrrrs    z._discover)r9rrrrr_discover_polymorphic_submodelss z)BaseModel._discover_polymorphic_submodelscCs|jstd|j|S)NzP_get_model_by_discriminator_value can only be called on polymorphic base classes)r9rrr)rkeyrrr!_get_model_by_discriminator_values z+BaseModel._get_model_by_discriminator_valuecsjrtfdd|D}jre|j}|dur!tdjr&nj}| |durH| | |durHtd ||j t sWtd j j tfdd|D}nd i|}|jdd |S) z method used to construct instances from query results this is where polymorphic deserialization occurs c3s&|]\}}j|||fVqdSr )_db_maprrrvrrrrs$z0BaseModel._construct_instance..Nz+discriminator value was not found in valuesz3unrecognized discriminator column {0} for class {1}z{0} is not a subclass of {1}c3s*|]\}}|jvr||fVqdSr rrr)rrrrs(Tforcer)rdictrr8rr:r!r9_polymorphic_baserrrr issubclass_set_persisted)rrdisc_key poly_baser+r)rrr_construct_instances0       zBaseModel._construct_instancecs6fdd|jDD] }|d|_q d|_dS)Ncsg|] }|js r|qSrchanged)rrrrr z,BaseModel._set_persisted..FT)rrreset_previous_valuerr)r'rrrrrrs zBaseModel._set_persistedcs"jsdStfddjDS)z Called by the save function to check if this should be persisted with update or insert :return: Fcsg|] }j|j qSr)rrrr}rrrrz)BaseModel._can_update..)rallrr}rr}r _can_updateszBaseModel._can_updatecCs |jptS)zU Returns the manual keyspace, if set, otherwise the default keyspace ) __keyspace__DEFAULT_KEYSPACErrrr _get_keyspaces zBaseModel._get_keyspacecCs |j|S)z Returns the column matching the given name, raising a key error if it doesn't exist :param name: the name of the column to return :rtype: Column )rrrArrrrs zBaseModel._get_columncCs|j|j||S)z= Returns the column, mapped by db_field name )rrrrrrr_get_column_by_db_namesz BaseModel._get_column_by_db_namecsPjjkrdStj}tj}||krdStfdd|DS)NFc3s(|]}t|dt|dkVqdSr )r)rrotherr'rrrs&z#BaseModel.__eq__..)rsetrrr)r'rr other_keysrrr__eq__s zBaseModel.__eq__cCs || Sr )r)r'rrrr__ne__! zBaseModel.__ne__cCs8t|}|r|}|stddt||S|S)z Returns the column family name if it's been defined otherwise, it creates it from the module and class name zModel keyspace is not set and no default is available. Set model keyspace or setup connection before attempting to generate a query.z{0}.{1})r _raw_column_family_namerrr)rinclude_keyspacecf_namekeyspacerrrcolumn_family_name$s zBaseModel.column_family_namecs|js`|jr-|jrtdt|j|_|jS|j}|j|kr'td|j||_|jS|jr<|j s<|j |_|jSt dfdd}||j}|dd}|}t dd|}||_|jS) Nz;Model __table_name_case_sensitive__ will be removed in 4.0.zzModel __table_name__ will be case sensitive by default in 4.0. You should fix the __table_name__ value of the '{0}' model.z([a-z])([A-Z])csdd|S)NcSsd|d|dS)Nz{0}_{1})rgrouplower)rrrrFzEBaseModel._raw_column_family_name....)sub)s camelcaserrrFz3BaseModel._raw_column_family_name..iz^_+) _table_name__table_name____table_name_case_sensitive__rPendingDeprecationWarningrrrr8r9rrrecompiler)r table_nameccaserrrrr4s0      z!BaseModel._raw_column_family_namecCs||j|_dS)zKFunction to change a column value without changing the value manager statesN)rr)r'rArrrr_set_column_valueQszBaseModel._set_column_valuecCsZ|jD]%\}}t||}|dur|j|js|jr|}||}|||qdS)z7 Cleans and validates the field values N) rrrrrrrvalidater)r'rAcolrvalrrrrUs  zBaseModel.validateccs|jD]}|VqdS)z Iterate over column ids. Nr)r' column_idrrr__iter__aszBaseModel.__iter__cCs*t|tst||jvrtt||S)z Returns column's value. )rstr TypeErrorrrKeyErrorr)r'rrrr __getitem__fs  zBaseModel.__getitem__cCs,t|tst||jvrtt|||S)z Sets a column's value. )rrrrrrsetattr)r'rrrrr __setitem__ns  zBaseModel.__setitem__cCs*z|jWSt|j|_|jYS)zF Returns the number of columns defined on that model. )_lenrFrrr}rrr__len__vs  zBaseModel.__len__cCsdd|DS)z Returns a list of column IDs. cSg|]}|qSrrrrrrrrz"BaseModel.keys..rr}rrrrszBaseModel.keyscfddDS)z Returns list of column values. csg|]}|qSrrrr}rrrsz$BaseModel.values..rr}rr}rrzBaseModel.valuescr)z+ Returns a list of column ID/value tuples. csg|]}||fqSrrrr}rrrz#BaseModel.items..rr}rr}rrrzBaseModel.itemscCs8|jpi}|jD]\}}|t||d||<q |S)z1 Returns a map of column names to cleaned values N)_dynamic_columnsrr to_databaser)r'rrArrrr_as_dicts zBaseModel._as_dictcKs>t|t|j}|rtd||jjdi|S)z Create an instance of this model in the database. Takes the model column values as keyword arguments. Setting a value to `None` is equivalent to running a CQL `DELETE` on that column. Returns the instance. zIncorrect columns passed: {0}Nr)rrrrrrGcreate)rr2 extra_columnsrrrr s zBaseModel.createcCs |jS)z Returns a queryset representing all stored objects This is a pass-through to the model objects().all() )rGrrrrrrs z BaseModel.allcO|jj|i|S)z Returns a queryset based on filter parameters. This is a pass-through to the model objects().:method:`~cqlengine.queries.filter`. )rGr>rr1r2rrrr>zBaseModel.filtercOr )z Returns a single object based on the passed filter constraints. This is a pass-through to the model objects().:method:`~cqlengine.queries.get`. )rGrrrrrrrz BaseModel.getcCs|jdus Jd||_|S)zu Sets a timeout for use in :meth:`~.save`, :meth:`~.update`, and :meth:`~.delete` operations N/Setting both timeout and batch is not supported)rr)r'timeoutrrrrszBaseModel.timeoutc Csr|jr|jr tdt||j|j||j|j||j |j |j |j |j |j|j|jd |d|_ |S)a Saves an object to the database. .. code-block:: python #create a person instance person = Person(first_name='Kimberly', last_name='Eggleston') #saves it to Cassandra person.save() z"cannot save polymorphic base model)batchttlr`rrri conditionalrrnN)r8r9r!rr:rr __dmlquery__rrrVr^rprdrIrrlsaverr}rrrrs&  zBaseModel.savec Ks|D]8\}}|j|}|durtd|j|jj||jr6t ||}||kr6td||j|jjt |||q|j rO|j rGt dt ||j|j||j|j||j|j|j|j|j|j|jd |d|_|S)ao Performs an update on the model instance. You can pass in values to set on the model for updating, or you can call without values to execute an update against any modified fields. If no fields on the model have been modified since loading, no query will be performed. Model validation is performed normally. Setting a value to `None` is equivalent to running a CQL `DELETE` on that column. It is possible to do a blind update, that is, to update a field without having first selected the object out of the database. See :ref:`Blind Updates ` Nz {0}.{1} has no column named: {2}z4Cannot apply update to primary key '{0}' for {1}.{2}z$cannot update polymorphic base model)rrr`rrrrrn)rrrrrrrris_primary_keyrrr8r9r!r:rrrrrVr^rprIrrlupdater)r'rrrr current_valuerrrrsF     zBaseModel.updatec Cs0|j|j||j|j|j|j|j|jddS)z6 Deletes the object from the database )rr`rrrrrnN) rrrr^rprrIrldeleter}rrrrs  zBaseModel.deletecCsdd|jDS)zb Returns a list of the columns that have been updated since instantiation or save cSsg|] \}}|jr|qSrrrrrrr/rz1BaseModel.get_changed_columns..)rrr}rrrget_changed_columns+szBaseModel.get_changed_columnscCs |j|Sr )rGr)rrrrr _class_batch1rzBaseModel._class_batchcCs,|jtjus Jd|jrtd||_|S)Nrz3Cannot specify a connection on model in batch mode.)rrrrvrr)r'rrrr _inst_batch5s zBaseModel._inst_batchcCrr )__connection__rrrr_class_get_connection>szBaseModel._class_get_connectioncCs |jp|jSr )rvrr}rrr_inst_get_connectionBrzBaseModel._inst_get_connection)Frh)Urrrr4 _DoesNotExistr _MultipleObjectsReturnedr r6rGrRrrorrrDrHr]r`rbrirkrnrurxrrrrr __options____compute_routing_key__r ModelQuerySetr7DMLQueryrrpr^rdrlrrvr(rr classmethodrrrrrrrrrrrrrrrrrrrrrrr r rr>rrrrrrrrr$rrr _get_connectionrrrrrLs     *             $5    rcseZdZfddZZS)ModelMetaClassc)sttd}t}|D]}t|diD] \}}|||qq dd} d<dd<dp:i} | dd<dd D} t| d d d } td d | D} dd |D| } dd | D} t| dk}t| dkrtd t| dr|std| r| dnd\}}t |t j t j frtdd}|r| sfdd|}t| }| stdd | Dstddd |D}dd |D}|r|rtdtdd| D}fdd }d}| D]T\}}|tjvrtd! ||js|jrt |t j rtd"|s&|jr&d#|_d#}|jr1||_|d7}|}|rE|j|_|j|_|j|_|||qtd$dDtd%dD}d&d#rd'd D}td(d|D}d)d |Dtfd*d }ni}td+d }tdkr| std,tdkrd-d Dd}|d.<nfd/d }fd0d }t||d.<t}D]@}|j|vrtd1 ||j|jr|jr|jrtd2|jr |jd3vr td4 t|j|j||jqi}D]\} }!|!j}"|"| kr*| ||"<qd5<d6<|d<|d7<|d8<id9<d:<|d;<|d<<|d=<t|dkd><| d?<|d@<|dA<|dB<|dC<| ruinddD<d}#|D]}t|dEd}#|#durnq}|#p dEtj!}#t"dE|#fidE<d}$|D]}t|dFd}$|$durnq|$p dFtj#}$t"dF|$fidF<t$t%|&|||}%g}&D] }'t '|'|&qt|&D] }(|((|%)q|%S)GNrrFrr#default_time_to_livercSs$g|]\}}t|tjr||fqSr)rrColumnrrrrras$z*ModelMetaClass.__new__..cSs |djS)Nr)position)xrrrrbs z(ModelMetaClass.__new__..)rcSsg|]}|djqSrdiscriminator_columnrcrrrrdcSrrrrr-rrrrfrcSsg|] }|djr|qSr.r/r1rrrrgrrrzBonly one discriminator_column can be defined in a model, {0} foundz]__discriminator_value__ specified, but no base columns defined with discriminator_column=True)NNzEcounter and container columns cannot be used as discriminator columnscs8|D]}t|ddr|S|j}|r|SqdS)Nr9F)r __bases__)basesbaser)_get_polymorphic_baserrr8ws  z5ModelMetaClass.__new__.._get_polymorphic_basecSsg|]\}}|jqSr) primary_keyrrrrrr3z#At least 1 primary key is required.cSsg|] }t|tjr|qSr)rrCounterr1rrrrrcSs"g|] }|jst|tjs|qSr)r9rrr:r1rrrrs"z(counter models may not have data columnscss|]\}}|jVqdSr r<rrrrrsz)ModelMetaClass.__new__..cs0||<|jr ||<||t||<dSr )r9set_column_namer)col_namecol_obj)r column_dict primary_keysrr_transform_columns  z1ModelMetaClass.__new__.._transform_columnz5column '{0}' conflicts with built-in attribute/methodz.counter columns cannot be used as primary keysTcss|] }|djr|VqdSrNr;rrrrrcss|] }|djs|VqdSrBr;rrrrrrCr$cSrrrr1rrrrrcss|] }|j|jfVqdSr )r|_partition_key_index)rrrrrrscSsg|]}|jqSr)cql_typer1rrrrscsfddt|DS)Ncsg|] \}}||qSr) to_binary)rtp proto_versionrrrrz...)zippartsrJ) key_cql_typesrIrrrcSsdSr rrLrrrrsz*at least one partition key must be definedcSrrrr4rrrrrpkcstfddDS)Nc3s|] }j|VqdSr )rrr1r}rrrrC;ModelMetaClass.__new__....)tuplerr}partition_keysr}rrrcs tfddt|DS)Nc3s$|] \}}j||VqdSr )rr)rr2rr}rrrs"rP)rQrKr)r'rrRr}rrs z+{0} defines the column '{1}' more than oncezBclustering_order may be specified only for clustering primary keys)ascdescz/invalid clustering order '{0}' for column '{1}'rrr_pk_namer_partition_keysrDr_clustering_keys _has_counterr9r8rr;r:rr r )*r rr setdefaultrsortedanyrFr rrrrr:rr__dict__r9r<rDr,r staticmethodrrpropertyrr|clustering_orderrrepraddpopr rr superr)__new__ resolve_udtsregister_for_keyspacer))rrAr6rpk_nameinherited_columnsr7rr is_abstractoptionscolumn_definitionsis_polymorphic_basediscriminator_columnsis_polymorphicdiscriminator_column_namer0polymorphic_basedefined_columnscounter_columns data_columnshas_partition_keysrApartition_key_index overridingclustering_keyskey_colskey_serializer_get_set col_namesdb_mapr=fielddb_fieldDoesNotExistBaseMultipleObjectsReturnedBaserudtsr user_typer)r8rr?rNrSr@rreJs                    zModelMetaClass.__new__)rrrre __classcell__rrrrr)Hsr)c@s:eZdZdZ dZ dZ dZ dZ dZ dZ dZ dS)ModelTNF) rrrrrrrrr#rr$rrrrrs"r) metaclass).loggingrwarningsrcassandra.cqlenginerrrrrcassandra.cqlengine.queryr r!r r"cassandra.metadatar cassandra.utilr getLoggerrlogrrr r!Warningr"robjectr$r6rDrRr]rbrkroruAbstractQueryableColumnryrrrr)rrrrrsH           2M