gl zdZddlZddlZddlmZmZmZdedeefdZdedee eeffdZ d deeddfd Z y) zCModule for checking differences between Prisma schema and database.N)ListOptionalTuple diff_outputreturnc|jdDcgc]#}|js|j%}}g}d}d}|D]}|jdr(|r#|r!|j|jd}d}<|s?|j dr)||z }|j|jd}d}y||dzz }|r|j|j|Scc}w)z Extract SQL commands from the Prisma migrate diff output. Args: diff_output (str): The full output from prisma migrate diff. Returns: List[str]: A list of SQL commands extracted from the diff output.  Fz-- T; )splitstrip startswithappendendswith)rlinelines sql_commandscurrent_command in_sql_blocks U/var/www/openai/venv/lib/python3.12/site-packages/litellm/proxy/db/check_migration.pyextract_sql_commandsrs'2&7&7&= N&=dTZZ\&=E NLOL ??5 !##O$9$9$;<"$L }}S!4'##O$9$9$;<"$$ 4#:- O1134 3 Os C,C,db_urlc P tjdddd|dddgddd }t|j}|r,t d t d |D] }t |d|fSd gfS#tj $r(}d |j }t |d gfcYd}~Sd}~wwxYw)aChecks for differences between current database and Prisma schema. Returns: A tuple containing: - A boolean indicating if differences were found (True) or not (False). - A string with the diff output or error message. Raises: subprocess.CalledProcessError: If the Prisma command fails. Exception: For any other errors during execution. prismamigratediffz --from-urlz--to-schema-datamodelz./schema.prismaz--scriptT)capture_outputtextcheckzChanges to DB Schema detectedzRequired SQL commands:Fz*Failed to generate migration diff. Error: N) subprocessrunrstdoutprintCalledProcessErrorstderr)rresultrcommande error_messages rcheck_prisma_schema_diff_helperr+-s'!   ",FMM:  1 2 * +'g(% %"9   ( (DQXXJO  mbys$A#A*&A**B%=B B% B%c|"tjd}| tdt|\}}|rtdj |y)z2Main function to run the Prisma schema diff check.N DATABASE_URLzDATABASE_URL not setz[prisma schema out of sync with db. Consider running these sql_commands to sync the two - {})osgetenv Exceptionr+format)rhas_diffmessages rcheck_prisma_schema_diffr4Ys[ ~>* >23 37?Hg i p p   )N) __doc__r.r!typingrrrstrrboolr+r4r5rr;sdI (("c"d3i"J)C)E$S /4J)X  Xc]  d  r5