a 8Xh"@sddlZddlmZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZdd lmZdd lmZGd d d e ZdS) Ncount)settings) BaseCommand CommandError)DEFAULT_DB_ALIAS)TEST_DATABASE_PREFIX)SQLITE_ENGINESPOSTGRESQL_ENGINES MYSQL_ENGINES)parse_mysql_cnf) signalcommand)RemovedInNextVersionWarningcs,eZdZdZfddZeddZZS)Commandz%Drops test database for this project.cst||jddddddd|jdd d d dd d|jd dd dddd|jddd dddd|jddd dtdd|jdtdtddS)Nz --noinputz --no-input store_false interactiveTz:Tells Django to NOT prompt the user for input of any kind.)actiondestdefaulthelpz-Uz--userstoreuserz=Use another user for the database then defined in settings.pyz-Pz --passwordpasswordzAUse another password for the database then defined in settings.pyz-Dz--dbnamedbnamez5Use another database name then defined in settings.pyz-Rz--routerrouterz:Use this router-database other then defined in settings.pyz --databasezGNominates a database to run command for. Defaults to the "%s" database.)rr)super add_arguments add_argumentr)selfparser __class__}/var/www/viveiro_mudafortebrasil/venv/lib/python3.9/site-packages/django_extensions/management/commands/drop_test_database.pyrs2     zCommand.add_argumentsc s$|d}|dtkr,tjdtdd|d}tj|}|durLtd||d}d }}}} |d krt|\}}}} |d p|d p|}|d p|dp|}z|ddWnt ydYn0durt |dp|ddus d krtd|dp |}|dp0| } |d} |drVt dj d} nd} | dkrpt ddSfdd} |tvrtd|dd} z@| | D]2}tj|sqtd |t|qWntyYdS0n |tvrd!dl}||d"}|d#r*||d$<n||d%<| rDt| |d&<|jfi|}|}| d'j D]L}d(|}||}|d)krqd*|}td+|d,||qfnL|tvrtj d-}|rd!dl!}nd!dl"}dd.i}|r||d <|r ||d <|r||d%<| r&| |d&<|jfi|}|rDd/|_#n |$d!|}| d'j D]}d0|}zF|||j%d)krWqd1|}td+|d,||Wn>|j&y}z"t'd2t(|WYd}~dSd}~00q`n td3|| dks|dr t d4dS)5z$Drop test database for this project.databaserz2--router is deprecated. You should use --database.) stacklevelNzUnknown database %sZENGINEZmysqlrUSERrZPASSWORDTESTNAMErz?You need to specify DATABASE_NAME in your Django settings file.ZHOSTZPORT verbosityraG You have requested to drop all test databases. This will IRREVERSIBLY DESTROY ALL data in the database "{db_name}" and all cloned test databases generated via the "--parallel" flag (these are sequentially named "{db_name}_1", "{db_name}_2", etc.). Are you sure you want to do this? Type 'yes' to continue, or 'no' to cancel: )db_nameyeszReset cancelled.c3s(VfddtdDEdHdS)a} Return a generator of all possible test database names. e.g., 'test_foo', 'test_foo_1', test_foo_2', etc. formatter: func returning a clone db name given the primary db name and the clone's number, e.g., 'test_foo_1' for mysql/postgres, and 'test_foo_1..sqlite3' for sqlite (re: double dots, see comments). c3s|]}|VqdS)Nr").0n) database_name formatterr"r# qz=Command.handle..get_database_names..Nrr1r0r5r#get_database_namesgs z*Command.handle..get_database_nameszUnlinking %s databasescSstj|\}}d|||S)Nz{}_{}.{})ospathsplitextformat)namenumberfilenameextr"r"r#format_filename{sz'Command.handle..format_filenamezUnlinking database named "%s"r)rpasswd/Z unix_sockethostportz{}_{}zKSELECT SCHEMA_NAME FROM INFORMATION_SCHEMA.SCHEMATA WHERE SCHEMA_NAME='%s';r4zDROP DATABASE IF EXISTS `%s`z Executing: ""psycopgZ template1Tz>SELECT datname FROM pg_catalog.pg_database WHERE datname='%s';zDROP DATABASE IF EXISTS "%s";z Error: %szUnknown database engine %szReset successful.))rwarningswarnrrZ DATABASESgetrr KeyErrorrinputr;printr logginginfor8r9isfileunlinkOSErrorr ZMySQLdb startswithintconnectcursorexecuter importlibutil find_specrFZpsycopg2Z autocommitZset_isolation_levelZrowcountZProgrammingError exceptionstr)rargsoptionsr$ZdbinfoZenginerrZ database_hostZ database_portr+Zconfirmr7r@r,ZDatabasekwargs connectionrUZ exists_queryZ row_countZ drop_queryZ has_psycopg3Z conn_paramser"r6r#handle1s                            zCommand.handle)__name__ __module__ __qualname__rrr ra __classcell__r"r"r r#rs r)importlib.utilrW itertoolsrr8rMrGZ django.confrZdjango.core.management.baserrZ django.dbrZ django.db.backends.base.creationrZdjango_extensions.settingsr r r Z"django_extensions.management.mysqlr Z"django_extensions.management.utilsr Z#django_extensions.utils.deprecationrrr"r"r"r#s