U ad+ @sldZddlZddlmZmZddlmZddlmZddlm Z ddl m Z ddl m Z Gd d d eZdS) z Recreates the public schema for current database (PostgreSQL only). Useful for Docker environments where you need to reset database schema while there are active connections. N) BaseCommand CommandError)DEFAULT_DB_ALIAS) connections)settings)POSTGRESQL_ENGINES)RemovedInNextVersionWarningcs,eZdZdZdZfddZddZZS)Commandz&`reset_schema` command implementation.z-Recreates the public schema for this project.csft||jddddddd|jdd d d td d|jd tdtd|jddd dddddS)Nz --noinputz --no-input store_false interactiveTz:Tells Django to NOT prompt the user for input of any kind.)actiondestdefaulthelpz-Rz--routerstorerouterzBUse this router-database instead of the one defined in settings.pyz --databasezGNominates a database to run command for. Defaults to the "%s" database.)rrz-Sz--schemaschemapublicz$Drop this schema instead of "public")super add_arguments add_argumentr)selfparser __class__V/tmp/pip-unpacked-wheel-portabvm/django_extensions/management/commands/reset_schema.pyrs< zCommand.add_argumentsc Os|d}|dtkr,tjdtdd|d}tj|}|dkrLtd||d}|tkrftd|d }|d }|d rt d ||}nd }|d krt ddSt | &} | d || d |W5QRXdS)Ndatabaserz2--router is deprecated. You should use --database.) stacklevelzUnknown database %sZENGINEz8This command can be used only with PostgreSQL databases.NAMErr z You have requested a database schema reset. This will IRREVERSIBLY DESTROY ALL data in the "{}" schema of database "{}". Are you sure you want to do this? Type 'yes' to continue, or 'no' to cancel: yeszReset cancelled.zDROP SCHEMA {} CASCADEzCREATE SCHEMA {})rwarningswarnrrZ DATABASESgetrrinputformatprintrcursorexecute) rargsoptionsrZdbinfoZengineZ database_namerZconfirmr(rrrhandle,s0    zCommand.handle)__name__ __module__ __qualname____doc__rrr, __classcell__rrrrr s r )r0r"Zdjango.core.managementrrZ django.dbrrZ django.confrZdjango_extensions.settingsrZ#django_extensions.utils.deprecationrr rrrrs