o ǯh'@sddlmZddlmZddlmZddlmZmZm Z ddl m Z ddl m Z ddlmZdd lmZmZdd lmZmZdd lmZmZmZmZGd d d ejZGdddejZdS))settings)get_current_site) call_command)models connections transaction)reverse) CloneSchema)_check_schema_name)post_schema_syncschema_needs_to_be_sync)get_creation_fakes_migrationsget_tenant_base_schema) schema_existsget_tenant_domain_modelget_public_schema_nameget_tenant_database_aliascseZdZdZdZ dZ dZ ejddde gdZ dZ dZ gZ Gdd d Zd d Zd d ZddZddZeddZd'fdd ZddZd(ddZddZd(fdd Z  d)dd Zd!d"Zd#d$Zd%d&ZZS)* TenantMixinz4 All tenant models must inherit this class. FTDATA?) max_lengthuniquedb_index validatorsNc@eZdZdZdS)zTenantMixin.MetaTN__name__ __module__ __qualname__abstractr!r!R/var/www/Befach/backend/venv/lib/python3.10/site-packages/django_tenants/models.pyMeta5r#cC|jSN) schema_nameselfr!r!r"__str__8zTenantMixin.__str__cCs$tt}|j|j||S)a Syntax sugar which helps in celery tasks, cron jobs, and other scripts Usage: with Tenant.objects.get(schema_name='test') as tenant: # run some code in tenant test # run some code in previous tenant (public probably) )rr_previous_tenantappendtenantactivater) connectionr!r!r" __enter__;s zTenantMixin.__enter__cCstt}||jdSr&)rr set_tenantr,pop)r)exc_typeexc_valexc_tbr1r!r!r"__exit__Js zTenantMixin.__exit__cCstt}||dS)z Syntax sugar that helps at django shell with fast tenant changing Usage: Tenant.objects.get(schema_name='test').activate() N)rrr3r0r!r!r"r/Os zTenantMixin.activatecCstt}|dS)z Syntax sugar, return to public schema Usage: test_tenant.deactivate() # or simpler Tenant.deactivate() N)rrset_schema_to_public)clsr1r!r!r" deactivateYs zTenantMixin.deactivater csDtt}|jj}t|d}|r|r|jtkrtd|j|r3|s3|j|jtfvr3td|jtj |i||re|re|j rez|j d|dt j t|dWdStyd|jddw|rrtj t|ddS|s|j rt|jsz|j d|dt j t|dWdSty|wdSdSdS)Nr'zDCan't create tenant outside the public schema. Current schema is %s.zWCan't update tenant outside it's own schema or the public schema. Current schema is %s.T)check_if_exists verbosity)senderr.) force_drop)rr_stateaddinghasattrr'r Exceptionsupersaveauto_create_schema create_schemar sendrserializable_fieldsdeleter r _drop_schema)r)r=argskwargsr1is_new has_schema __class__r!r"rEfs>      zTenantMixin.savecCs|S)zX in certain cases the user model isn't serializable so you may want to only send the id r!r(r!r!r"rIszTenantMixin.serializable_fieldscCstt}t|d}|r|j|jtfvrtd|j|r:t|jr<|js(|r>|| }| d|jdSdSdSdS)z Drops the schemar'zWCan't delete tenant outside it's own schema or the public schema. Current schema is %s.zDROP SCHEMA "%s" CASCADEN) rrrBr'rrCrauto_drop_schemapre_dropcursorexecute)r)r?r1rOrTr!r!r"rKs   zTenantMixin._drop_schemacCsdS)zz This is a routine which you could override to backup the tenant schema before dropping. :return: Nr!r(r!r!r"rSszTenantMixin.pre_dropcs||tj|i|S)zt Deletes this row. Drops the tenant's schema if the attribute auto_drop_schema set to True. )rKrDrJ)r)r?rLrMrPr!r"rJs zTenantMixin.deletec Cstt}t|j|}|rt|jrdSt}|rK|r9t}t}| ||j|j t ddd|jd|dn| d|jt dd|jd|d| dS)z Creates the schema 'schema_name' for this tenant. Optionally checks if the schema already exists before creating it. Returns true if the schema was created, false otherwise. Fmigrate_schemasT)r.faker' interactiver=zCREATE SCHEMA "%s")r.r'rXr=N)rrr r'rTrrrr clone_schema clone_moderrUr9) r)r< sync_schemar=r1rTfake_migrations base_schemarYr!r!r"rGs8   zTenantMixin.create_schemacCs.z |jjdd}|WStjyYdSw)z: Returns the primary domain of the tenant T is_primaryN)domainsgetr DoesNotExist)r)domainr!r!r"get_primary_domains zTenantMixin.get_primary_domaincCs8|rdnd}t|j}d||jd|t|f}|S)z1 Returns the URL of this tenant. zhttps://zhttp://.) is_securerrcjoinr'r)r)request view_name http_typercurlr!r!r"rs zTenantMixin.reversecCs t|tjS)zd Get the type of tenant. Will only work for multi type tenants :return: str )getattrrMULTI_TYPE_DATABASE_FIELDr(r!r!r"get_tenant_types zTenantMixin.get_tenant_type)r )F)FTr )rrr__doc__rRrFrZr CharFieldr r' domain_urldomain_subfolderr,r#r*r2r8r/ classmethodr;rErIrKrSrJrGrdrro __classcell__r!r!rPr"rsD   %  ,  rcspeZdZdZejddddZejej ddej dZ ej dddZ ejfdd ZGd d d Zd d ZZS) DomainMixinzC All models that store the domains must inherit this class T)rrrr`)r related_name on_delete)defaultrcsX|jjj|jddj|jd}|jp| |_|jr!|jddt j |i|dS)NT)r.r_)pkFr^) rQobjectsfilterr.excluder{r_existsupdaterDrE)r)rLrM domain_listrPr!r"rEs  zDomainMixin.savec@r)zDomainMixin.MetaTNrr!r!r!r"r# r$r#cCr%r&)rcr(r!r!r"r*r+zDomainMixin.__str__)rrrrprrqrc ForeignKeyr TENANT_MODELCASCADEr. BooleanFieldr_ratomicrEr#r*rur!r!rPr"rvs  rvN) django.confrdjango.contrib.sites.shortcutsrdjango.core.managementr django.dbrrr django.urlsrdjango_tenants.cloner postgresql_backend.baser signalsr r utilsrrrrrrModelrrvr!r!r!r"s     h