deepblue/GoogleDriveManagement
1
0
Fork 0
mirror of https://github.com/GAM-team/GAM.git synced 2025-07-08 13:43:35 +00:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Clean up parsing transferSecCals, transferDriveFiles

Browse Source
This commit is contained in:
Ross Scroggs
2015-12-22 22:16:33 -08:00
parent 1f1329c536
commit 96dfa52dba
1 changed files with 17 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/gam.py
View File

@ -3933,11 +3933,15 @@ def showDriveFileInfo(users):
def transferSecCals(users): def transferSecCals(users):
target_user = sys.argv[5] target_user = sys.argv[5]
try:
if sys.argv[6].lower() == u'keepuser':
remove_source_user = False
except IndexError:
remove_source_user = True remove_source_user = True
i = 6
while i < len(sys.argv):
if sys.argv[i].lower() == u'keepuser':
remove_source_user = False
i += 1
else:
print u'ERROR: %s is not a valid argument for "gam <users> transfer seccals"' % sys.argv[i]
sys.exit(2)
for user in users: for user in users:
source_cal = buildGAPIServiceObject(u'calendar', user) source_cal = buildGAPIServiceObject(u'calendar', user)
source_calendars = callGAPIpages(service=source_cal.calendarList(), function=u'list', minAccessRole=u'owner', showHidden=True, fields=u'items(id),nextPageToken') source_calendars = callGAPIpages(service=source_cal.calendarList(), function=u'list', minAccessRole=u'owner', showHidden=True, fields=u'items(id),nextPageToken')
@ -3950,11 +3954,14 @@ def transferSecCals(users):
def transferDriveFiles(users): def transferDriveFiles(users):
target_user = sys.argv[5] target_user = sys.argv[5]
remove_source_user = True remove_source_user = True
try: i = 6
if sys.argv[6].lower() == u'keepuser': while i < len(sys.argv):
if sys.argv[i].lower() == u'keepuser':
remove_source_user = False remove_source_user = False
except IndexError: i += 1
pass else:
print u'ERROR: %s is not a valid argument for "gam <users> transfer drive"' % sys.argv[i]
sys.exit(2)
target_drive = buildGAPIServiceObject(u'drive', target_user) target_drive = buildGAPIServiceObject(u'drive', target_user)
target_about = callGAPI(service=target_drive.about(), function=u'get', fields=u'quotaBytesTotal,quotaBytesUsed,rootFolderId') target_about = callGAPI(service=target_drive.about(), function=u'get', fields=u'quotaBytesTotal,quotaBytesUsed,rootFolderId')
target_drive_free = int(target_about[u'quotaBytesTotal']) - int(target_about[u'quotaBytesUsed']) target_drive_free = int(target_about[u'quotaBytesTotal']) - int(target_about[u'quotaBytesUsed'])