#!/usr/bin/env python3

"""

# Purpose: Get file counts/size for Team Drives

# Customize: DOMAIN_LIST

# Python: Use python or python3 below as appropriate to your system; verify that you have version 3

# $ python -V or python3 -V

# Python 3.x.y

# Usage:

# 1: If you want to include all Team Drives, do this step and then skip to step 4, otherwise start at step 2.

# $ gam redirect csv ./TeamDrives.csv print teamdrives fields id,name

# 2: If you want Team Drives for a specific set of organizers, replace <UserTypeEntity> with your user selection in the command below

# $ gam redirect csv ./AllTeamDrives.csv <UserTypeEntity> print teamdrives role organizer fields id,name

# 3: Delete duplicate Team Drives (some may have multiple organizers). Make sure that ID_FIELD = 'id' in DeleteDuplicateRows.py

# $ python3 DeleteDuplicateRows.py ./AllTeamDrives.csv ./TeamDrives.csv

# 4: Customize GetTeamDriveOrganizers.py for this task:

# Set DOMAIN_LIST as required

# Set ONE_ORGANIZER = True

# Set SHOW_GROUP_ORGANIZERS = False

# Set SHOW_USER_ORGANIZERS = True

# 5: Get ACLs for all Team Drives

# $ gam redirect csv ./TeamDriveACLs.csv multiprocess csv ./TeamDrives.csv gam print drivefileacls "~id" fields emailaddress,role,type

# 6: From that list of ACLs, output a CSV file with headers "id,name,organizer"

# that shows an organizer/fileOrganizer for each Team Drive

# $ python3 GetTeamDriveOrganizers.py TeamDriveACLs.csv TeamDrives.csv TeamDriveOrganizers.csv

# 7: From that list of organizers, get the file counts/size for all Team Drives that have an organizer (matchfield organizer "^.+$")

# If you don't need size informatiom

# $ gam config csv_input_row_filter "organizers:regex:^.+$" redirect csv ./TeamDriveFileCounts.csv multiprocess csv ./TeamDriveOrganizers.csv gam user "~organizers" print filecounts select teamdriveid "~id"

# If you need size information and have GAM 6.13.04 or later:

# $ gam config csv_input_row_filter "organizers:regex:^.+$" redirect csv ./TeamDriveFileCounts.csv multiprocess csv ./TeamDriveOrganizers.csv gam user "~organizers" print filecounts select teamdriveid "~id" showsize

# If you need size information and have GAM 6.13.03 or earlier

# $ gam config csv_input_row_filter "organizers:regex:^.+$" redirect csv ./TeamDriveFileCounts.csv multiprocess csv ./TeamDriveOrganizers.csv gam user "~organizers" print filelist select teamdriveid "~id" countsonly showsize showsource

# 8: You can identify all Team Drives without an organizer

# $ gam csv ./TeamDriveOrganizers.csv skipfield organizers "^.+$" gam info teamdrive teamdriveid "~id"

"""

import csv

import re

import sys

# If you want to limit organizers to a specific list of domains, use the list below, e.g., DOMAIN_LIST = ['domain.com',] DOMAIN_LIST = ['domain1.com', 'domain2.com',]

DOMAIN_LIST = []

QUOTE_CHAR = '"' # Adjust as needed

LINE_TERMINATOR = '\n' # On Windows, you probably want '\r\n'

PERMISSIONS_N_ROLE = re.compile(r"permissions.(\d+).role")

if (len(sys.argv) > 3) and (sys.argv[3] != '-'):

outputFile = open(sys.argv[3], 'w', encoding='utf-8', newline='')

else:

outputFile = sys.stdout

outputCSV = csv.DictWriter(outputFile, ['id', 'name', 'organizer'], lineterminator=LINE_TERMINATOR, quotechar=QUOTE_CHAR)

outputCSV.writeheader()

teamDriveNames = {}

inputFile = open(sys.argv[2], 'r', encoding='utf-8')

for row in csv.DictReader(inputFile, quotechar=QUOTE_CHAR):

teamDriveNames[row['id']] = row['name']

inputFile.close()

if (len(sys.argv) > 1) and (sys.argv[1] != '-'):

inputFile = open(sys.argv[1], 'r', encoding='utf-8')

else:

inputFile = sys.stdin

for row in csv.DictReader(inputFile, quotechar=QUOTE_CHAR):

organizer = ''

for k, v in iter(row.items()):

mg = PERMISSIONS_N_ROLE.match(k)

if mg and v in ['organizer', 'fileOrganizer']:

permissions_N = mg.group(1)

if row[f'permissions.{permissions_N}.type'] != 'user':

continue

emailAddress = row[f'permissions.{permissions_N}.emailAddress'].lower()

if DOMAIN_LIST:

domain = emailAddress[emailAddress.find('@')+1:]

if domain not in DOMAIN_LIST:

continue

organizer = emailAddress

break

outputCSV.writerow({'id': row['id'],

'name': teamDriveNames.get(row['id'], row['id']),

'organizer': organizer})

if inputFile != sys.stdin:

inputFile.close()

if outputFile != sys.stdout:

outputFile.close()