B `R@sddlZddlZddlZddlZddlZddlmZmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZmZmZddlmZddlmZmZddlmZddlmZdd lmZ e!e"Z#ej$Gd d d e Z%Gd d d eZ&ej$GdddeZ'ddZ(ej$GdddeZ)ej$Gddde)Z*ej$Gddde)Z+Gddde)Z,Gddde Z-Gddde Z.ddZ/d d!Z0e d"d#d$ed%d&Z1dS)'N) BooleanField StringField) view_config) UploadEngineNoTransactionEngineUserInitialPost UserWTFormCredsLoadedFormSessionDBHandler LogSQLCommandSQLData NoOpProcessor ExecuteSQLformat_exception) AuthFailError)CHECKED UNCHECKED)errors_base_view)utils) exceptionscsTeZdZdZejddZejddZejddZejddZ iffdd Z Z S)CopySchemaInitialPostzOGet values from settings when there's not user-supplied values in POST N)defaultFcs:t||dd}|d|_|d|_|d|_|S)Npgwui pgwui_copyZdefault_schemaZdefault_source_dbZdefault_target_db)superbuildschemafrom_dbto_db)selfsettingsZ copy_settings) __class__]/home/kop/projects/devel/pgwui/test_venv/lib/python3.7/site-packages/pgwui_copy/views/copy.pyrLs      zCopySchemaInitialPost.build) __name__ __module__ __qualname____doc__attribrrrforcer __classcell__r"r")r!r#rBs     rc@s8eZdZdZedZedZedZedZ edZ dS)CopySchemaWTFormz@The wtform used to connect to the "gombemi" db to authenticate .zSchema To Copy:zFrom Database:z To Database:zOverwrite Destination Schema:zAllow Copy To Sensitive DBs:N) r$r%r&r'rrrrr overwriter*r"r"r"r#r,Vs r,cs,eZdZdZfddZfddZZS)CopySchemaFormz| Acts like a dict, but with extra methods. Attributes: uh The UploadHandler instance using the form cslt|jjj|d<|jjj|d<|jjj|d<|jj j|d<|jj j|d<dS)z0 Read form data from the client rrrr-r*N) rreadZ_formrdatalowerstriprrr-r*)r)r!r"r#r/js  zCopySchemaForm.readcsjt||}|d|d<|d|d<|d|d<|dr@t}nt}||d<|drZt}nt}||d<|S)zH Produces the dict pyramid will use to render the form. rrrr*r-)rwriterr)rresulterrorsresponseZ force_checkedZoverwrite_checked)r!r"r#r3ys   zCopySchemaForm.write)r$r%r&r'r/r3r+r"r")r!r#r.bs r.cCs|d|f|dk S)zDoes the schema exist?z.SELECT 1 FROM pg_namespace WHERE nspname = %sN)executefetchone)currr"r"r# schema_existssr:csTeZdZdZddZfddZddZfdd Zd d Zd d Z ddZ Z S)CopySchemaHandlerzA Abstract class to copy a schema from one db to another. cCstj|ttdS)N)Zfcip)r.rr,r)rr"r"r# make_formszCopySchemaHandler.make_formcs.t||}|d o$|jddk|d<|S)a Setup dict to render resulting html form Returns: Dict pyramid will use to render the resulting form Reserved keys: errors A list of UploadError exceptions. db_changed Boolean. Whether the db was changed. r5action db_changed)rr3uf)rr4r5r6)r!r"r#r3s  zCopySchemaHandler.writecCs t||S)zrMake a db loader function from an UploadEngine. Input: Side Effects: Yes, lots. )r)ruer"r"r#factoryszCopySchemaHandler.factorycst||}||fS)zInstead of rendering, just our results so we can decide what to do next. Input: errors List of Error instances result Db connection result dict )rrender)rr5r4r6)r!r"r#rDszCopySchemaHandler.rendercCsdj|jd|jddS)z?Return a string for the log detailing an SQL execution attempt.z"In database {db}: By user ({user})ruser)dbrE)formatrA)rr"r"r# to_db_detailszCopySchemaHandler.to_db_detailcCstd||dS)z6Write a success message to the log when executing sql.z{0}: {1}N)logwarningrGrH)rmsgr"r"r#log_sql_successsz!CopySchemaHandler.log_sql_successcCs td||t|dS)z7Write a failture message to the log when executing sql.z {0}: {1}{2}N)rIinforGrHr)rrKexr"r"r#log_sql_failuresz!CopySchemaHandler.log_sql_failure) r$r%r&r'r=r3rCrDrHrLrOr+r"r")r!r#r;s  r;csHeZdZdZejddZfddZfddZdd Z d d Z Z S) FromSchemaHandlerz^Handler to check the source schema and do initial validation. Attributes: cur N)rcst|jjj|_dS)N)rinitrequestregistryr )r)r!r"r#rQs zFromSchemaHandler.initcs|j|_|j}|j}|d}|d|f|dkrXtdd|}|d|||d}t||st dd |}|d||t |S) zMake a db loader function from an UploadEngine. Having the UploadEngine we can test the db content. Input: Side Effects: Yes, lots. rEz7SELECT 1 FROM pg_roles WHERE rolname = %s AND rolsuperNzMust be a superuserz/The supplied user ({0}) is not an administratorzUser not a superuserrzSource schema does not existz/The supplied source schema ({0}) does not exist) r9rAr7r8copy_exZNonAdminUserErrorrGrOr:ZInvalidSchemaErrorrrC)rrBrAr9rErNr)r!r"r#rCs,     zFromSchemaHandler.factorycCs d|_dS)z4We do all the work validating input so need no data.r"N)r0)rr"r"r#get_dataszFromSchemaHandler.get_datac Cs@|j}g}|d}|dkr:td}|d||||jddd}|ddkrztd}|d|||nP|d|kr|d std d |d}|d |dd ||||ddkrt d}|d|||nD|d|dkrAborted: The "Allow..." box must be checked to copy into the "z " databaserzNo from db suppliedz$The from db is the same as the to dbz)Both the from db and the to db are ({0}).zFrom db and to db are the same)rArTZ NoSchemaErrorrOappendr Z NoToDBErrorr1ZLiveDBTargetErrorrGZ NoFromDBErrorZ SameDBError markupsafeescape)rrAr5rrNrVr"r"r# val_inputsB               zFromSchemaHandler.val_input) r$r%r&r'r(r)r rQrCrUrZr+r"r")r!r#rPs    &rPcsTeZdZdZejddZfddZddZdd Z d d Z d d Z ddZ Z S)DropSchemaHandlerz Handler to drop the target schema from the target db. (Extra) Attributes: drop_attempted True when the handler will attempt to drop the schema. F)rcs|j}|jd}t||rp|jdr6d|_t|Std||jdt ddt ||jdn"td||jdt ||Sd S) zMake a db loader function from an UploadEngine. Having the UploadEngine we can test the db content. Input: Side Effects: Yes, lots. rr-Tz2Abort: Schema ({0}) already exists in the ({1}) dbrzThe target schema existsz&The schema ({0}) exists in the {1} db.z+Schema ({0}) does not exist in the ({1}) dbN)r9rAr:drop_attemptedrrCrIrMrGrTZSchemaExistsErrorrXrYr )rrBr9r)r!r"r#rCNs     zDropSchemaHandler.factorycCs|d|jddS)zz,DropSchemaHandler.get_data..) log_success log_failurez VACUUM FULLcs t|S)N)rTZVacuumFullError)rN)rr"r#rarbN) rArWr rGr]r^r_r`r r0)rrAsqlr")rrr#rU}s"    zDropSchemaHandler.get_data)r$r%r&r'r(r)r\rCr]r^r_r`rUr+r"r")r!r#r[Cs  r[c@s(eZdZdZddZddZddZdS) VacuumHandlerz6 Handler to add a user schema to the test db. cCs|ddS)NzSuccessful VACUUM ANALYZE)rL)rr"r"r#log_vacuumanalyze_successsz'VacuumHandler.log_vacuumanalyze_successcCs|d|dS)NzFailed VACUUM ANALYZE)rO)rrNr"r"r#log_vacuumanalyze_failuresz'VacuumHandler.log_vacuumanalyze_failurec sD|j}g}|d|tddfdd|j|jdt||_dS)zq Build and stash the SQL to be executed. Returns: List of SQCommand instances rzVACUUM ANALYZEr"cs t|S)N)rTZVacuumAnalyzeError)rN)rr"r#rarbz(VacuumHandler.get_data..)rcrdN)rArWr rgrhr r0)rrArer")rr#rUs  zVacuumHandler.get_dataN)r$r%r&r'rgrhrUr"r"r"r#rfsrfcs(eZdZdZfddZddZZS)CheckFromSchemaEnginez+Customize credential failure error message.cs$t}td|jjd|jS)NzUnable to login to the ({0}) dbr)rauthfailerror_factoryrrGuhrAdescr)rorig)r!r"r#rjs   z+CheckFromSchemaEngine.authfailerror_factorycCs|jjd|_dS)z-The handler has two db's, save the right one.rN)rkrArF)rr"r"r#read_dbszCheckFromSchemaEngine.read_db)r$r%r&r'rjrnr+r"r")r!r#ris ric@seZdZdZddZdS) ToDBEnginezUse the to_db to login.cCs|jjd|_dS)Nr)rkrArF)rr"r"r#rnszToDBEngine.read_dbN)r$r%r&r'rnr"r"r"r#rosroc Cs^|d}|d}|dd}|d}|d}|d}|d}|d } tjd d d } | j} | d j|||||d| d||| |} yr?zProcessing form submissionzFailed feasiblity testr@rz%Target schema dropped but copy failedzqThe ({schema}) schema was dropped from the {db} db but the copy failed so now it no longer exists in the {db} db.rr)rrFr5FrrEz$Successfully copied schema: Schema (z ): From DB (z ): To DB (z ): By user zVACUUM ANALYZE failedzVACUUM ANALYZE failed in the z db, performance may sufferzVacuum analyzed zFinished processing request)POSTrIdebugrPrQrirunrSr rr[rorrArTZCopyCommandErrorr\rWZSchemaDroppedErrorrGrXrYextendrMrfZSchemaCopiedErrorupdate)rRpostZform_submittedZ from_handlerr6r5r Z drop_handlerrNZ copyerrorsrrrrEZvacuum_handlerZ vacuumoutZ vacuumerrorsr"r"r#copy_schema_viewKs^     "  r)2r(loggingrrrXZwtformsrrZ pyramid.viewrZpgwui_core.corerrrrr r r r r rrZpgwui_core.exceptionsrZpgwui_core.constantsrrZpgwui_common.viewrrrrrT getLoggerr$rIsrr,r.r:r;rPr[rfrirorrrr"r"r"r#sB 4      1?fY"s