From 447a807f69de604e385eebeacf904669143d9eea Mon Sep 17 00:00:00 2001
From: Jay Lee <jay0lee@gmail.com>
Date: Wed, 19 Nov 2014 11:12:41 -0500
Subject: [PATCH] 'gam license <sku>' to perform actions for users with given
 skus

---
 gam.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/gam.py b/gam.py
index c7ca2725..32f143eb 100644
--- a/gam.py
+++ b/gam.py
@@ -6387,10 +6387,9 @@ def doPrintCrosDevices():
     cros_attributes.append(crosdevice)
   output_csv(cros_attributes, titles, 'CrOS', todrive)
 
-def doPrintLicenses(return_list=False):
+def doPrintLicenses(return_list=False, skus=None):
   lic = buildGAPIObject(u'licensing')
   products = [u'Google-Apps', u'Google-Drive-storage', u'Google-Coordinate', u'Google-Vault']
-  skus = None
   licenses = []
   lic_attributes = [{}]
   todrive = False
@@ -6963,6 +6962,14 @@ def getUsersToModify(entity_type=None, entity=None, silent=False, return_uids=Fa
       else:
         users.append(member[u'primaryEmail'])
     if not silent: sys.stderr.write(u"done.\r\n")
+  elif entity_type in [u'license', u'licenses']:
+    users = []
+    licenses = doPrintLicenses(return_list=True, skus=entity.split(u','))
+    for row in licenses[1:]: # skip header
+      try:
+        users.append(row[u'userId'])
+      except KeyError:
+        pass
   elif entity_type == u'file':
     users = []
     filename = entity