dZddlmZddlmZddlmZddlmZddlmZddl m Z Gdd ejZ d Z d Zd Zd Zy)z%Base class for Organization commands.)absolute_import)division)unicode_literals) organizations) exceptions) resourcesc eZdZdfd ZxZS)UnknownOrganizationErrorcPdj|}tt|||y)NzjCannot determine Organization ID from [{0}]. Try `gcloud organizations list` to find your Organization ID.)formatsuperr __init__)self org_argumentmetavarmessage __class__s 9lib/googlecloudsdk/command_lib/organizations/org_utils.pyrz!UnknownOrganizationError.__init__s,O|$  "D27GD)ORGANIZATION_ID)__name__ __module__ __qualname__r __classcell__)rs@rr r sEErr cHd}|j|r|t|dS|S)Nzorganizations/) startswithlen)org_idprefixs rStripOrgPrefixr $s+ & v #f+,  Mrctjj}|jddt |j }|j dd|id}|jS)z0Get the Organization URI for the given resource.cloudresourcemanagerv1NorganizationsIdz"cloudresourcemanager.organizations)params collection)rREGISTRYCloneRegisterApiByNamer nameParseSelfLink)resourceregistryrorg_refs rOrganizationsUriFuncr0,sk    % % '( 3T: (-- (& NN V6  7'    rctj}t|}|jr|j |S|j |S)aGet the Organization object for the provided Organization argument. Returns the organization object for a given organization ID or will search for and return the organization object associated with the given domain name. Args: org_argument: The value of the organization argument. Returns: An object representing an organization, or None if the organization could not be determined. )rClientr isdigitGet GetByDomain)r orgs_clientrs rGetOrganizationr7:sF$$&+ , '& ^^ ??6 ""  " "6 **rctj}t|}|jr|S|j |}|rt|j Sy)aGet the Organization ID for the provided Organization argument. Numeric values will be returned, values like 'organizations/123456789' will return '123456789' and a value like 'example.com' will search for the organization ID associated with that domain. Args: org_argument: The value of the organization argument. Returns: A string containing the numeric organization ID, or None if the organization ID could not be determined. N)rr2r r3r5r*)rr6r org_objects rGetOrganizationIdr:PsP$$&+ , '& ^^ M((0J JOO ,, rN)__doc__ __future__rrr+googlecloudsdk.api_lib.cloudresourcemanagerrgooglecloudsdk.calliopergooglecloudsdk.corerBadArgumentExceptionr r r0r7r:rrrBsD ,&'E.)Ez>>E +,r