W h&NSSKrSSKrSSKJr SSKJr SSKJr SSKJr SSK J r SSK J r J r SSKJr S r"S S \R"5r\"5rS rS rSr"SS5r"SS5r\RS5r"SS5r"SS5r\S5r\S5rg)N)contextmanager) ReadConcern) WriteConcern)_no_dereference_for_fields) _import_class)DEFAULT_CONNECTION_NAMEget_db)count_documents) switch_dbswitch_collectionno_dereferenceno_sub_classes query_counterset_write_concernset_read_write_concern!no_dereferencing_active_for_classc\rSrSrSrSrg)MyThreadLocalsc0UlgNno_dereferencing_classselfs dC:\Users\julio\Documents\inmuebles-mzt\backend\env\Lib\site-packages\mongoengine/context_managers.py__init__MyThreadLocals.__init__s &(#rN)__name__ __module__ __qualname____firstlineno__r__static_attributes__rrrrs)rrc(U[R;$r) thread_localsrclss rrr"s -66 66rcz[RRUS5 [RU==S- ss'g)Nr)r'r setdefaultr(s r$_register_no_dereferencing_for_classr-&s.((33C;((-2-rc[RU==S-ss'[RUS:Xa [RRU5 gg)Nr+r)r'rpopr(s r&_unregister_no_dereferencing_for_classr0+sD((-2-++C0A5,,0056rc*\rSrSrSrSrSrSrSrg)r 1aswitch_db alias context manager. Example :: # Register connections register_connection('default', 'mongoenginetest') register_connection('testdb-1', 'mongoenginetest2') class Group(Document): name = StringField() Group(name='test').save() # Saves in the default db with switch_db(Group, 'testdb-1') as Group: Group(name='hello testdb!').save() # Saves in testdb-1 cXlUR5UlX lURR S[ 5Ulg)zConstruct the switch_db context manager :param cls: the class to change the registered db :param db_alias: the name of the specific database to use db_aliasN)r)_get_collection collectionr4_metagetr ori_db_alias)rr)r4s rrswitch_db.__init__Cs6 --/ IIMM*6MNrcURURRS'SURlUR$)z4Change the db_alias and clear the cached collection.r4N)r4r)r7 _collectionrs r __enter__switch_db.__enter__Ns-%)]]z"#xxrcURURRS'URURlg)z"Reset the db_alias and collection.r4N)r9r)r7r6r<rtvalue tracebacks r__exit__switch_db.__exit__Ts*%)%6%6z"#r)r)r6r4r9N r r!r"r#__doc__rr=rDr$r%rrr r 1s" O /rr c*\rSrSrSrSrSrSrSrg)r Za(switch_collection alias context manager. Example :: class Group(Document): name = StringField() Group(name='test').save() # Saves in the default db with switch_collection(Group, 'group1') as Group: Group(name='hello testdb!').save() # Saves in group1 collection chXlUR5UlURUlX lg)zConstruct the switch_collection context manager. :param cls: the class to change the registered db :param collection_name: the name of the collection to use N)r)r5ori_collection_get_collection_nameori_get_collection_namecollection_name)rr)rNs rrswitch_collection.__init__hs. !113'*'?'?$.rc~^[U4Sj5nUTRlSTRlTR$)z@Change the _get_collection_name and clear the cached collection.c>TR$r)rN)r)rs rrL9switch_collection.__enter__.._get_collection_namevs'' 'rN) classmethodr)rLr<)rrLs` rr=switch_collection.__enter__ss:  (  ()=%#xxrcpURURlURURlg)zReset the collection.N)rKr)r<rMrLr@s rrDswitch_collection.__exit__~s&#22(,(D(D%r)r)rNrKrMNrFr%rrr r Zs  / Err c #|# Un[S5n[S5n[S5nURR5VVs/sHupE[XQX#45(dMUPM nnn[ U5 [ U6 Sv SSS5 [ U5 gs snnf!,(df  N =f![ U5 f=f7f)zno_dereference context manager. Turns off all dereferencing in Documents for the duration of the context manager:: with no_dereference(Group): Group.objects() ReferenceFieldGenericReferenceFieldComplexBaseFieldN)r_fieldsitems isinstancer-rr0)r)rXrYrZnamefield deref_fieldss rr r s4&'78 -.E F();< #{{002 2 (=P 2   -S1 ' 6J7 /s3 7 6 /s3sMB<AB,B B&B,<BB, B<B, B)%B,, B99B<c*\rSrSrSrSrSrSrSrg)rzno_sub_classes context manager. Only returns instances of this class and no sub (inherited) classes:: with no_sub_classes(Group) as Group: Group.objects.find() cXlSUlg)zdConstruct the no_sub_classes context manager. :param cls: the class to turn querying subclasses on Nr)cls_initial_subclasses)rr)s rrno_sub_classes.__init__s &*#rcURRUlURR4URlUR$)z8Change the objects default and _auto_dereference values.)r) _subclassesre _class_namers rr=no_sub_classes.__enter__s6&*hh&:&:# $ 4 46xxrc:URURlg)z/Reset the default and _auto_dereference values.N)rer)rhr@s rrDno_sub_classes.__exit__s#::rrdNrFr%rrrrs+ ;rrcr\rSrSrSr\4SjrSrSrSr Sr Sr S r S r S rS rS rSrSrSrSrg)raHQuery_counter context manager to get the number of queries. This works by updating the `profiling_level` of the database so that all queries get logged, resetting the db.system.profile collection at the beginning of the context and counting the new entries. This was designed for debugging purpose. In fact it is a global counter so queries issued by other threads/processes can interfere with it Usage: .. code-block:: python class User(Document): name = StringField() with query_counter() as q: user = User(name='Bob') assert q == 0 # no query fired yet user.save() assert q == 1 # 1 query was fired, an 'insert' user_bis = User.objects().first() assert q == 2 # a 2nd query was fired, a 'find_one' Be aware that: - Iterating over large amount of documents (>101) makes pymongo issue `getmore` queries to fetch the next batch of documents (https://www.mongodb.com/docs/manual/tutorial/iterate-a-cursor/#cursor-batches) - Some queries are ignored by default by the counter (killcursors, db.system.indexes) c[US9UlSUlSUlSSURR-0SS0SS0S.Ulg) N)aliasrz$nez%s.system.indexes killcursorsz$existsF)nsopzcommand.killCursors)r dbinitial_profiling_level_ctx_query_counterr^_ignored_query)rrps rrquery_counter.__init__sPu%'+$"#- <=-($-u#5 rcURRSS05nUSUlURRRR 5 URRSS05 g)Nprofilerwas)rtcommandrusystemrzdrop)rprofile_update_ress r_turn_on_profiling query_counter._turn_on_profilingsT!WW__i^<'9%'@$ ##% A'rcRURRSUR05 g)Nrz)rtr}rurs r_resets_profilingquery_counter._resets_profilings D$@$@ABrc&UR5 U$r)rrs rr=query_counter.__enter__s ! rc$UR5 gr)rr@s rrDquery_counter.__exit__s  rc*UR5nX:H$r _get_count)rrBcounters r__eq__query_counter.__eq__s//#rc.URU5(+$r)rrrBs r__ne__query_counter.__ne__s;;u%%%rc(UR5U:$rrrs r__lt__query_counter.__lt__ 5((rc(UR5U:*$rrrs r__le__query_counter.__le__ E))rc(UR5U:$rrrs r__gt__query_counter.__gt__rrc(UR5U:$rrrs r__ge__query_counter.__ge__ rrc"UR5$rrrs r__int__query_counter.__int__ s  rc(SUR5-$)z,repr query_counter as the number of queries.z%srrs r__repr__query_counter.__repr__sdoo'''rc[URRRUR5UR - nU=R S- slU$)zGet the number of queries by counting the current number of entries in db.system.profile and substracting the queries issued by this context. In fact everytime this is called, 1 query is issued so we need to balance that r+)r rtr~rzrwrv)rcounts rrquery_counter._get_countsR DGGNN22D4G4G H%% &     r)rvrwrtruN)r r!r"r#rGrrrrr=rDrrrrrrrrrr$r%rrrrsS85  (C! &)*)*!( rrc## [URRR55nUR U5 UR [ S0UD6S9v g7f)N) write_concernr%)dictrdocumentr\update with_optionsr)r6write_concernscombined_concernss rrr#sPZ55>>DDFG^,  ! ! 0Q?P0Q ! RRsAAc#X# [URRR55nUbUR U5 [UR RR55nUbUR U5 UR [S0UD6[S0UD6S9v g7f)N)r read_concernr%) rrrr\rrrrr)r6r read_concernscombined_write_concernscombined_read_concernss rrr*s":#;#;#D#D#J#J#LM!&&~6!*"9"9"B"B"H"H"JK %%m4  ! !"=%<= :#9: " sB(B*) contextlib threadingrpymongo.read_concernrpymongo.write_concernrmongoengine.base.fieldsrmongoengine.commonrmongoengine.connectionrr mongoengine.pymongo_supportr __all__localrr'rr-r0r r r rrrrr%rrrs%,.>,B7 )Y__)  73 6 &/&/R'E'ET 44@;;8__DSS r