o /h!@sdZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZmZddlmZddlmZdd lmZGd d d e ZdS) z] reset_db command originally from https://www.djangosnippets.org/snippets/828/ by dnordberg N)settings) BaseCommand CommandError)DEFAULT_DB_ALIAS)SQLITE_ENGINESPOSTGRESQL_ENGINES MYSQL_ENGINES)parse_mysql_cnf) signalcommand)RemovedInNextVersionWarningcs,eZdZdZfddZeddZZS)Commandz%Resets the database for this project.cst||jddddddd|jdd d d d d|jd dddddd|jddddddd|jddddddd|jddddddd|jdddd td!d|jd"td#td$|jd%d&d d'd d(ddS))Nz --noinputz --no-input store_false interactiveTz:Tells Django to NOT prompt the user for input of any kind.)actiondestdefaulthelpz --no-utf8 store_trueno_utf8_supportFz3Tells Django to not create a UTF-8 charset databasez-Uz--userstoreuserz=Use another user for the database than defined in settings.pyz-Oz--ownerownerz[Use another owner for creating the database than the user defined in settings or via --userz-Pz --passwordpasswordzAUse another password for the database than defined in settings.pyz-Dz--dbnamedbnamez5Use another database name than defined in settings.pyz-Rz--routerrouterz:Use this router-database other than defined in settings.pyz --databasez>Nominates a database to run command for. Defaults to the "%s".)rrz-cz--close-sessionsclose_sessionszXClose database connections before dropping database (currently works on PostgreSQL only))super add_arguments add_argumentr)selfparser __class__i/var/www/html/env_mimamsha/lib/python3.10/site-packages/django_extensions/management/commands/reset_db.pyrs   zCommand.add_argumentsc Os |d}|dtkrtjdtdd|d}tj|}|dur&td||d}d }}}} } |d krBt|\}}}} } |d pL|d pL|}|d pW|dpW|}|dp]|} |dph|dph|}|d krqtd|dpw| } |dp~| } |d} |drt d|f} nd} | dkrt ddS|t vrzt d|t|WnFtyYn=w|tvrddl}||d}| dr| |d<n| |d<| rt| |d <|jd4i|}d!|}|d"rd nd#}d$||f}t d%|||t d%|||n|tvrtjd&}|r"ddl}nddl}dd'i}|r1||d <|r8||d <| r?| |d<| rF| |d <|jd4i|}|rUd(|_n|d| }|d)rd*|}t d%|z|!|Wn|j"y}zt #d+t$|WYd}~nd}~wwd,|}t d%|z|!|Wn|j"y}zt #d+t$|WYd}~nd}~wwd-|}| r|d.| 7}|d/7}tj%r|d0tj%7}n|d17}t d%||!|ntd2|| dks|drt d3dSdS)5z Reset the database for this project. Note: Transaction wrappers are in reverse as a work around for autocommit, anybody know how to do this the right way? databaserz2--router is deprecated. You should use --database.) stacklevelNzUnknown database %sENGINEmysqlrUSERrPASSWORDrrNAMEz?You need to specify DATABASE_NAME in your Django settings file.HOSTPORT verbosityrz You have requested a database reset. This will IRREVERSIBLY DESTROY ALL data in the database "%s". Are you sure you want to do this? Type 'yes' to continue, or 'no' to cancel: yeszReset cancelled.zUnlinking %s databaser)rpasswd/ unix_sockethostportzDROP DATABASE IF EXISTS `%s`rzCHARACTER SET utf8zCREATE DATABASE `%s` %szExecuting... "%s"psycopg template1Trz SELECT pg_terminate_backend(pg_stat_activity.pid) FROM pg_stat_activity WHERE pg_stat_activity.datname = '%s'; z Error: %szDROP DATABASE "%s";zCREATE DATABASE "%s"z WITH OWNER = "%s" z ENCODING = 'UTF8'z TABLESPACE = %s;;zUnknown database engine %szReset successful.r#)&rwarningswarnr r DATABASESgetrr inputprintrlogginginfoosunlinkOSErrorrMySQLdb startswithintconnectquerystripr importlibutil find_specr7psycopg2 autocommitset_isolation_levelcursorexecuteProgrammingError exceptionstrDEFAULT_TABLESPACE)rargsoptionsr%dbinfoenginerr database_name database_host database_portrr0confirmDatabasekwargs connection drop_query utf8_support create_query has_psycopg3 conn_paramsrQclose_sessions_queryer#r#r$handleds                            zCommand.handle)__name__ __module__ __qualname__rrr ri __classcell__r#r#r!r$r s  Jr )__doc__importlib.utilrKrBr@r: django.confrdjango.core.management.baserr django.dbrdjango_extensions.settingsrrr"django_extensions.management.mysqlr "django_extensions.management.utilsr #django_extensions.utils.deprecationr r r#r#r#r$s