Initial commit

1 files changed, 92 insertions, 0 deletions

diff --git a/git-gmail b/git-gmail
new file mode 100755
index 0000000..29bc29d
--- /dev/null
+++ b/git-gmail
@@ -0,0 +1,92 @@
+#!/usr/bin/env python
+#
+# Grab email from a gmail label and write it in a local dir
+#
+# Author: tomeu@sugarlabs.org
+# Based on code by: ted@tedkulp.com
+# Based on code by: follower@myrealbox.com
+#
+# License: GPL 2.0
+#
+import os
+import sys
+from getpass import getpass
+import re
+import email
+import subprocess
+
+import libgmail
+
+PATCHES_LABEL = 'Patches'
+PATCHES_DIR = os.path.expanduser('~/patches')
+
+def _may_be_patch(message):
+    found_minuses = False
+    for line in message.source.split('\n'):
+        if found_minuses and line.startswith('+++'):
+            return True
+        if line.startswith('---'):
+            found_minuses = True
+        else:
+            found_minuses = False
+    return False 
+
+
+def _write_patch(message):
+    if not os.path.exists(PATCHES_DIR):
+        os.mkdir(PATCHES_DIR)
+
+    full_message = str(message.source.replace('\r', '').strip())
+    email_message = email.message_from_string(full_message)
+    subject = email_message['Subject'].replace('\n\t', ' ')
+
+    # TODO: Probably should remove all tags between []
+    subject = subject.replace('[Sugar-devel] ', '')
+    subject = subject.replace('[PATCH] ', '')
+
+    file_path = subject.replace(' ', '-')
+    file_path += '.patch'
+    file_path = os.path.join(PATCHES_DIR, file_path)
+
+    print "Writing patch: %s" % file_path
+
+    patch_file = open(file_path, 'w+')
+    patch_file.write(full_message)
+    patch_file.close()
+
+if __name__ == "__main__":
+
+    call_args = ['git', 'config', 'user.gmail']
+    pipe = subprocess.Popen(call_args, stdout=subprocess.PIPE)
+    user_name = pipe.communicate()[0].strip()
+    if not user_name:
+        print 'Need a user name configured in .gitconfig: ' \
+              'git config --global user.gmail <name>@gmail.com'
+        exit(0)
+
+    password = getpass('Password: ')
+    ga = libgmail.GmailAccount(user_name, password)
+     
+    print "Please wait, logging in as %s..." % user_name
+     
+    try:
+        ga.login()
+    except libgmail.GmailLoginFailure:
+        print "Login failed. (Wrong username/password?)"
+        sys.exit(2)
+    else:
+        print "Log in successful"
+     
+    query = 'label: %s' % PATCHES_LABEL
+    if len(sys.argv) > 1:
+        query += ' %s' % sys.argv[1]
+    result = ga.getMessagesByQuery(query, True)
+     
+    if len(result):
+        for thread in reversed(result):
+            for msg in reversed(thread):
+                if _may_be_patch(msg):
+                    _write_patch(msg)
+    else:
+        print "No threads found in `%s`." % new_label
+