a ܌xd6@sdZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z mZddlmZddlmZdd lmZGd d d eZdS) z\ reset_db command originally from http://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||jdddddd|jddd d d d|jd d ddddd|jddddddd|jddddddd|jddddddd|jddddtd d|jd!td"td#|jd$d%dd&d d'ddS)(Nz --noinput 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 --databasezGNominates a database to run command for. Defaults to the "%s" database.)rrz-cz--close-sessionsclose_sessionszEClose database connections before dropping database (PostgreSQL only))super add_arguments add_argumentr)selfparser __class__f/var/www/html/Ranjet/env/lib/python3.9/site-packages/django_extensions/management/commands/reset_db.pyrsN       zCommand.add_argumentsc Os|d}|dtkr,tjdtdd|d}tj|}|durLtd||d}d }}}} } |d krt|\}}}} } |d p|d p|}|d p|dp|}|dp|} |dp|dp|}|d krtd|dp| } |dp| } |d} |dr t d|f} nd} | dkr:t ddS|t vrxzt d|t|WntyrYn0nV|tvr.ddl}||d}| dr| |d<n| |d<| rt| |d <|jfi|}d!|}|d"rd nd#}d$||f}t d%|||t d%|||n|tvrddl}dd&i}|rV||d <|rd||d <| rr| |d<| r| |d <|jfi|}|d|}|d'r d(|}t d%|z||Wn8|jy }zt d)t|WYd}~n d}~00d*|}t d%|z||Wn8|jyf}zt d)t|WYd}~n d}~00d+|}| r|d,| 7}|d-7}tj r|d.tj 7}n|d/7}t d%|||n td0|| dks|drt d1dS)2z 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 %sZENGINEmysqlrUSERrZPASSWORDrrNAMEz?You need to specify DATABASE_NAME in your Django settings file.ZHOSTZPORT 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/Z unix_sockethostportzDROP DATABASE IF EXISTS `%s`rzCHARACTER SET utf8zCREATE DATABASE `%s` %szExecuting... "%s"Z template1rz 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.)!rwarningswarnr rZ DATABASESgetrr inputprintrlogginginfoosunlinkOSErrorrZMySQLdb startswithintconnectquerystriprZpsycopg2Zset_isolation_levelcursorexecuteZProgrammingError exceptionstrZDEFAULT_TABLESPACE)rargsoptionsr%ZdbinfoZenginerrZ database_nameZ database_hostZ database_portrr,ZconfirmZDatabasekwargs connectionZ drop_queryZ utf8_supportZ create_queryZ conn_paramsrBZclose_sessions_queryer#r#r$handleAs                     & &    zCommand.handle)__name__ __module__ __qualname__rrr rK __classcell__r#r#r!r$r s )r )__doc__r:r8r3Z django.confrZdjango.core.management.baserrZ django.dbrZdjango_extensions.settingsrrrZ"django_extensions.management.mysqlr Z"django_extensions.management.utilsr Z#django_extensions.utils.deprecationr r r#r#r#r$s