ó ¸9—^c@sfdZddlmZmZddlmZddlmZddlm Z defd„ƒYZ dS( s° 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. iÿÿÿÿ(t BaseCommandt CommandError(t connections(tsettings(tinputtCommandcBs&eZdZdZd„Zd„ZRS(s&`reset_schema` command implementation.s-Recreates the public schema for this project.c Cstt|ƒj|ƒ|jddddddtddƒ|jd d dd dd dddd ƒ|jdddd ddddddƒdS(Ns --noinputtactiont store_falsetdestt interactivetdefaultthelps:Tells Django to NOT prompt the user for input of any kind.s-Rs--routertstoretroutersBUse this router-database instead of the one defined in settings.pys-Ss--schematschematpublics$Drop this schema instead of "public"(tsuperRt add_argumentst add_argumenttTrue(tselftparser((sT/tmp/pip-unpacked-wheel-a0M10Y/django_extensions/management/commands/reset_schema.pyRs  c Os|d}tjj|ƒ}|dkr;td|ƒ‚n|jdƒjdƒd}|dkrrtd ƒ‚n|d }|d }|d r«td j||ƒƒ}nd}|dkrÆdGHdSt|j ƒ3} | j dj|ƒƒ| j dj|ƒƒWdQXdS(NR sUnknown database router %stENGINEt.iÿÿÿÿt postgresqltpostgresql_psycopg2tpostgiss8This command can be used only with PostgreSQL databases.tNAMERR sÈ 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: tyessReset cancelled.sDROP SCHEMA {} CASCADEsCREATE SCHEMA {}(RRR( Rt DATABASEStgettNoneRtsplitRtformatRtcursortexecute( RtargstoptionsR tdbinfotenginet database_nameRtconfirmR"((sT/tmp/pip-unpacked-wheel-a0M10Y/django_extensions/management/commands/reset_schema.pythandle#s&       (t__name__t __module__t__doc__R RR*(((sT/tmp/pip-unpacked-wheel-a0M10Y/django_extensions/management/commands/reset_schema.pyRs N( R-tdjango.core.managementRRt django.dbRt django.confRt six.movesRR(((sT/tmp/pip-unpacked-wheel-a0M10Y/django_extensions/management/commands/reset_schema.pyts