h dZddlZddlZddlmZmZmZddlmZde dee fdZ de dee ee ffd Z d dee ddfd Z y) zCModule for checking differences between Prisma schema and database.N)ListOptionalTuple)verbose_logger 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 \/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/proxy/db/check_migration.pyextract_sql_commandsr s'2&7&7&= NdTZZ\ NE NLOL ??5 !##O$9$9$;<"$L }}S!4'##O$9$9$;<"$$ 4#:- O1134 3 Os C,C,db_urlc ztjd tjdddd|dddgd d d }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. z"Checking for Prisma schema diff...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) rdebug subprocessrunrstdoutprintCalledProcessErrorstderr)rresultrcommande error_messages rcheck_prisma_schema_diff_helperr-/s=>'!   ",FMM:  1 2 * +'g(% %"9   ( (DQXXJO  mbys$A#A?;A??B:B5/B:5B:c|"tjd}| tdt|\}}|r%t j dj |yy)z2Main function to run the Prisma schema diff check.N DATABASE_URLzDATABASE_URL not setuh🚨🚨🚨 prisma schema out of sync with db. Consider running these sql_commands to sync the two - {})osgetenv Exceptionr-r exceptionformat)rhas_diffmessages rcheck_prisma_schema_diffr7\s\ ~>* >23 37?Hg  v } }  )N)__doc__r0r#typingrrrlitellm._loggingrstrrboolr-r7r8rr?sgI ((+"c"d3i"J*C*E$S /4J*Z  Xc]  d  r8