3 g"@sddlZddlmZddlZddlZddlZddlmZddl m Z m Z ddl m Z ddlmZddlmZmZmZddlmZdd lmZdd lmZGd d d e ZdS) N)count)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.cstj||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__v/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django_extensions/management/commands/drop_test_database.pyrs&      zCommand.add_argumentscs|d}|dtkr,tjdtdd|d}tjj|}|dkrLtd||jd}d }}}} |d krt|\}}}} |d p|jd p|}|d p|jdp|}y|ddWnt k rdYnXdkrt |dp|jddks d krtd|jdp"|}|jdp2| } |d} |drXt dj d} nd} | dkrrt ddSfdd} |tkrtjd|dd} y@x:| | D].}tjj|sPtjd |tj|qWWntk rdSXn|tkrd!dl}||d"}|jd#r,||d$<n||d%<| rFt| |d&<|jf|}|j}xV| d'j D]H}d(|}|j|}|d)krPd*|}tjd+|d,|j|qfWn>|tkrtjj d-}|rd!dl!}nd!dl"}dd.i}|r||d <|r||d <|r||d%<| r$| |d&<|jf|}|r>d/|_#n |j$d!|j}x| d'j D]}d0|}y@|j||j%d)krPd1|}tjd+|d,|j|Wn6|j&k r}ztj'd2t(|dSd}~XnXq\Wn td3|| dks|drt d4dS)5z$Drop test database for this project.databaserz2--router is deprecated. You should use --database.) stacklevelNzUnknown database %sZENGINEZmysqlrUSERrZPASSWORDZTESTNAMErz?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! qsz=Command.handle..get_database_names..N)r)r.)r-)r.r!get_database_namesgs z*Command.handle..get_database_nameszUnlinking %s databasescSstjj|\}}dj|||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';r0zDROP 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.))rwarningswarnr rZ DATABASESgetrr KeyErrorrinputr5printrlogginginfor2r3isfileunlinkOSErrorr ZMySQLdb startswithintconnectcursorexecuter importlibutil find_specr@Zpsycopg2Z autocommitZset_isolation_levelZrowcountZProgrammingError exceptionstr)rargsoptionsr"ZdbinfoZenginerrZ database_hostZ database_portr(Zconfirmr1r:r)ZDatabasekwargs connectionrOZ exists_queryZ row_countZ drop_queryZ has_psycopg3Z conn_paramser )r-r!handle1s                          zCommand.handle)__name__ __module__ __qualname__rrr r[ __classcell__r r )rr!rs r)importlib.utilrQ itertoolsrr2rGrAZ django.confrZdjango.core.management.baserrZ django.dbrZ django.db.backends.base.creationrZdjango_extensions.settingsrr r Z"django_extensions.management.mysqlr Z"django_extensions.management.utilsr Z#django_extensions.utils.deprecationr rr r r r!s