o ưi @sdZddlZddlZddlmZmZmZddlmZde dee fddZ d e dee ee ffd d Z dd ee ddfd d Z dS)zCModule for checking differences between Prisma schema and database.N)ListOptionalTuple)verbose_logger diff_outputreturncCsdd|dD}g}d}d}|D]4}|dr)|r&|r&||d}d}q|rF|dr@||7}||d}d}q||d 7}q|rP|||S) 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. cSsg|] }|r|qS)strip).0linerrW/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/db/check_migration.py sz(extract_sql_commands.. Fz-- T; )split startswithappendr endswith)rlines sql_commandsZcurrent_commandZ in_sql_blockr rrr extract_sql_commands s*    rdb_urlc Cstdz3tjdddd|dddgd d d d }t|j}|r4td td |D]}t|q(d |fWSd gfWStjyY}zd|j}t|d gfWYd}~Sd}~ww)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...ZprismaZmigratediffz --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_messagerrr check_prisma_schema_diff_helper/s<       r)cCsJ|durtd}|durtdt|\}}|r#td|dSdS)z2Main function to run the Prisma schema diff check.NZ 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)rZhas_diffmessagerrr check_prisma_schema_diff\s  r0)N)__doc__r*rtypingrrrZlitellm._loggingrstrrboolr)r0rrrr s %-