o "Æi<ã@sxdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z ddl mZe e¡ZGdd „d ƒZdS) u 租户数æ®å¯¼å…¥æœåŠ¡ éN)ÚPath)ÚDictÚAny)ÚSession)Úsettings)Ú AppExceptionc@sŠeZdZdZdefdd„Zdededee e ffdd „Z dedee e ffd d „Z dedee e ffd d „Z dedee e ffdd„ZdS)ÚTenantImportServiceu租户数æ®å¯¼å…¥æœåŠ¡ÚdbcCs ||_dS)N)r )Úselfr ©r úS/lsinfo/ai/hellotax_ai/base_platform/app/services/platform/tenant_import_service.pyÚ__init__s zTenantImportService.__init__Ú tenant_idÚ export_dirÚreturnc Csöddlm}d|›}ddlm}|tjƒ}|jpd}|jpd}|j  d¡} |j } |j } |j   |d|›ƒ¡|j  ¡d | i} dd l} |  d ¡pJd }|d d}| ¡rÓt|dƒ }| ¡}Wd ƒn1siwY| d|¡}dd l}|jdddd}| |¡|j}Wd ƒn1s“wYz0|d|dt|ƒd| d| d|g }tj|| ddd}|jdkr¾td|j›ƒ‚Wdd l}|  |¡n dd l}|  |¡w|d d}| ¡r÷|d|dt|ƒd| d| dt|ƒg }tj|| ddd}ddiS) u导入 PostgreSQL æ•°æ®r)ÚtextÚtenant_)ÚurlparseÚ localhosti8ú/zCREATE SCHEMA IF NOT EXISTS Ú PGPASSWORDNÚpsqlz(/opt/homebrew/opt/postgresql@15/bin/psqlÚ postgresqlz tenant_1.sqlÚrÚtenant_1Úwz.sqlF)ÚmodeÚsuffixÚdeletez-hz-pz-Uz-dz-fT)ÚenvÚcapture_outputrzPostgreSQL import failed: zpublic_tenant_data.sqlÚstatusÚsuccess)!Ú sqlalchemyrÚ urllib.parserrÚ DATABASE_URLÚhostnameÚportÚpathÚlstripÚusernameÚpasswordr ÚexecuteÚcommitÚshutilÚwhichÚexistsÚopenÚreadÚreplaceÚtempfileÚNamedTemporaryFileÚwriteÚnameÚstrÚ subprocessÚrunÚ returncoderÚstderrÚosÚunlink)r rrrÚ schema_namerÚparsedÚdb_hostÚdb_portÚdb_nameÚdb_userÚ db_passwordrr.rÚ schema_fileÚfÚ sql_contentr4ÚtmpÚtmp_fileÚcmdÚresultr=Ú public_filer r r Úimport_postgresqlsh           ÿ  þú ÿ ÿ  úz%TenantImportService.import_postgresqlc Cs¼ddlm}|dd}| ¡sddiS|ƒ}t|ddd }| ¡ d ¡}Wd ƒn1s0wY| ¡}|D] }| ¡rG| |¡qs