3 g)@s^ddlZddlZddlmZmZddlmZmZmZm Z ddl m Z GdddedZ Gdd d e Z Gd d d e edZGd d d eZGddde edZGdddeZGdddeedZGdddeZGdddeZGdddeeZGdddeeZGdddeeZGdddeedZGd d!d!eZGd"d#d#eZGd$d%d%ZdS)&N)abstractmethodABCMeta)DictListOptionalTuple) import_stringc@s(eZdZdZeddZeddZdS)BaseCRa Abstract base collision resolver. All collision resolvers needs to inherit from this class. To write custom collision resolver you need to overwrite resolve_collisions function. It receives Dict[str, List[str]], where key is model name and values are full model names (full model name means: module + model_name). You should return Dict[str, str], where key is model name and value is full model name. cCst|}|jjj|jfS)N)r_metaZ app_configname__name__)clsZfull_model_path model_classrc/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/collision_resolvers.pyget_app_name_and_modelszBaseCR.get_app_name_and_modelcCsdS)Nr)self namespacerrrresolve_collisionsszBaseCR.resolve_collisionsN)r __module__ __qualname____doc__ classmethodrrrrrrrr s r ) metaclassc@seZdZdZddZdS)LegacyCRz\ Default collision resolver. Model from last application in alphabetical order is selected. cCs*i}x |jD]\}}|d||<qW|S)N)items)rrresultr modelsrrrr%szLegacyCR.resolve_collisionsN)r rrrrrrrrr"src@s eZdZdZddZddZdS) AppsOrderCRNcCsV|jdk stdi}x:|jD].\}}t|dkr |j|}|dd||<q W|S)Nz6You must define APP_PRIORITIES in your resolver class!rr)APP_PRIORITIESAssertionErrorrlen$_sort_models_depending_on_priorities)rrrr rZ sorted_modelsrrrr/s  zAppsOrderCR.resolve_collisionsc Csfg}xX|D]P}y|j|\}}|jj|}Wnttfk rJtj}YnX|j||fq Wt|S)N) rr!index ImportError ValueErrorsysmaxsizeappendsorted)rrZmodels_with_prioritiesmodelapp_name_positionrrrr$8s  z0AppsOrderCR._sort_models_depending_on_priorities)r rrr!rr$rrrrr ,s r c@seZdZdZeddZdS)InstalledAppsOrderCRah Collision resolver which selects first model from INSTALLED_APPS. You can set your own app priorities list by subclassing him and overwriting APP_PRIORITIES field. This collision resolver will select model from first app on this list. If both app's are absent on this list, resolver will choose model from first app in alphabetical order. cCsddlm}t|dgS)Nr)settingsZINSTALLED_APPS) django.confr1getattr)rr1rrrr!Ls z#InstalledAppsOrderCR.APP_PRIORITIESN)r rrrpropertyr!rrrrr0Dsr0cs,eZdZdZeddZfddZZS) PathBasedCRz Abstract resolver which transforms full model name into alias. To use him you need to overwrite transform_import function which should have one parameter. It will be full model name. It should return valid alias as str instance. cCsdS)Nr)r module_pathrrrtransform_importZszPathBasedCR.transform_importcsjtt|j|}xT|jD]H\}}t|dkr0qx0|D](}|j|}t|tsVtd|||<q6WqW|S)Nrz'result of transform_import must be str!) superr5rrr#r7 isinstancestrr")rrZ base_importsr rr,new_name) __class__rrr^s   zPathBasedCR.resolve_collisions)r rrrrr7r __classcell__rr)r<rr5Rs r5c@seZdZdZddZdS) FullPathCRa Collision resolver which transform full model name to alias by changing dots to underscores. He also removes 'models' part of alias, because all models are in models.py files. Model from last application in alphabetical order is selected. cCs$|jdd\}}||}|jddS)Nz.modelsr.r.)rsplitreplace)rr6moduler,rrrr7qszFullPathCR.transform_importN)r rrrr7rrrrr>jsr>c@seZdZdZdZddZdS) AppNameCRah Abstract collision resolver which transform pair (app name, model_name) to alias by changing dots to underscores. You must define MODIFICATION_STRING which should be string to format with two keyword arguments: app_name and model_name. For example: "{app_name}_{model_name}". Model from last application in alphabetical order is selected. NcCs<|jdk std|j|\}}|jdd}|jj||dS)Nz;You must define MODIFICATION_STRING in your resolver class!r?r.)r- model_name)MODIFICATION_STRINGr"rrAformat)rr6r-rDrrrr7s zAppNameCR.transform_import)r rrrrEr7rrrrrCwsrCc@seZdZdZdZdS)AppNamePrefixCRa  Collision resolver which transform pair (app name, model_name) to alias "{app_name}_{model_name}". Model from last application in alphabetical order is selected. Result is different than FullPathCR, when model has app_label other than current app. z{app_name}_{model_name}N)r rrrrErrrrrGsrGc@seZdZdZdZdS)AppNameSuffixCRz Collision resolver which transform pair (app name, model_name) to alias "{model_name}_{app_name}" Model from last application in alphabetical order is selected. z{model_name}_{app_name}N)r rrrrErrrrrHsrHc@seZdZdZdS)AppNamePrefixCustomOrderCRz Collision resolver which is mixin of AppNamePrefixCR and InstalledAppsOrderCR. In case of collisions he sets aliases like AppNamePrefixCR, but sets default model using InstalledAppsOrderCR. N)r rrrrrrrrIsrIc@seZdZdZdS)AppNameSuffixCustomOrderCRz Collision resolver which is mixin of AppNameSuffixCR and InstalledAppsOrderCR. In case of collisions he sets aliases like AppNameSuffixCR, but sets default model using InstalledAppsOrderCR. N)r rrrrrrrrJsrJc@seZdZdZdS)FullPathCustomOrderCRz Collision resolver which is mixin of FullPathCR and InstalledAppsOrderCR. In case of collisions he sets aliases like FullPathCR, but sets default model using InstalledAppsOrderCR. N)r rrrrrrrrKsrKc@seZdZdZdZddZdS) AppLabelCRa Abstract collision resolver which transform pair (app_label, model_name) to alias. You must define MODIFICATION_STRING which should be string to format with two keyword arguments: app_label and model_name. For example: "{app_label}_{model_name}". This is different from AppNameCR when the app is nested with several level of namespace: Gives sites_Site instead of django_contrib_sites_Site Model from last application in alphabetical order is selected. NcCs:|jdk stdt|}|jj|j}}|jj||dS)Nz;You must define MODIFICATION_STRING in your resolver class!) app_labelrD)rEr"rr rMr rF)rr6rrMrDrrrr7szAppLabelCR.transform_import)r rrrrEr7rrrrrLsrLc@seZdZdZdZdS)AppLabelPrefixCRz Collision resolver which transform pair (app_label, model_name) to alias "{app_label}_{model_name}". Model from last application in alphabetical order is selected. z{app_label}_{model_name}N)r rrrrErrrrrNsrNc@seZdZdZdZdS)AppLabelSuffixCRz Collision resolver which transform pair (app_label, model_name) to alias "{model_name}_{app_label}". Model from last application in alphabetical order is selected. z{model_name}_{app_label}N)r rrrrErrrrrOsrOc@sLeZdZddZddZeddZeddZed d Zed d Z d S)CollisionResolvingRunnercCsdS)Nr)rrrr__init__sz!CollisionResolvingRunner.__init__cCs|j|}|j|S)N)_get_dictionary_of_names_get_dictionary_of_modules)rmodels_to_importdictionary_of_namesrrrrun_collision_resolvers z/CollisionResolvingRunner.run_collision_resolvercCs@ddlm}tt|dd}|j||j|}|j||S)Nr)r1Z!SHELL_PLUS_MODEL_IMPORTS_RESOLVERz.django_extensions.collision_resolvers.LegacyCR)r2r1rr3+_assert_is_collision_resolver_class_correctr,_assert_is_collision_resolver_result_correct)r rTr1collision_resolver_classrrrrrRs    z1CollisionResolvingRunner._get_dictionary_of_namescCsXt|tstdx@|jD]4\}}t|ts:td|t|tstd|qWdS)Nz5Result of resolve_collisions function must be a dict!z5key in collision resolver result should be str not %sz7value in collision resolver result should be str not %s)r9dictr"rr:)r rkeyvaluerrrrXszECollisionResolvingRunner._assert_is_collision_resolver_result_correctcCs>tj|rt|tstdttj|jjdks:tddS)Nz=SHELL_PLUS_MODEL_IMPORTS_RESOLVER must be subclass of BaseCR!z3resolve_collisions function must take one argument!) inspectisclass issubclassr r"r#getfullargspecrargs)r rYrrrrWs zDCollisionResolvingRunner._assert_is_collision_resolver_class_correctcCsLi}xB|jD]6\}}|jdd\}}|j|g||j||fqW|S)Nr?r)rr@ setdefaultr*)r rUZdictionary_of_modulesaliasr,r6rDrrrrSs  z3CollisionResolvingRunner._get_dictionary_of_modulesN) r rrrQrVrrRrXrWrSrrrrrPs    rP)r^r(abcrrtypingrrrrZdjango.utils.module_loadingrr rr r0r5r>rCrGrHrIrJrKrLrNrOrPrrrrs(