o uvXj6@s|dZddlZddlZddlmZddlmZmZddlm Z ddl m Z m Z m Z ddlmZddlmZGd d d eZdS) z\ reset_db command originally from http://www.djangosnippets.org/snippets/828/ by dnordberg N)settings) BaseCommand CommandError)input)SQLITE_ENGINESPOSTGRESQL_ENGINES MYSQL_ENGINES)parse_mysql_cnf) signalcommandcs,eZdZdZfddZeddZZS)Commandz%Resets the database for this project.cstt|||jdddddd|jddd d d d|jd d ddddd|jddddddd|jddddddd|jddddddd|jddddd d!d|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--routerrouterrz:Use this router-database other than defined in settings.pyz-cz--close-sessionsclose_sessionszEClose database connections before dropping database (PostgreSQL only))superr add_arguments add_argument)selfparser __class__k/var/www/html/myproject/venv/lib/python3.10/site-packages/django_extensions/management/commands/reset_db.pyrsF       zCommand.add_argumentsc Os|d}tj|}|durtd||d}d}}}} } |dkr0t|\}}}} } |dp:|dp:|}|d pE|d pE|}|d pK|} |d pV|d pV|}|dkr_td|dpe| } |dpl| } |d} |dr}td|f} nd} | dkrtddS|tvrzt d|t |Wn0t yYn'w|t vrddl}||d}| dr| |d<n| |d<| rt| |d<|jd0i|}d|}|drdnd}d ||f}t d!|||t d!|||n|tvrddl}d"d#i}|r||d<|r||d <| r| |d<| r%| |d<|jd0i|}|d|}|d$rkd%|}t d!|z||Wn|jyj}ztd&t|WYd}~nd}~wwd'|}t d!|z||Wn|jy}ztd&t|WYd}~nd}~wwd(|}| r|d)| 7}|d*7}tjr|d+tj7}n|d,7}t d!|||ntd-|| d.ks|drtd/dSdS)1z 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? rNzUnknown database router %sENGINEmysqlrUSERrPASSWORDrrNAMEz?You need to specify DATABASE_NAME in your Django settings file.HOSTPORT verbosityr z 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"database 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.r")r DATABASESgetrr rprintrlogginginfoosunlinkOSErrorrMySQLdb startswithintconnectquerystriprpsycopg2set_isolation_levelcursorexecuteProgrammingError exceptionstrDEFAULT_TABLESPACE)rargsoptionsrdbinfoenginerr database_name database_host database_portrr,confirmDatabasekwargs connection drop_query utf8_support create_query conn_paramsrGclose_sessions_queryer"r"r#handle;s                      zCommand.handle)__name__ __module__ __qualname__rrr r^ __classcell__r"r"r r#r s  %r )__doc__r<r: django.confrdjango.core.management.baserr six.movesrdjango_extensions.settingsrrr"django_extensions.management.mysqlr "django_extensions.management.utilsr r r"r"r"r#s