a ZEg^@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 --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__g/var/www/sistema_ama/venv/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|WntyrYn0n|tvr.ddl}||d}| dr| |d<n| |d<| rt| |d <|jfi|}d!|}|d"rd nd#}d$||f}t d%|||t d%|||n|tvrtjd&}|rTddl}nddl}dd'i}|rr||d <|r||d <| r| |d<| r| |d <|jfi|}|rd(|_n |d| }|d)r6d*|}t d%|z|!|Wn8|j"y4}zt #d+t$|WYd}~n d}~00d,|}t d%|z|!|Wn8|j"y}zt #d+t$|WYd}~n d}~00d-|}| r|d.| 7}|d/7}tj%r|d0tj%7}n|d17}t d%||!|n td2|| dks |drt 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_specr1Zpsycopg2Z autocommitZset_isolation_levelcursorexecuteZProgrammingError exceptionstrZDEFAULT_TABLESPACE)rargsoptionsr%ZdbinfoZenginerrZ database_nameZ database_hostZ database_portrr+ZconfirmZDatabasekwargs connectionZ drop_queryZ utf8_supportZ create_queryZ has_psycopg3Z conn_paramsrEZclose_sessions_queryer#r#r$handleBs                       & &    zCommand.handle)__name__ __module__ __qualname__rrr rN __classcell__r#r#r!r$r s )r )__doc__importlib.utilrBr: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