U 8T_b@sddlZddlmZddlmZddlmZmZddl m Z ddl m Z ddl mZddlmZmZdd lmZdd lmZdd lmZdd lmZd dlmZd ddZddZddZddZGdddZ GdddeZ!GdddeZ"GdddZ#dS)!N)contextmanager) AppConfig)Appsapps)settings)models)RECURSIVE_RELATIONSHIP_CONSTANT) DEFAULT_NAMESnormalize_together)make_model_tuple)cached_property) import_string)get_docs_version)InvalidBasesErrorcCsJt|tr6|dd}t|dkr*t|S||dfS|jj|jjfSdS)N.rr) isinstancestrsplitlentuple_meta app_label model_name)modelrrr>/tmp/pip-unpacked-wheel-dv63caxw/django/db/migrations/state.py_get_app_label_and_model_names   rcCsdd|D}t}|jjdddD]8}|jr(|jdk r(t|jts(||j | |jq(|j}|j r||kr| |j |S)zEReturn all models that have a direct relationship to the given model.cSsg|]}t|tjr|qSr) issubclassrModel).0subclassrrr s z'_get_related_models..T)Zinclude_parentsZinclude_hiddenN) __subclasses__setrZ get_fields is_relation related_modelrraddrappendproxyZconcrete_model)mrelated_modelsZrelated_fields_modelsfoptsrrr_get_related_modelss  r0cCsddt|DS)zq Return a list of typical (app_label, model_name) tuples for all related models for the given model. cSsh|]}|jj|jjfqSr)rrr)r"rel_modrrr 2sz,get_related_models_tuples..)r0rrrrget_related_models_tuples-sr4cCsjt}t|}|D]@}|jj|jj}}||f|kr6q|||f|t|q||jj|jjfhS)a Return all models that have a direct or indirect relationship to the given model. Relationships are either defined by explicit relational fields, like ForeignKey, ManyToManyField or OneToOneField, or by inheriting from another model (a superclass is related to its subclasses, but not vice versa). Note, however, that a model inheriting from a concrete model is also related to its superclass through the implicit *_ptr OneToOneField on the subclass. )r&r0rrrr)extend)rseenqueuer1 rel_app_labelrel_model_namerrrget_related_models_recursive8s  r:c@seZdZdZdddZddZddZd d d Zd!d d Zd"ddZ ddZ ddZ ddZ e ddZeddZeddZddZdS)# ProjectStatez Represent the entire project's overall state. This is the item that is passed around - do it here rather than at the app level so that cross-app FKs/etc. resolve properly. NcCs|pi|_|pg|_d|_dSNF)r real_apps is_delayed)selfrr=rrr__init__Us  zProjectState.__init__cCs6|j|j}}||j||f<d|jkr2|||dSNr)r name_lowerr__dict__ reload_model)r? model_staterrrrr add_model[s zProjectState.add_modelcCs2|j||f=d|jkr.|j|||jdSrA)rrCrunregister_model clear_cacher?rrrrr remove_modelas  zProjectState.remove_modelFc Cs|r d|_t}z|j||}Wntk r6YnX|rFt|}nt|}|j||f}t}|j D]:}|j rl|j j t krqlt|j|\} } || | fql|||D]R\} } z|j| | } Wntk rYqX|r|t| q|t| q|||f|S)NT)r>r&r get_model LookupErrorr4r:rfieldsvaluesr' remote_fieldrrrr(r)lowerupdate) r?rrdelayr-Z old_modelrEZdirect_related_modelsfieldr8r9Z rel_modelrrr_find_reload_modelis:    zProjectState._find_reload_modelcCs&d|jkr"||||}||dSrA)rCrT_reload)r?rrrRr-rrrrDs zProjectState.reload_modelTcCs@d|jkr||q>|D]:\}}z|j||f}Wntk rYqbX||qb|j |dSN) r bulk_updaterG real_modelsrrBr*rKeyErrorrender_multiple)r?r-r8r9Zstates_to_be_renderedrErrrrUs      zProjectState._reloadcCs@tdd|jD|jd}d|jkr4|j|_|j|_|S)z*Return an exact copy of this ProjectState.cSsi|]\}}||qSrclone)r"kvrrr sz&ProjectState.clone..)rr=r)r;ritemsr=rCrr]r>)r?Z new_staterrrr]s  zProjectState.clonecCs|jrd|jkr|jd=dSrA)r>rCr?rrrclear_delayed_apps_cachesz%ProjectState.clear_delayed_apps_cachecCst|j|jSrW) StateAppsr=rrbrrrrszProjectState.appscCst|j|jdd|_|jS)NT)ignore_swappable)rdr=rrrbrrr concrete_appsszProjectState.concrete_appscCs8i}|jddD]}t|}|||j|jf<q||S)z3Take an Apps and return a ProjectState matching it.T)Zinclude_swapped) get_models ModelState from_modelrrB)clsrZ app_modelsrrErrr from_appss  zProjectState.from_appscCs |j|jkot|jt|jkSrW)rr&r=r?otherrrr__eq__szProjectState.__eq__)NN)F)F)T)__name__ __module__ __qualname____doc__r@rFrJrTrDrVrUr]rcr rpropertyrf classmethodrkrnrrrrr;Ns"  0      r;cs,eZdZdZdZfddZddZZS) AppConfigStubzAStub of an AppConfig. Only provides a label and a dict of models.rcs||_t|ddSrW)labelsuperr@)r?rv __class__rrr@szAppConfigStub.__init__cCs|jj|j|_dSrW)r all_modelsrvrrbrrr import_modelsszAppConfigStub.import_models)rorprqrrpathr@r{ __classcell__rrrxrrus rucsNeZdZdZdfdd ZeddZddZd d Zd d Z d dZ Z S)rdzo Subclass of the global Apps registry class to better handle dynamic model additions and removals. Fc sg|_|D]2}t|}|D]}|jtj|ddq q dd|D}ddt||D}t |d|_ d|_ | ||jddlm} |rttjhnt} | || d } | rtd d d | DdS) NT) exclude_relscSsh|] }|jqSr)r)r"rErrrr2sz%StateApps.__init__..cSsg|] }t|qSr)ru)r"rvrrrr$sz&StateApps.__init__..r)_check_lazy_references)ignore css|] }|jVqdSrW)msg)r"errorrrr sz%StateApps.__init__..)rY global_appsZget_app_configrgr*rhrirNsortedrwr@_lockZ ready_eventr[Zdjango.core.checks.model_checksrr rZAUTH_USER_MODELr& ValueErrorjoin) r?r=rrerZapprZ app_labels app_configsrrerrorsrxrrr@s      zStateApps.__init__ccs,|j}d|_z dVW5||_|XdSr<)readyrH)r?rrrrrXs  zStateApps.bulk_updatec Cs|sdS|r|}|r~g}|D]2}z||Wq"tk rR||Yq"Xq"t|t|krxtd|tf|}qW5QRXdS)NzCannot resolve bases for %r This can happen if you are inheriting models from an app with migrations (e.g. contrib.auth) in an app with no migrations; see https://docs.djangoproject.com/en/%s/topics/migrations/#dependencies for more)rXrenderrr*rr)r?Z model_statesZunrendered_modelsZnew_unrendered_modelsrrrrr[$s$ zStateApps.render_multiplecCsHtgi}t|j|_t|j|_|jD] }||_q0|j|_|S)z Return a clone of this registry.)rdcopydeepcopyrzrrNrrY)r?r]Z app_configrrrr]>s zStateApps.clonecCsl||j||jj<||jkrBt||j|<||j|_i|j|_||j|j|jj<|||dSrW) rzrrrrurrZdo_pending_operationsrH)r?rrrrrregister_modelJs    zStateApps.register_modelcCs8z|j||=|j|j|=Wntk r2YnXdSrW)rzrrrZrIrrrrGTs  zStateApps.unregister_model)F) rorprqrrr@rrXr[r]rrGr}rrrxrrds   rdc@sleZdZdZdddZeddZeddd Zd d Z d d Z ddZ ddZ ddZ ddZddZdS)rha Represent a Django Model. Don't use the actual Model class as it's not designed to have its options changed - instead, mutate this one and then render it into a Model as required. Note that while you are allowed to mutate .fields, you are not allowed to mutate the Field instances inside there themselves - you must instead assign new ones, as these are not detached during a clone. Nc Cs||_||_t||_|pi|_|jdg|jdg|pFtjf|_|pPg|_ |j D]\\}}t |dr|t d||j rt |jdrt d||jr^t |jjdr^t d|q^|jdD]}|jst d|qdS) Nindexes constraintsrz7ModelState.fields cannot be bound to a model - "%s" is.rz_ModelState.fields cannot refer to a model class - "%s.to" does. Use a string reference instead.zdModelState.fields cannot refer to a model class - "%s.through" does. Use a string reference instead.zKIndexes passed to ModelState require a name attribute. %r doesn't have one.)rnamedictrMoptions setdefaultrr!basesmanagersrahasattrrr'r(Z many_to_manyrOZthrough) r?rrrMrrrrSindexrrrr@gs@    zModelState.__init__cCs |jSrW)rrPrbrrrrBszModelState.name_lowerFc sXg}jjD]z}t|ddr"|r"q t|tjr0q |j}z|||fWq t k r}zt d|jj |fW5d}~XYq Xq |sjj D]Z}|j}z|||fWqt k r}zt d|jj |fW5d}~XYqXqi}t D]}|dkr qq|jjkr|dkr>jjd}tt|||<q|dkrfjjd} tt| ||<q|dkrd d jjD} | D]} | js| q| |d<q|d krd d jjD|d <qjj|||<q|rd D]} | |kr|| =qn$|dddjjDkr&|d=fddttfddd} tdd| D}tdd|Ds|tjf}g}t}d}jjD]}|j|krqnb|jrt|}|nF|jks|jkrt }|j!|_!|j|_|jkr |}nq|"|j||j|fq|d|fgkr>g}|jj#jj ||||S)z3Given a model, return a ModelState representing it.rONz'Couldn't reconstruct field %s on %s: %sz+Couldn't reconstruct m2m field %s on %s: %s)rrunique_togetherindex_togetherrcSsg|] }|qSrr\)r"idxrrrr$sz)ModelState.from_model..rcSsg|] }|qSrr\)r"conrrrr$s)rrorder_with_respect_torcSsh|] }|jqSr)r)r"rSrrrr2sz(ModelState.from_model..cs@g}|jD]0}t|dr0|jjr0||q ||q |S)Nr) __bases__rrZabstractr5r*)rrbase) flatten_basesrrrs   z,ModelState.from_model..flatten_basescs j|SrW)__mro__r)xr3rrz'ModelState.from_model..keycss$|]}t|dr|jjn|VqdS)rN)rrZ label_lowerr"rrrrrs z(ModelState.from_model..css$|]}t|tpt|tjVqdSrW)rrr rr!rrrrrsZobjects)$rZ local_fieldsgetattrrrZOrderWrtrr*r] TypeErrorrvZlocal_many_to_manyZ object_namer Zoriginal_attrsr&r rZset_name_with_modelrgetZprivate_fieldsrranyr!rZuse_in_migrationsrZ_set_creation_counterZ _base_managerZ_default_managerManagerrr)r)rjrr~rMrSrerutitrrrZflattened_basesrrZ manager_namesZdefault_manager_shimmanagerZ new_managerr)rrrris                      zModelState.from_modelc csjt|jddd}|D]N\}}|\}}}}}|rLt|} || fVqt|} || ||fVqdS)z-Deep-clone the managers using deconstruction.cSs |djS)Nr)Zcreation_counter)r_rrrrrz/ModelState.construct_managers..rN)rr deconstructr as_manager) r?Zsorted_managersZmgr_namerrZ manager_pathZqs_pathargskwargsZqs_classZ manager_classrrrconstruct_managerss zModelState.construct_managerscCs.|j|j|jt|jt|j|jt|jdS)z(Return an exact copy of this ModelState.)rrrMrrr) ryrrrrMrrlistrrbrrrr]szModelState.clonecs|jd|j}tdd|}ztfdd|jD}Wn$tk r^td|jfYnXdd|jD}||d<d |d <| | t|j ||S) zACreate a Model object from our current state into the given apps.)rrZMetarc3s&|]}t|tr|n|VqdSrW)rrrKrrrrr0sz$ModelState.render..z(Cannot resolve one or more bases from %rcSsi|]\}}||qSrr\)r"rrSrrrr`7sz%ModelState.render..Z__fake__rp) rrtyperrrLrrMrarQrr)r?rZ meta_contentsmetarbodyrrrr)s   zModelState.rendercCs8|jdD]}|j|kr |Sq td||jfdS)NrzNo index named %s on model %srrr)r?rrrrrget_index_by_name@s  zModelState.get_index_by_namecCs8|jdD]}|j|kr |Sq td||jfdS)Nrz"No constraint named %s on model %sr)r?r constraintrrrget_constraint_by_nameFs  z!ModelState.get_constraint_by_namecCsd|jj|j|jfS)Nz <%s: '%s.%s'>)ryrorrrbrrr__repr__LszModelState.__repr__cCs||j|jkoz|j|jkozt|jt|jkoztddtt|jt|jDoz|j|jkoz|j |j koz|j |j kS)NcssB|]:\\}}\}}||ko8|dd|ddkVqdS)rN)r)r"Zk1f1Zk2f2rrrrTsz$ModelState.__eq__..) rrrrMallziprrarrrrlrrrrnOs$       zModelState.__eq__)NNN)F)rorprqrrr@r rBrtrirr]rrrrrnrrrrrh\s "   rh)r)$r contextlibrZ django.appsrZdjango.apps.registryrrrZ django.confrZ django.dbrZdjango.db.models.fields.relatedrZdjango.db.models.optionsr r Zdjango.db.models.utilsr Zdjango.utils.functionalr Zdjango.utils.module_loadingr Zdjango.utils.versionr exceptionsrrr0r4r:r;rurdrhrrrrs*            h