o ǯh+@sddlZddlmZddlmZmZddlmZddlm Z ddl m Z m Z ddl mZmZmZddlmZz dd lmZejZWneyPdd lmZYnwdd lmZd d ZddZddZddZddZddZddZ ddZ!ddZ"ddZ#d d!Z$d"d#Z%d$d%Z&d&d'Z'd(d)Z(Gd*d+d+eZ)Gd,d-d-eZ*d.d/Z+d0d1Z,d2d3Z-d4d5Z.efd6d7Z/ed8fd9d:Z0ed;d<d=d>Z1d?d@Z2dAdBZ3dCdDZ4d8dEdFdGdHZ5dIdJZ6dS)KN)ContextDecorator) lru_cachewraps) ModuleType)settings)ImproperlyConfiguredValidationError)DEFAULT_DB_ALIAS connection connections) import_stringapps) get_model)mailcC ttjSN)rr TENANT_MODELrrQ/var/www/Befach/backend/venv/lib/python3.10/site-packages/django_tenants/utils.pyget_tenant_model rcCrr)rrTENANT_DOMAIN_MODELrrrrget_tenant_domain_modelrrcCs ttdtS)NTENANT_DB_ALIAS)getattrrr rrrrget_tenant_database_alias rcC ttddS)NPUBLIC_SCHEMA_NAMEpublicrrrrrrget_public_schema_name rr"cCs ttdiS)N TENANT_TYPESr!rrrrget_tenant_types$rr$cCsttdd}t|S)NTENANT_BASE_MIGRATE_COMMANDz/django.core.management.commands.migrate.Command)rrr ) class_pathrrr%get_tenant_base_migrate_command_class(s r'cCr)NHAS_MULTI_TYPE_TENANTSFr!rrrrhas_multi_type_tenants1rr)cCr)NMULTI_TYPE_DATABASE_FIELDr!rrrr"get_multi_type_database_field_name5rr,cCs"tr ttdSttddS)NURLCONFPUBLIC_SCHEMA_URLCONF urls_public)r)r$r"rrrrrrget_public_schema_urlconf9s r0cCs&tsJdt}dd|DS)z6This is to allow a choice field for the type of tenantFzBget_tenant_type_choices should only be used for multi type tenantscSsg|]}||fqSrr).0krrr Gz+get_tenant_type_choices..)r)r$keys) tenant_typesrrrget_tenant_type_choices@sr7cCr)NTENANT_LIMIT_SET_CALLSFr!rrrrget_limit_set_callsJrr9cCsttddpd}|dS)NTENANT_SUBFOLDER_PREFIXr+z/ )rrstrip)subfolder_prefixrrrget_subfolder_prefixNs r=cC(ttdd}|rttddstd|S) If TENANT_CREATION_FAKES_MIGRATIONS, tenants will be created by cloning an existing schema specified by TENANT_CLONE_BASE. TENANT_CREATION_FAKES_MIGRATIONSFTENANT_BASE_SCHEMAzdYou must specify a schema name in TENANT_BASE_SCHEMA if TENANT_CREATION_FAKES_MIGRATIONS is enabled.rrr)fakedrrrget_creation_fakes_migrationsS  rDcCr>)r?rAFr@z\TENANT_CREATION_FAKES_MIGRATIONS setting must be True to use TENANT_BASE_SCHEMA for cloning.rB)schemarrrget_tenant_base_schemabrErGcCs ttddS)NTENANT_MIGRATION_ORDERr!rrrrget_tenant_migration_orderqrrIc,eZdZfddZddZddZZS)schema_contextc(|d|_|dt|_tdSNrdatabase) schema_namegetrrNsuper__init__selfargskwargs __class__rrrRw zschema_context.__init__cCs&t|j|_tj|_|j|jdSr)r rNr tenantprevious_tenant set_schemarOrTrrr __enter__| zschema_context.__enter__cG*|jdur |jdS|j|jdSrr[r set_schema_to_public set_tenantrTexcrrr__exit__ zschema_context.__exit____name__ __module__ __qualname__rRr^rf __classcell__rrrWrrKu rKcrJ)tenant_contextcrLrM)rZrPrrNrQrRrSrWrrrRrYztenant_context.__init__cCs&t|j|_tj|_|j|jdSr)r rNr rZr[rcr]rrrr^r_ztenant_context.__enter__cGr`rrardrrrrfrgztenant_context.__exit__rhrrrWrrnrmrncCs2ttdrtjr|tjr|ttjd}|S)z; Removes the TENANT_TOKEN from a particular string r.N)hasattrrr. startswithlen) url_stringrrrclean_tenant_urls  rscCst|S)zL Legacy function - just in case someone is still using the old name ) remove_wwwhostnamerrrremove_www_and_devsrwcCs|dr |ddS|S)z Removes www. from the beginning of the address. Only for routing purposes. www.test.com/login/ and test.com/login/ should find the same tenant. zwww.N)rprurrrrts  rtcCs ttdS)z I know this is very ugly! I'm looking for more elegant solutions. See: http://stackoverflow.com/questions/6957016/detect-django-testing-mode outbox)rorrrrrdjango_is_in_test_modes rzcCsHt|}|}d}|||f|}|r|d}nd}||S)NzUSELECT EXISTS(SELECT 1 FROM pg_catalog.pg_namespace WHERE LOWER(nspname) = LOWER(%s))rF)r cursorexecutefetchoneclose)rOrN _connectionr{sqlrowexistsrrr schema_existss rTcCsddlm}t|}|}t|rtd||stddtj |j tj |}| || ||_ |rA| dSdS)zR This renames a schema to a new name. It checks to see if it exists first r)is_valid_schema_namezNew schema name already existsz(Invalid string used for the schema name.zALTER SCHEMA {0} RENAME TO {1}N)&django_tenants.postgresql_backend.baserr r{rrformatr ops quote_namerOr|r~save)rZnew_schema_namerNrrrr{rrrr schema_renames    r)maxsizecsddlm}|dd}zt|}Wn ty|YSwt|tr&|n|jfdd|D}t |dkr<|S|dS)Nrr .csg|] }|jkr|jqSr)namelabel)r1 app_configapp_namerrr3s  z!get_app_label..) django.appsrsplitr ImportError isinstancerrget_app_configsrq)appr candidate imported_app app_labelrrr get_app_labels     rcCsdd|DS)z= Returns a list of app labels of the given apps_list cSsg|]}t|qSr)r)r1rrrrr3r4zapp_labels..r) apps_listrrr app_labelssrc Cs4z|tjWSttfytj|tjYSw)a Convenience function for parsing django-tenants' path configuration strings. If the string contains '%s', then the current tenant's schema name will be inserted at that location. Otherwise the schema name will be appended to the end of the string. :param config_path: A configuration path string that optionally contains '%s' to indicate where the tenant schema name should be inserted. :return: The formatted string containing the schema name )r rO TypeError ValueErrorospathjoin) config_pathrrrparse_tenant_config_path s  rcCsttdd}ttdg}|s_|rat|vrtdtt}t.}|d|jj g| rJt | |j jddd}|rJtd t|Wdn1sTwYtdSdSdS) NSKIP_PG_EXTRA_VALIDATIONFPG_EXTRA_SEARCH_PATHSz0%s can not be included on PG_EXTRA_SEARCH_PATHS.z>SELECT 1 FROM information_schema.tables WHERE table_name = %s;rOT)flatz._tenant_migration..wrapper)r)rrrr)rr_tenant_migrationFs z+tenant_migration.._tenant_migrationrr)rqcallable)rrrUrrrrtenant_migration<s  rcCst|dr|jSdS)z,This gets the tenant object from the requestrZN)rorZ)requestrrr get_tenantZs r)7r contextlibr functoolsrrtypesr django.confrdjango.core.exceptionsrr django.dbr r r django.utils.module_loadingr rrrrdjango.db.models.loading django.corerrrrr"r$r'r)r,r0r7r9r=rDrGrIrKrnrsrwrtrzrrrrrrrrrrrrsZ