hdZddlZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z ddlmZddlmZGddZGd d Z dd e e eefd efd Zy)zs This file contains the PrismaWrapper class, which is used to wrap the Prisma client and handle the RDS IAM token. N)datetime timedelta)AnyOptionalUnion)verbose_proxy_logger) str_to_boolcjeZdZdedefdZdeedefdZdeefdZ dd ed eefd Z d efd Z y) PrismaWrapperoriginal_prismaiam_token_db_authc ||_||_yN)_original_prismar )selfr r s Z/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/db/prisma_client.py__init__zPrismaWrapper.__init__s /!2 token_urlreturnc4|ytjj|}tjj|}tjj |j }|j ddgd}| tdt|}|j ddgd}|s td tj|d}|t| z} tj} | | kDS#t$r} td | d} ~ wwxYw) NTz X-Amz-Expiresrz.X-Amz-Expires parameter is missing or invalid.z X-Amz-Datez+X-Amz-Date parameter is missing or invalid.z%Y%m%dT%H%M%SZzInvalid X-Amz-Date format: )seconds) urllibparseunquoteurlparseparse_qsqueryget ValueErrorintrstrptimerutcnow) rr decoded_url parsed_url query_paramsexpires expires_inttoken_time_str token_timeeexpiration_time current_times ris_token_expiredzPrismaWrapper.is_token_expireds  ll**95 \\**;7 ||,,Z-=-=> ""?TF;A> ?MN N'l &)),=a@JK K @!**>;KLJ %y'EE ( o-- @:1#>? ? @s=C;; DDDc p|jrddlm}tjd}tjd}tjd}tjd}tjd}||||}d |d |d |d |d | }|r|d |z }|tj d<|Sy)Nr)generate_iam_auth_token DATABASE_HOST DATABASE_PORT DATABASE_USER DATABASE_NAMEDATABASE_SCHEMA)db_hostdb_portdb_userz postgresql://:@/z?schema= DATABASE_URL)r litellm.proxy.auth.rds_iam_tokenr1osgetenvenviron) rr1r7r8r9db_name db_schematoken_db_urls rget_rds_iam_tokenzPrismaWrapper.get_rds_iam_token?s  ! ! Pii0Gii0Gii0Gii0G "34I+'E &gYaway'!G9UGXi[11)0BJJ~ &NrN new_db_url http_clientcKddlm}||||_n ||_|jjd{y7w)Nr)Prisma)http)prismarJrconnect)rrGrHrJs rrecreate_prisma_clientz$PrismaWrapper.recreate_prisma_clientVs@ "  "$* $eZdZedefdZeddedefdZy) PrismaManagerrctjjt}tjj tjj |}|S)z(Get the path to the migrations directory)r?pathabspath__file__dirname)rednames r_get_prisma_dirzPrismaManager._get_prisma_dirxs:''//(+ 89 r use_migratec\ttjdxs|}tdD]}tj}t j }tj| |rD ddlm }t j }|j| tj|cStjgd d d  tj|y y#t$r9}tjd|dYd}~tj|yd}~wwxYw#tj $rHtj"d|dzdt%j&t)j*ddYnutj,$r`}d|z }|dkDrd|dnd}tj"d|d|t%j&t)j*ddYd}~nd}~wwxYwtj|#tj|wxYw)z Set up the database using either prisma migrate or prisma db push Returns: bool: True if setup was successful, False otherwise USE_PRISMA_MIGRATEr)ProxyExtrasDBManagerz3LiteLLM: Failed to import proxy extras. Got zNF)rj)rLdbpushz--accept-data-loss<T)timeoutcheckzAttempt z timed outz Retrying... (z attempts left)rz(The process failed to execute. Details: .)r r?r@rangegetcwdrbrichdirlitellm_proxy_extras.utilsrn ImportErrorrerrorsetup_database subprocessrVTimeoutExpiredwarningtimesleeprandom randrangeCalledProcessError)rjattempt original_dir prisma_dirrnr, attempts_left retry_msgs rrzPrismaManager.setup_databases"")),@"ABQk QxG99;L&668J HHZ $ '%S"/!>!>!@J/>>;>W0&+NNF ""  &Q RC'%,22TUVTWW^_ %8&A%",, 4$,,x! }J-OP 6++Ar2300 4 !G %q(%]O?C %,,>qc9+N 6++Ar233 4&&sh/D.2C)8%D.5D.) D+2D& D.&D++D..AG:HG:AG50H5G::HH+N)F)r[r\r] staticmethodr_rir^rr`rrrbrbws=S 2D2T22rrbdisable_updatesrc|tjdd}t|tr t |}t | S)ao Determines if Prisma Schema updates should be applied during startup. Args: disable_updates: Controls whether schema updates are disabled. Accepts boolean or string ('true'/'false'). Defaults to checking DISABLE_SCHEMA_UPDATE env var. Returns: bool: True if schema updates should be applied, False if updates are disabled. Examples: >>> should_update_prisma_schema() # Checks DISABLE_SCHEMA_UPDATE env var >>> should_update_prisma_schema(True) # Explicitly disable updates >>> should_update_prisma_schema("false") # Enable updates using string DISABLE_SCHEMA_UPDATEfalse)r?r@ isinstancer_r r^)rs rshould_update_prisma_schemars=$))$;WE/3'%o6O$ $$rr)__doc__rRr?rrrr urllib.parserrtypingrrrlitellm._loggingrlitellm.secret_managers.mainr r rbr^r_rr`rrrsp  (''14aaH;;~37%eD#I./% %r