Listing and getting policies

  """Sample script to demonstrate the use of the List method in the Policy API."""

  from collections.abc import Mapping, Sequence
  import json
  import pprint
  import time
  from typing import Any
  import urllib.request
  from absl import app
  from absl import flags
  import google.auth.transport.requests
  from google.oauth2 import service_account

  SCOPES = ['https://www.googleapis.com/auth/cloud-identity.policies']
  BASE_URL = 'https://cloudidentity.googleapis.com/'
  VERSIONED_BASE_URL = f'{BASE_URL}v1/'

  _SA_FILE = flags.DEFINE_string(
      name='sa_file',
      default=None,
      help='Service account credentials file path',
      required=True,
  )

  _ADMIN_EMAIL = flags.DEFINE_string(
      name='admin_email',
      default=None,
      help='Administrator email to call as',
      required=True,
  )

  _PAGE_SIZE = flags.DEFINE_integer(
      name='page_size',
      default=50,
      help='Page size for the List API request',
      required=False,
  )

  # To list all policies, set FILTER to '';
  # To list policies for a specific customer, set FILTER to
  # 'customer == "customers/{obfuscated_target_customer_id}"';
  # To list policies for a specific Application, such as Gmail, set FILTER to
  # 'setting.type.matches("gmail.*")';
  # To list policies for a specific Setting, such as service_status, set FILTER to
  # 'setting.type.matches(".*service_status")'.
  _LIST_FILTER = flags.DEFINE_string(
      name='list_filter',
      default='',
      help='Filter for the List API request',
      required=False,
  )


  def create_delegated_credentials(
      sa_file: str, user_email: str
  ) -> service_account.Credentials:
    """Creates delegated credentials for the user.

    Args:
      sa_file: The service account credentials file path.
      user_email: The administrator email to call as.

    Returns:
      The delegated credentials for the user.
    """
    credentials = service_account.Credentials.from_service_account_file(
        sa_file, scopes=SCOPES
    )
    delegated_credentials = credentials.with_subject(user_email)
    return delegated_credentials


  def build_list_policies_request(
      page_size: int, list_filter: str, page_token: str, access_token: str
  ) -> urllib.request.Request:
    """Builds the request for the List Policies API.

    Args:
      page_size: The page size for the request.
      list_filter: The filter for the request.
      page_token: The page token for the request.
      access_token: The access token for the API.

    Returns:
      The request for the List Policies API.
    """
    list_url = (
        f'{VERSIONED_BASE_URL}policies?page_size={page_size}'
        f'&filter={list_filter}&page_token={page_token}'
    )
    request = urllib.request.Request(list_url)
    request.add_header('Authorization', 'Bearer ' + access_token)
    return request


  def call_list_policies_api(
      request: urllib.request.Request,
  ) -> Mapping[str, Any]:
    """Calls the List Policies API.

    Args:
      request: The request for the List Policies API.

    Returns:
      The response for the List Policies API.
    """
    content = urllib.request.urlopen(request).read()
    return json.loads(content)


  def call_list_policies_api_till_last_page(
      access_token: str, page_size: int, list_filter: str
  ) -> None:
    """Calls the List Policies API until the last page.

    Args:
      access_token: The access token for the API.
      page_size: The page size for the request.
      list_filter: The filter for the request.
    """
    page_token = ''
    # Paginate until the last page.
    while True:
      list_policies_request = build_list_policies_request(
          page_size, list_filter, page_token, access_token
      )
      list_policies_response = call_list_policies_api(list_policies_request)
      print_list_policies_response(list_policies_response)
      if (
          'nextPageToken' not in list_policies_response
          or not list_policies_response['nextPageToken']
      ):
        print('This is the last page.')
        break
      page_token = list_policies_response['nextPageToken']
      time.sleep(1)


  def print_list_policies_response(response: Mapping[str, Any]) -> None:
    pp = pprint.PrettyPrinter(indent=4)

    if 'policies' in response:
      for policy in response['policies']:
        pp.pprint(policy)
    if 'nextPageToken' in response:
      print('Next page token: ' + response['nextPageToken'])


  def main(argv: Sequence[str]):
    if len(argv) > 4:
      raise app.UsageError('Too many command-line arguments.')

    dc = create_delegated_credentials(_SA_FILE.value, _ADMIN_EMAIL.value)
    dc.refresh(google.auth.transport.requests.Request())
    call_list_policies_api_till_last_page(
        dc.token,
        _PAGE_SIZE.value,
        _LIST_FILTER.value,
    )


  if __name__ == '__main__':
    app.run(main)

  """Sample script to demonstrate the use of the get method in the Policy API."""

  from collections.abc import Sequence
  import json
  import pprint
  import urllib.request
  from absl import app
  from absl import flags
  import google.auth.transport.requests
  from google.oauth2 import service_account

  SCOPES = ['https://www.googleapis.com/auth/cloud-identity.policies']
  BASE_URL = 'https://cloudidentity.googleapis.com/'
  VERSIONED_BASE_URL = f'{BASE_URL}v1/'

  _SA_FILE = flags.DEFINE_string(
      name='sa_file',
      default=None,
      help='Service account credentials file path',
      required=True,
  )

  _ADMIN_EMAIL = flags.DEFINE_string(
      name='admin_email',
      default=None,
      help='Administrator email to call as',
      required=True,
  )

  _POLICY_NAME = flags.DEFINE_string(
      name='policy_name',
      default=None,
      help='Policy name of the policy to get',
      required=True,
  )


  def create_delegated_credentials(
      sa_file: str, user_email: str
  ) -> service_account.Credentials:
    """Creates delegated credentials for the user.

    Args:
      sa_file: The service account credentials file path.
      user_email: The administrator email to call as.

    Returns:
      The delegated credentials for the user.
    """
    credentials = service_account.Credentials.from_service_account_file(
        sa_file, scopes=SCOPES
    )
    delegated_credentials = credentials.with_subject(user_email)
    return delegated_credentials


  def build_get_policy_request(
      policy_name: str, access_token: str
  ) -> urllib.request.Request:
    """Builds the request for the Get Policy API.

    Args:
      policy_name: The policy name to get.
      access_token: The access token for the API.

    Returns:
      The request for the Get Policy API.
    """
    list_url = f'{VERSIONED_BASE_URL}{policy_name}'
    request = urllib.request.Request(list_url)
    request.add_header('Authorization', 'Bearer ' + access_token)
    return request


  def call_get_policy_api(access_token: str, policy_name: str) -> None:
    """Calls the Get Policy API.

    Args:
      access_token: The access token for the API.
      policy_name: The policy name to get.
    """
    request = build_get_policy_request(policy_name, access_token)
    content = urllib.request.urlopen(request).read()
    response = json.loads(content)
    print_get_policy_response(response)


  def print_get_policy_response(response: str) -> None:
    pp = pprint.PrettyPrinter(indent=4)
    pp.pprint(response)


  def main(argv: Sequence[str]):
    if len(argv) > 3:
      raise app.UsageError('Too many command-line arguments.')

    dc = create_delegated_credentials(_SA_FILE.value, _ADMIN_EMAIL.value)
    dc.refresh(google.auth.transport.requests.Request())
    call_get_policy_api(dc.token, _POLICY_NAME.value)


  if __name__ == '__main__':
    app.run(main)