U adD@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||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 --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__R/tmp/pip-unpacked-wheel-portabvm/django_extensions/management/commands/reset_db.pyrs zCommand.add_argumentsc Os|d}|dtkr,tjdtdd|d}tj|}|dkrLtd||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 krzzt d|t|Wntk rtYnXnJ|tkr,ddl}||d}| dr| |d<n| |d<| rt| |d <|jf|}d!|}|d"rd nd#}d$||f}t d%|||t d%|||n|tkrddl}dd&i}|rT||d <|rb||d <| rp| |d<| r~| |d <|jf|}|d|}|d'rd(|}t d%|z||Wn6|jk r}zt d)t|W5d}~XYnXd*|}t d%|z||Wn6|jk r\}zt d)t|W5d}~XYnXd+|}| rx|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 %sZENGINEZmysqlrUSERrZPASSWORDrrNAMEz?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_paramsrAZclose_sessions_queryer#r#r$handleAs                       " "    zCommand.handle)__name__ __module__ __qualname__rrr rJ __classcell__r#r#r!r$r s )r )__doc__r9r7r2Z 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