PdZddlmZddlmZddlmZddlmZdZdZdZ d Z y ) zGHelper functions for constructing and validating AlloyDB user requests.)absolute_import)division)unicode_literals) propertiesc|j}|jjdtjj j j|j|j|j}|j|_ |jr|j|_ t||j|_|j"r|j"|_|j&r|j$j)d|j*r|j*|_|j/||j|jS)agValidates command line input arguments and passes parent's resources. Args: client: Client for api_utils.py class. alloydb_messages: Messages module for the API client. cluster_ref: parent resource path of the resource being created args: Command line input arguments. Returns: Fully-constructed request to create an AlloyDB user. z)alloydb.projects.locations.clusters.users) projectsId locationsId clustersIdusersIdalloydbsuperuser)useruserIdparent)Userresource_parserCreaterVALUEScoreproject GetOrFailregionclusterusername RelativeNamenamepassword_ParseUserTypetypeuserTypedb_roles databaseRoles superuserappendkeep_extra_roleskeepExtraRoles2AlloydbProjectsLocationsClustersUsersCreateRequest)clientalloydb_messages cluster_refargs user_resourceuser_refs 5lib/googlecloudsdk/command_lib/alloydb/user_helper.pyConstructCreateRequestFromArgsr.s#'')-  # # * *1""''//99++mm +( ,,.- ]]!]]M)*:DIIF- ]]"&--M ^^&&'9: #'#8#8M  L L  ]]  % % ' M cft|||\}}|j||j|S)a)Constructs the request to update an AlloyDB instance. Args: alloydb_messages: Messages module for the API client. user_ref: parent resource path of the resource being updated args: Command line input arguments. Returns: Fully-constructed request to update an AlloyDB user. )r r updateMask)ConstructUserAndMaskFromArgs1AlloydbProjectsLocationsClustersUsersPatchRequestr)r(r,r*r+masks r-ConstructPatchRequestFromArgsr5DsE5$-  K K x4464 L r/cd}d}|j}|j|_d|jvr|j|_||fSd|jvr2|j |_|jr|j|_||fSd|jvrX|jr|j jdn|j jd|j |_||fS|dfS)a"Validates command line arguments and creates the user and field mask. Args: alloydb_messages: Messages module for the API client. user_ref: resource path of the resource being updated args: Command line input arguments. Returns: An AlloyDB user and mask for update. rdatabase_rolesz set-passwordz set-rolesz set-superuserr N) rrr command_pathrr r!r$r%r"r#remove)r(r,r* password_pathdatabase_roles_pathr+s r-r2r2Xs-("'')-,,.-t(((!]]M - ''D%%%"&--M %)%:%:m" - --))) ~~ mm-. mm-."&--M - --  r/c|dk(r |jjjS|dk(r |jjjSy)NBUILT_IN IAM_BASED)rUserTypeValueValuesEnumALLOYDB_BUILT_INALLOYDB_IAM_USER)r( user_types r-rrsE*  8 8 I IIK  8 8 I II r/N) __doc__ __future__rrrgooglecloudsdk.corerr.r5r2rr/r-rGs-N&'*)X('Tr/