o -ho @sddZddlZddlmZddlmZddlmZeZ ddZ d d Z dd d Z Gd ddZ dS)zi Tenant-aware database router for schema separation. Routes database queries to tenant-specific schemas. N)settings) connection)TenantcCs|r+|t_d|j}t}|d|dWddS1s$wYdSdt_t}|dWddS1sCwYdS)z'Set the current tenant for this thread.tenant_zSET search_path TO "z ", publicNzSET search_path TO public)_localtenant tenant_codercursorexecute)r schema_namer r #/var/www/html/pos/tenants/router.py set_tenant s  "  "rcCs ttddS)z'Get the current tenant for this thread.rN)getattrrr r r rget_current_tenants rcCs"|s t}|s dS|j}d|S)z!Get the schema name for a tenant.publicr)rr )r rr r rget_tenant_schema_name s  rc@sBeZdZdZddZddZdddZd d Zd d Zd dZ dS)TenantDatabaseRouterzI Database router that routes queries to tenant-specific schemas. cK||r |SdS)z"Suggest the database to read from.N_is_tenant_model_get_tenant_databaseselfmodelhintsr r r db_for_read. z TenantDatabaseRouter.db_for_readcKr)z!Suggest the database to write to.Nrrr r r db_for_write4rz!TenantDatabaseRouter.db_for_writeNcKs|dkrdS||r dSdS)z3Determine if migration should run on this database.tenantsTN)_is_tenant_app)rdb app_label model_namerr r r allow_migrate:s  z"TenantDatabaseRouter.allow_migratecCsgd}|jj|vS)z&Check if model should be tenant-aware. customers inventorysalesorders suppliershrbilling)_metar#)rr tenant_appsr r rrFs z%TenantDatabaseRouter._is_tenant_modelcCsgd}||vS)z*Check if app contains tenant-aware models.r&r )rr#r/r r rr!Lsz#TenantDatabaseRouter._is_tenant_appcCsdS)zIGet the database for current tenant (always default for schema approach).defaultr )rr r rrQsz)TenantDatabaseRouter._get_tenant_databaseN) __name__ __module__ __qualname____doc__rrr%rr!rr r r rr)s   rr1)r5 threading django.confr django.dbrmodelsrlocalrrrrrr r r rs