deepblue/GoogleDriveManagement
1
0
Fork 0
mirror of https://github.com/GAM-team/GAM.git synced 2025-07-09 06:03:35 +00:00
Code Issues Actions 19 Packages Projects Releases Wiki Activity

Label processing cleanup

Browse Source
This commit is contained in:
Ross Scroggs
2016-03-17 17:47:05 -07:00
parent a3c509ce61
commit 087c6775e3
1 changed files with 13 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/gam.py
View File

@ -4458,14 +4458,16 @@ PROCESS_MESSAGE_FUNCTION_TO_ACTION_MAP = {u'delete': u'deleted',
u'trash': u'trashed', u'untrash': u'untrashed', u'modify': u'modified'} u'trash': u'trashed', u'untrash': u'untrashed', u'modify': u'modified'}
def labelsToLabelIds(gmail, labels): def labelsToLabelIds(gmail, labels):
allLabels = {u'INBOX': u'INBOX', u'SPAM': u'SPAM', u'TRASH': u'TRASH', allLabels = {
u'INBOX': u'INBOX', u'SPAM': u'SPAM', u'TRASH': u'TRASH',
u'UNREAD': u'UNREAD', u'STARRED': u'STARRED', u'IMPORTANT': u'IMPORTANT', u'UNREAD': u'UNREAD', u'STARRED': u'STARRED', u'IMPORTANT': u'IMPORTANT',
u'SENT': u'SENT', u'DRAFT': u'DRAFT', u'SENT': u'SENT', u'DRAFT': u'DRAFT',
u'CATEGORY_PERSONAL': u'CATEGORY_PERSONAL', u'CATEGORY_PERSONAL': u'CATEGORY_PERSONAL',
u'CATEGORY_SOCIAL': u'CATEGORY_SOCIAL', u'CATEGORY_SOCIAL': u'CATEGORY_SOCIAL',
u'CATEGORY_PROMOTIONS': u'CATEGORY_PROMOTIONS', u'CATEGORY_PROMOTIONS': u'CATEGORY_PROMOTIONS',
u'CATEGORY_UPDATES': u'CATEGORY_UPDATES', u'CATEGORY_UPDATES': u'CATEGORY_UPDATES',
u'CATEGORY_FORUMS': u'CATEGORY_FORUMS'} u'CATEGORY_FORUMS': u'CATEGORY_FORUMS',
}
labelIds = list() labelIds = list()
for label in labels: for label in labels:
if label not in allLabels: if label not in allLabels:
@ -4493,7 +4495,7 @@ def labelsToLabelIds(gmail, labels):
parent_label = label[:label.rfind('/')] parent_label = label[:label.rfind('/')]
while True: while True:
if not parent_label in allLabels: if not parent_label in allLabels:
label_result = callGAPI(service=gmail.users().labels(), 'create', label_result = callGAPI(gmail.users().labels(), 'create',
userId='me', body={'name': parent_label}) userId='me', body={'name': parent_label})
allLabels[parent_label] = label_result['id'] allLabels[parent_label] = label_result['id']
if parent_label.find('/') == -1: if parent_label.find('/') == -1:
@ -4577,6 +4579,7 @@ def doProcessMessages(users, function):
def doDeleteLabel(users): def doDeleteLabel(users):
label = sys.argv[5] label = sys.argv[5]
label_name_lower = label.lower()
for user in users: for user in users:
gmail = buildGAPIServiceObject(u'gmail', user) gmail = buildGAPIServiceObject(u'gmail', user)
print u'Getting all labels for %s...' % user print u'Getting all labels for %s...' % user
@ -4596,13 +4599,11 @@ def doDeleteLabel(users):
elif p.match(del_label[u'name']): elif p.match(del_label[u'name']):
del_labels.append(del_label) del_labels.append(del_label)
else: else:
got_label = False
for del_label in labels[u'labels']: for del_label in labels[u'labels']:
if label.lower() == del_label[u'name'].lower(): if label_name_lower == del_label[u'name'].lower():
del_labels.append(del_label) del_labels.append(del_label)
got_label = True
break break
if not got_label: else:
print u' Error: no such label for %s' % user print u' Error: no such label for %s' % user
continue continue
del_me_count = len(del_labels) del_me_count = len(del_labels)
@ -4672,6 +4673,7 @@ def showGmailProfile(users):
def updateLabels(users): def updateLabels(users):
label_name = sys.argv[5] label_name = sys.argv[5]
label_name_lower = label_name.lower()
body = {} body = {}
i = 6 i = 6
while i < len(sys.argv): while i < len(sys.argv):
@ -4701,14 +4703,13 @@ def updateLabels(users):
for user in users: for user in users:
gmail = buildGAPIServiceObject(u'gmail', user) gmail = buildGAPIServiceObject(u'gmail', user)
labels = callGAPI(service=gmail.users().labels(), function=u'list', userId=user, fields=u'labels(id,name)') labels = callGAPI(service=gmail.users().labels(), function=u'list', userId=user, fields=u'labels(id,name)')
label_id = None
for label in labels[u'labels']: for label in labels[u'labels']:
if label[u'name'].lower() == label_name.lower(): if label[u'name'].lower() == label_name_lower:
label_id = label[u'id'] callGAPI(service=gmail.users().labels(), function=u'patch', soft_errors=True,
userId=user, id=label[u'id'], body=body)
break break
if not label_id: else:
print 'Error: user does not have a label named %s' % label_name print 'Error: user does not have a label named %s' % label_name
callGAPI(service=gmail.users().labels(), function=u'patch', soft_errors=True, userId=user, id=label_id, body=body)
def renameLabels(users): def renameLabels(users):
search = u'^Inbox/(.*)$' search = u'^Inbox/(.*)$'