sdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddl m Z dd l m Z dd lmZddlm Zdd lmZdd lmZdd lmZddlZGddej0ZGddeZy)z5Database Migration Service conversion workspaces API.)absolute_import)division)unicode_literalsN)api_util)filter_rewrite) exceptions) api_helper) labels_util)log)resource_property)filesceZdZdZy))UnsupportedConversionWorkspaceDBTypeErrorzHError raised when the conversion workspace database type is unsupported.N)__name__ __module__ __qualname____doc__Mlib/googlecloudsdk/api_lib/database_migration/conversion_workspaces_client.pyrr$sPrrc eZdZdZdgZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZd)dZd)dZdZd)dZdZ d)dZd)dZd)dZd)dZdZ d)d Z!d!Z"d)d"Z#d#Z$d$Z%d)d%Z&d&Z'd'Z(d)d(Z)y)*ConversionWorkspacesClientz2Client for connection profiles service in the API. display_namectj||_tj||_tj ||_|jj|_|jj|_ |jj|_ ||_ t|j jj j"|j jj j$g|_yN)r GetApiVersion _api_versionGetClientInstanceclientGetMessagesModulemessages'projects_locations_conversionWorkspaces_serviceprojects_locations_location_service4projects_locations_conversionWorkspaces_mappingRules_mapping_rules_service_release_track frozenset EntityIssueSeverityValueValuesEnumISSUE_SEVERITY_WARNINGISSUE_SEVERITY_ERRORhigh_severity_issues)self release_tracks r__init__z#ConversionWorkspacesClient.__init__-s ..}=D,,];DK..}=DMKKGGDM![[;;D HH (D ) !!99PP !!99NN+!Drc`|jjjj|Sr)r!DatabaseEngineInfoEngineValueValuesEnumlookup_by_name)r/database_engines r_GetDatabaseEnginez-ConversionWorkspacesClient._GetDatabaseEngine=s( ((>>MM  rcZ|jj|j||S)N)engineversion)r!r3r7)r/r6database_versions r_GetDatabaseEngineInfoz1ConversionWorkspacesClient._GetDatabaseEngineInfoDs/ == + +&&7  , rc|jj}|ji|_d|jd<d|jd<tj||j d}|j |j|j}|j |j|j}|||j||S)zReturns a conversion workspace.*filtertruev2global_settings)globalSettings displayNamesource destination) r!ConversionWorkspacerBr ParseCreateArgsGlobalSettingsValuer<source_database_enginesource_database_versiondestination_database_enginedestination_database_versionr)r/argsconversion_workspace_typerBrErFs r_GetConversionWorkspacez2ConversionWorkspacesClient._GetConversionWorkspaceJs $ A A #d%(D"!'D!11 ';;=NO ( ( ##T%A%AF-- (($*K*KK %&%%  rct|jDcgc](}|j|rtj|*}}|Scc}w)z)Returns update mask for specified fields.)sorted _FIELDS_MAP IsSpecifiedr ConvertToCamelCase)r/rNfield update_fieldss r_GetUpdateMaskz)ConversionWorkspacesClient._GetUpdateMaskbsUD,,--E   E " ,,U3-  s-A cp|j|}|jdr|j|_||fS)z@Returns updated conversion workspace and list of updated fields.r)rXrTrrD)r/conversion_workspacerNrWs r_GetUpdatedConversionWorkspacez9ConversionWorkspacesClient._GetUpdatedConversionWorkspaceks:''-M ')-):):&  ..rcp|jj|}|jj|S)Nnamer!ConversionWorkspacesClient._GetApplyConversionWorkspaceRequestsqMMAA<IIKBM )#1#:#:#<#D#D ++$ dk=+m rc|dk(r*|jjjjS|dk(r*|jjjjS|jjjj S)z,Returns the tree type for database entities.SOURCEDRAFT)r!QDatamigrationProjectsLocationsConversionWorkspacesDescribeDatabaseEntitiesRequestTreeValueValuesEnum SOURCE_TREE DRAFT_TREEDB_TREE_TYPE_UNSPECIFIED)r/ tree_types r _GetTreeTypez'ConversionWorkspacesClient._GetTreeTypesH -- i i } } J JG -- i i } } I I gg{{ U Urc |jj|j||j|j |j |||jjj j}|jdrDatamigrationProjectsLocationsConversionWorkspacesPatchRequestjoinr#Patch) r/r^rN current_cwrZrWrupdate_req_type update_reqs rUpdatez!ConversionWorkspacesClient.UpdateEs55d;J*.*M*MD+'-++-J TT!0 88M* J ==  z **rctj}|jj}|||}|jj |S)zDeletes a conversion workspace. Args: name: str, the name of the resource to delete. Returns: Operation: the operation for deleting the conversion workspace. )r^r)rrr!?DatamigrationProjectsLocationsConversionWorkspacesDeleteRequestr#Delete)r/r^rdelete_req_type delete_reqs rrz!ConversionWorkspacesClient.DeletecsG++-J UU!djAJ ==   ++rc|jj}||j||}|jj |S)a$Commits a conversion workspace. Args: name: str, the reference of the conversion workspace to commit. args: argparse.Namespace, the arguments that this command was invoked with. Returns: Operation: the operation for committing the conversion workspace ) commitConversionWorkspaceRequestr^)r!?DatamigrationProjectsLocationsConversionWorkspacesCommitRequestrir#Commit)r/r^rNcommit_req_type commit_reqs rrz!ConversionWorkspacesClient.CommitusO UU!)-)R)R *  J ==   ++rc|jj}|||jj}|jj |S)zRollbacks a conversion workspace. Args: name: str, the reference of the conversion workspace to rollback. Returns: Operation: the operation for rollbacking the conversion workspace )r^"rollbackConversionWorkspaceRequest)r!ADatamigrationProjectsLocationsConversionWorkspacesRollbackRequest"RollbackConversionWorkspaceRequestr#Rollback)r/r^rollback_req_type rollback_reqs rrz#ConversionWorkspacesClient.RollbacksL WW% +/==+[+[+]L == ! !, //rc|jj}|||j|||}|jj |S)aSeeds a conversion workspace from a connection profile. Args: name: str, the reference of the conversion workspace to seed. source_connection_profile_ref: a Resource reference to a datamigration.projects.locations.connectionProfiles resource for source connection profile. destination_connection_profile_ref: a Resource reference to a datamigration.projects.locations.connectionProfiles resource for destination connection profile. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Operation: the operation for seeding the conversion workspace )r^seedConversionWorkspaceRequest)r!=DatamigrationProjectsLocationsConversionWorkspacesSeedRequestrur#Seed)r/r^rrrsrN seed_req_typeseed_reqs rrzConversionWorkspacesClient.SeedsS0 SS '+'N'N ) . ( H ==  h ''rc|jj}|||j|}|jj |S)a=Import rules in a conversion workspace. Args: name: str, the reference of the conversion workspace to import rules in. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Operation: the operation for importing rules in the conversion workspace )rimportMappingRulesRequest)r!KDatamigrationProjectsLocationsConversionWorkspacesMappingRulesImportRequestrr'Import)r/r^rNimport_rules_req_typeimport_rules_reqs r ImportRulesz&ConversionWorkspacesClient.ImportRulessN aa-"&"D"DT"J  & & - -.> ??rc|jj}|||j|}|jj |S)aLConverts the source entities to draft entities in a conversion workspace. Args: name: str, the reference of the conversion workspace to seed. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Operation: the operation for converting the conversion workspace )r^!convertConversionWorkspaceRequest)r!@DatamigrationProjectsLocationsConversionWorkspacesConvertRequestrr#Convert)r/r^rNconvert_req_type convert_reqs rrz"ConversionWorkspacesClient.ConvertsM VV# *.*T*T + K ==  --rc|jj}|||j||}|jj |S)aapplies a conversion workspace onto the destination database. Args: name: str, the reference of the conversion workspace to seed. destination_connection_profile_ref: a Resource reference to a datamigration.projects.locations.connectionProfiles resource for destination connection profile. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Operation: the operation for applying the conversion workspace )r^applyConversionWorkspaceRequest)r!>DatamigrationProjectsLocationsConversionWorkspacesApplyRequestrr#Apply)r/r^rsrNapply_req_type apply_reqs rrz ConversionWorkspacesClient.ApplysO TT (,(P(P .) I ==  y ))rcp|jj|}|jj|S)zDescribe a conversion worksapce. Args: name: str, the name for conversion worksapce being described. Returns: Described conversion worksapce Resource. r]r_)r/r^ describe_reqs rDescribez#ConversionWorkspacesClient.Describe s8==]] ^L ==  \ **rc g}d}t}|j||||} |jj|}|j}|D]s} |j | j | j| jtj| jjdd|j| du|js |S|j|_)a)Describes database entities in a conversion worksapce. Args: name: str, the name for conversion worksapce being described. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Described entities for the conversion worksapce. DATABASE_ENTITY_TYPE_) parentEntity shortNamer entityTypestatus)strrr#DescribeDatabaseEntitiesdatabaseEntitiesr~r r rsix text_typer replace_GetEntityStatus nextPageTokenr) r/r^rN entity_resultrrrresponseentitiesentitys rDescribeEntitiesz+ConversionWorkspacesClient.DescribeEntitiessMIJ33 iTL 77 Eh**h&"//))KK--(9(9:BB'++F3   # #   (55l rcd}|jD]u}|j|jjjj k(ry|j|jjjj k(std}w|S)zBGet entity status (Action required/review recommended/ no issues). NO_ISSUESACTION_REQUIREDREVIEW_RECOMMENDED)issuesseverityr!r*r+r-r,)r/rrissues rrz+ConversionWorkspacesClient._GetEntityStatus=st F .. ]] & & > > S S T! .. ]] & & > > U U V& Mrc:g}d}|j||t|} |jj|}|jD]0}|j D]}|j d|ji!2|js |S|j|_ {)aDescribe DDLs in a conversion worksapce. Args: name: str, the name for conversion worksapce being described. args: argparse.Namespace, the arguments that this command was invoked with. Returns: DDLs for the entities of the conversion worksapce. rddl) rrr#rr entityDdlr~r$rr) r/r^rN entity_ddlsrrrr entity_ddls r DescribeDDLsz'ConversionWorkspacesClient.DescribeDDLsMsKI//iML 77 Eh--& **J   Z^^  +.  # #   (55l rc g}|jD]}|j|jvs|j|j|j t j|jjddt j|jjddt j|jjdd|j|jd|S)z%Get issues in a conversion worksapce.r r  ISSUE_TYPE_ISSUE_SEVERITY_)r r r  issueType issueSeverity issueCode issueMessage) r r!r.r~r r rrr rtypecodemessage)r/r entity_issuesr"s rGetIssuesHelperz*ConversionWorkspacesClient.GetIssuesHelperhsM 444 4"//))--(9(9:BB'uzz2::="M ]]5>>:BB!2!MM   rcg}|j||t||} |jj|}|jD]"}|j |j |$|js |S|j|_m)z*Describe issues in a conversion worksapce.) rrr#rrextendr4rr) r/r^rrNrr3rrrs rDescribeIssuesHelperz/ConversionWorkspacesClient.DescribeIssuesHelper}sM11    L 77 Eh--&T11&9:.  # #   (55l rcd}|j||||jjjj}|j |j||||jjjj |S)a@Describe database entity issues in a conversion worksapce. Args: name: str, the name for conversion worksapce being described. args: argparse.Namespace, The arguments that this command was invoked with. Returns: Issues found for the database entities of the conversion worksapce. r)r7r!rrrr6r)r/r^rNrr3s rDescribeIssuesz)ConversionWorkspacesClient.DescribeIssuessI--   gg{{ H H M  !!    MM k k   K K   rc  tj|}|j j||j j|}tjjd|zdz|jj|}tj||}d}|j r||k(rk|t"j"j%j'dzd z} tj(| |tjjd | zd }tj(||j tjjd |zd z|zn"tjjd|z|j*r,tjjd|j*z|S#tj$r&tjddj |wxYw)a"Converts application code. Args: name: str, the name of the location. source_file: str, the path of the source file to be converted. target_path: str, the path of the target file to be written. Returns: True if the file was converted, False otherwise. z --source-filerw) sourceCode)r^convertApplicationCodeRequestz Sending file z for conversionFz_%Y%m%d_%H%M%Sz.bakz'The original file content was saved to TzFile z was converted and saved in z No changes were made to the filezResult message: )r rzr{rr|r}r!;DatamigrationProjectsLocationsConvertApplicationCodeRequestConvertApplicationCodeRequestr rPrintr%ConvertApplicationCoder #GetTargetFileNameForApplicationCoder;datetimenowstrftimeWriteFileContents resultMessage) r/r^ source_file target_path source_codeconvert_application_code_reqconvert_application_code_resp target_file converted backup_files r ConvertApplicationCodeSingleFilez;ConversionWorkspacesClient.ConvertApplicationCodeSingleFiles**;7k$(==#l#l &*mm&Q&Q"'R' $m$ JJ_{25FFG 55 ( " @@[KI$//  # ##%../?@ A    [9  5 C i  4?? jj K "@ @; N jj9KGH$22 jj <JJ J g  ! !  + +  / 6 6{ C s G 9Hc~tj|d5}|j|dddy#1swYyxYw)NT)r~)r FileWriterwrite)r/ audit_file audit_contentfs rWriteAuditContentz,ConversionWorkspacesClient.WriteAuditContents7     ggm    s3<c T|jrL|jjdk7r/tjddj |j|j rL|j jdk7r/tjddj |j |j r(|j||j |jytjj|js/tjdd j |j|jrXtjj|js/tjd d j |j|js|j|_d }d }tjj|jd }|j|dtj|jDcgc]K}tjj!tjj|j|r|M}}t#|D] }|d k(r |j%dr|dz }|j|tjj|j||jrNt&j&j)j+ddz|zdz} |j|| |dz }t&j&j)j+ddz|zdz} |j|| t,j.j1d|zdz#|d k(rBt&j&j)j+ddz} |j|| t,j.j1dt3|zdzt3|zdzycc}w)zConverts application code. Args: name: str, the name of the location. args: argparse.Namespace, The arguments that this command was invoked with. Returns: None. POSTGRESQLz--target-dialectzLspecified target dialect [{}] is not supported. Only POSTGRESQL is supportedORACLEz--source-dialectzHspecified source dialect [{}] is not supported. Only ORACLE is supportedNz--source-folderz0specified source folder [{}] is not a directory.z --target-pathzOspecified target path [{}] is not a directory while source folder is specified.rzConversion-Audit.txtz -------- z.javaz%Y-%m-%d %H:%M:%Sz: File z converted z not converted zSkipping file z since it is not a java filez): No files found eligible for conversion zSent z files for conversion, z files were actually converted.)target_dialectupperrr|r}source_dialectrGrOrHrrisdir source_folderrrVlistdirisfilerRendswithrBrCrDr rr?r) r/r^rN total_filesconverted_filesrSrU source_filesfilerTs rr@z1ConversionWorkspacesClient.ConvertApplicationCodes t2288:lJ  + +  vd112   t2288:hF  + +  vd112    ++   $"2"2 WW]]4-- .--  > E E""     "''--0@0@"A--  #VD$4$45    --ko77<< 2 24JKj Z6::d0011a WW^^BGGLL););Q? @ 1 &$ ) )  == !  +  2 2BGGLL!3!3T:DrrsY<&' >D.Z<=#:* Q0E0EQw w r