o uvXj @sLdZddlmZmZddlmZddlmZddlm Z GdddeZ dS) 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. ) BaseCommand CommandError) connections)settings)inputcs,eZdZdZdZfddZddZZS)Commandz&`reset_schema` command implementation.z-Recreates the public schema for this project.csTtt|||jdddddd|jddd d d d d|jd dd dddddS)Nz --noinput store_false interactiveTz:Tells Django to NOT prompt the user for input of any kind.)actiondestdefaulthelpz-Rz--routerstorerouterr zBUse this router-database instead of the one defined in settings.pyz-Sz--schemaschemapublicz$Drop this schema instead of "public")superr add_arguments add_argument)selfparser __class__o/var/www/html/myproject/venv/lib/python3.10/site-packages/django_extensions/management/commands/reset_schema.pyrs   zCommand.add_argumentsc Os|d}tj|}|durtd||ddd}|dvr&td|d}|d }|d r;td ||}nd }|d krGtd dSt| } | d|| d|WddS1siwYdS)NrzUnknown database router %sENGINE.) postgresqlpostgresql_psycopg2postgisz8This 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 {}) r DATABASESgetrsplitrformatprintrcursorexecute) rargsoptionsrdbinfoengine database_namerconfirmr(rrrhandle#s*  "zCommand.handle)__name__ __module__ __qualname____doc__r rr0 __classcell__rrrrrs  rN) r4django.core.managementrr django.dbr django.confr six.movesrrrrrrs