3 g^@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.cstj||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__l/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/management/commands/reset_db.pyrs<        zCommand.add_argumentscOs|d}|dtkr,tjdtdd|d}tjj|}|dkrLtd||jd}d }}}} } |d krt|\}}}} } |d p|jd p|}|d p|jdp|}|dp|} |dp|jdp|}|d krtd|jdp| } |jdp| } |d} |dr t d|f} nd} | dkr:t ddS|t krzyt j d|tj|Wntk rtYnXnt|tkr,ddl}||d}| jdr| |d<n| |d<| rt| |d <|jf|}d!|}|d"rd nd#}d$||f}t j d%||j|t j d%||j|jn|tkrtjjd&}|rRddl}nddl}dd'i}|rp||d <|r~||d <| r| |d<| r| |d <|jf|}|rd(|_n |jd|j }|d)r.d*|}t j d%|jy|j!|Wn6|j"k r,}zt j#d+t$|WYdd}~XnXd,|}t j d%|y|j!|Wn6|j"k r}zt j#d+t$|WYdd}~XnXd-|}| r|d.| 7}|d/7}tj%r|d0tj%7}n|d17}t j d%||j!|n td2|| dks|dr t d3dS)4z 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"psycopgZ 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.)&rwarningswarnr rZ DATABASESgetrr inputprintrlogginginfoosunlinkOSErrorrZMySQLdb startswithintconnectquerystripr importlibutil find_specr0Zpsycopg2Z autocommitZset_isolation_levelcursorexecuteZProgrammingError exceptionstrZDEFAULT_TABLESPACE)rargsoptionsr$ZdbinfoZenginerrZ database_nameZ database_hostZ database_portrr*ZconfirmZDatabasekwargs connectionZ drop_queryZ utf8_supportZ create_queryZ has_psycopg3Z conn_paramsrDZclose_sessions_queryer"r"r#handleBs                         " "    zCommand.handle)__name__ __module__ __qualname__rrr rM __classcell__r"r")r!r#r s )r )__doc__importlib.utilrAr9r7r2Z 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