29 lines
4.5 KiB
Plaintext
29 lines
4.5 KiB
Plaintext
|
a
|
|||
|
|
0¦Hj{ ã @ s> d Z ddlmZ ddlmZmZ ddlmZmZm Z ddl
|
|||
|
|
m Z ddl m
|
|||
|
|
Z
|
|||
|
|
ddlmZ eƒ Zej d¡r~eejejd d
|
|||
|
|
�Zn eejejejd ejd d �Ze eed d d d
|
|||
|
|
�Ze ƒ Zedœdd„Zeeedf dœdd„ƒZG dd„ dƒZee ddœdd„Z eddœdd„Z e dd œd d „Z e dd œd d „Z e e!d œd!d"„Z"dS )#u‡
|
|||
|
|
Databaskonfiguration och session-hantering för LandveX Admin Backend.
|
|||
|
|
Inkluderar tenant schema-isolering via PostgreSQL search_path.
|
|||
|
|
é )Úasynccontextmanager)ÚAsyncGeneratorÚOptional)Úcreate_async_engineÚ AsyncSessionÚasync_sessionmaker)Údeclarative_base)Útext)Ú get_settingsZsqliteT)ÚechoÚfuture)Z pool_sizeZ max_overflowZ
|
|||
|
|
pool_pre_pingr r F)Úclass_Zexpire_on_commitZ autoflushZ
|
|||
|
|
autocommit)Úreturnc C sž t ƒ 4 I dH šv} zNz| V | ¡ I dH W n" tyL | ¡ I dH ‚ Y n0 W | ¡ I dH n| ¡ I dH 0 W d ƒI dH qš1 I dH s�0 Y dS )u" Dependency för FastAPI-endpoints.N©ÚAsyncSessionLocalÚcommitÚ ExceptionZrollbackÚclose©Úsession© r úE/home/bernt/.openclaw/workspace/landvex-admin-backend/app/database.pyÚget_db, s
|
|||
|
|
r Nc C sž t ƒ 4 I dH šv} zNz| V | ¡ I dH W n" tyL | ¡ I dH ‚ Y n0 W | ¡ I dH n| ¡ I dH 0 W d ƒI dH qš1 I dH s�0 Y dS )u& Context manager för databassessioner.Nr r r r r Úget_db_context9 s
|
|||
|
|
r c @ s6 e Zd ZdZd
|
|||
|
|
eee dœdd„Zdd„ Zdd „ Z dS ) Ú
|
|||
|
|
TenantContextu© Context manager för tenant schema-isolering.
|
|||
|
|
|
|||
|
|
Använder PostgreSQL search_path och application context
|
|||
|
|
för att säkerställa dataisolering mellan tenants.
|
|||
|
|
N©r Ú
|
|||
|
|
tenant_schemac C s || _ |ptj| _d S )N)r ÚsettingsÚDEFAULT_TENANT_SCHEMAr )Úselfr r r r r Ú__init__N s zTenantContext.__init__c à sB | j td| j› d�ƒ¡I dH | j tdƒd| ji¡I dH | S )u- Sätt tenant context pÃ¥ databasanslutningen.úSET search_path TO "ú ", publicNú6SELECT set_config('app.current_tenant', :tenant, true)Útenant)r Úexecuter r )r r r r Ú
|
|||
|
|
__aenter__R s ÿ
|
|||
|
|
þ
|
|||
|
|
zTenantContext.__aenter__c à s0 | j tdƒ¡I dH | j tdƒ¡I dH dS )u Ã…terställ tenant context.úSET search_path TO publicNú1SELECT set_config('app.current_tenant', '', true)F)r r% r )r Úexc_typeÚexc_valÚexc_tbr r r Ú __aexit__] s
|
|||
|
|
ÿ
|
|||
|
|
zTenantContext.__aexit__)N)
|
|||
|
|
Ú__name__Ú
|
|||
|
|
__module__Ú __qualname__Ú__doc__r r Ústrr r& r, r r r r r G s r )r r r c à s: | td|› d�ƒ¡I dH | tdƒd|i¡I dH dS )u) Sätt tenant schema för aktuell session.r! r" Nr# r$ ©r% r r r r r Úset_tenant_schemaf s ÿ
|
|||
|
|
þr3 )r r c à s, | tdƒ¡I dH | tdƒ¡I dH dS )u Ã…terställ schema till default.r' Nr( r2 r r r r Úreset_tenant_schemaq s ÿr4 )Ú schema_namer c à sx t ¡ 4 I dH šN}| td| › d�ƒ¡I dH | td| › d�ƒ¡I dH W d ƒI dH qt1 I dH sj0 Y dS )u% Skapa ett nytt schema för en tenant.Nz CREATE SCHEMA IF NOT EXISTS "ú"zGRANT USAGE ON SCHEMA "z " TO PUBLIC©ÚengineÚbeginr% r ©r5 Úconnr r r Úcreate_tenant_schemay s r< c à s\ t ¡ 4 I dH š2}| td| › d�ƒ¡I dH W d ƒI dH qX1 I dH sN0 Y dS )zTa bort ett tenant schema.NzDROP SCHEMA IF EXISTS "z " CASCADEr7 r: r r r Údrop_tenant_schema€ s r= c à sb t ¡ 4 I dH š8}| tdƒd| i¡I dH }| ¡ W d ƒI dH S 1 I dH sT0 Y dS )zKolla om ett schema finns.NzTSELECT EXISTS(SELECT 1 FROM information_schema.schemata WHERE schema_name = :schema)Úschema)r8 Úconnectr% r Úscalar)r5 r; Úresultr r r Ú
|