jg^dZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZmZddlmZddlmZdd lmZGd d e Zy) 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)RemovedInNextVersionWarningc2eZdZdZfdZedZxZS)Commandz%Resets the database for this project.ct|||jdddddd|jdd d d d |jd ddddd|jdddddd|jdddddd|jdddddd|jdddd td!|jd"td#tz$|jd%d&d d'd d(y))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__s i/var/www/html/Stanley/venv/lib/python3.12/site-packages/django_extensions/management/commands/reset_db.pyr zCommand.add_argumentssO f% mM    3DF   (7P    )G'4n    ,wZT    *W8TH    *W8EUM    "2Z]mm    $\@PZ_X  c|d}|dtk7r!tjdtd|d}tj j |}|td|z|j d}d x}x}x}x} } |d k(rt|\}}}} } |d xs|j d xs|}|d xs|j dxs|}|dxs|} |dxs|j dxs|}|d k(r td|j dxs| } |j dxs| } |d} |drtd|d} nd} | dk7r tdy|tvr. tjd|tj|n6|t"vrddl}||d}| j'dr| |d<n| |d <| rt)| |d!<|j*d6i|}d"|z}|d#rd nd$}d%|d&|}tjd'||j-|tjd'||j-|j/ns|t0vr\t2j4j7d(}|rddl}nddl}dd)i}|r||d <|r||d <| r| |d <| r| |d!<|j*d6i|}|rd*|_n|j?d|jA}|d+r;d,|z}tjd'|j/ |jC|d.|z}tjd'| |jC|d/|z}| r|d0| zz }|d1z }tjJr|d2tjJzz }n|d3z }tjd'||jC|ntd4|z| dk\s|dr td5yy#t $rY"wxYw#|jD$r)}tjFd-tI|Yd}~d}~wwxYw#|jD$r*}tjFd-tI|Yd}~ d}~wwxYw)7z 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 %sENGINEmysqlrUSERrPASSWORDrrNAMEz?You need to specify DATABASE_NAME in your Django settings file.HOSTPORT verbosityrz_ You have requested a database reset. This will IRREVERSIBLY DESTROY ALL data in the database "zQ". 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 `z` zExecuting... "%s"psycopg 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 r DATABASESgetrr inputprintrlogginginfoosunlinkOSErrorr MySQLdb startswithintconnectquerystripr importlibutil find_specr:psycopg2 autocommitset_isolation_levelcursorexecuteProgrammingError exceptionstrDEFAULT_TABLESPACE)r"argsoptionsr(dbinfoenginerr database_name database_host database_portrr3confirmDatabasekwargs connection drop_query utf8_support create_query has_psycopg3 conn_paramsrUclose_sessions_queryes r%handlezCommand.handleBs:& 8  0 0 MMNPkxy zx(H##''1 >4x?@ @H%JLLLxL-L-- W L[\bLc IT8]M=v<&**V"4<:&L&**Z*@LH (D)PVZZ-?P= B `a a 6*;m  6*;m K( = ! 3@ BCGG e  $ %  ^ #  4f= -(} $ &"F'',(5}%!.v!$]!3v)))3F3J7-GJ!():!;2AUL8E|TL LL,j 9   Z ( LL,l ;   \//1 2 ) )$>>33I>L*+#[1K&* F#*2 J'&3 F#&3 F#)))8K8J(, %..q1&&(F'(($ ($$  02F2L2L2NO;NN#781=@J LL,j 9 7z*4mCL 7% ?? 0 0L** 3h6Q6Q QQ # LL,l ; NN< (;fDE E >W]3 % &4e  r 00;%%k3q6::;,, 7!!+s1v66 7sB +OO)<P$ O&%O&)P!8PP!$Q3QQ)__name__ __module__ __qualname__rr r rm __classcell__)r$s@r%rrs$ 2D' RH'H'r&r)__doc__importlib.utilrOrFrDr> django.confrdjango.core.management.baserr django.dbrdjango_extensions.settingsrrr "django_extensions.management.mysqlr "django_extensions.management.utilsr #django_extensions.utils.deprecationr rr=r&r%r{s>   A&XX><Ku'ku'r&