Add script

2014-07-30 13:45:50 +02:00 · 2014-07-30 13:45:50 +02:00 · 4d131cdb53
commit 4d131cdb53
parent 353c2122a9
1 changed files with 156 additions and 0 deletions
--- a/emails_sms_free.py
+++ b/emails_sms_free.py
@ -0,0 +1,156 @@
+#!/usr/bin/env python3
+
+import email
+import imaplib
+import json
+import os.path
+import requests
+import sys
+
+msg_ids = {}
+
+
+def get_subject(subject_header):
+    tmp = email.header.decode_header(subject_header)[0]
+    if tmp[1] is None:
+        return str(tmp[0])
+    else:
+        return tmp[0].decode(tmp[1])
+
+
+def send(url, msg, i=0):
+    full_url = url.replace('{$msg}', msg)
+    r = requests.get(full_url, verify=False)
+    if r.status_code == 200:
+        return True
+    elif r.status_code == 400:
+        print('Un paramètre obligatoire est manquant.')
+        return False
+    elif r.status_code == 402:
+        if i < 3:
+            print('Trop de SMS ont été envoyés en trop peu de temps, ' +
+                'le script réessayera dans 30 secondes.')
+            time.sleep(30)
+            send(url, msg, i+1)
+        else:
+            print('Impossible d\'envoyer le message dans la dernière minute ' +
+                  'et demie.')
+            return False
+    elif r.status_code == 403:
+        print('Identifiants incorrects ou service non activé.')
+        return False
+    elif r.status_code == 500:
+        print('Erreur côté serveur.')
+        return False
+
+
+def get_emails(imap_server, imap_user, imap_password, inbox):
+    global msg_ids
+
+    print('Connecting to '+imap_server+'… ', end='')
+    conn = imaplib.IMAP4_SSL(imap_server)
+    print('Connected')
+    to_send = []
+
+    try:
+        print('Logging as '+imap_user+'… ', end='')
+        retcode, capabilities = conn.login(imap_user, imap_password)
+    except:
+        print('Failed')
+        print(sys.exc_info()[1])
+        sys.exit(1)
+    print('Logged in')
+
+    try:
+        conn.select(inbox, readonly=True)
+        typ, tmp_msg_ids = conn.uid('search', None, 'ALL')
+        tmp_msg_ids = [i.decode('utf-8')
+                       for i in tmp_msg_ids[0].split()]
+        if imap_server in msg_ids:
+            diff_msg_ids = [i
+                            for i in tmp_msg_ids
+                            if i not in msg_ids[imap_server]]
+        else:
+            diff_msg_ids = tmp_msg_ids
+        msg_ids[imap_server] = tmp_msg_ids
+        if len(diff_msg_ids) == 0:
+            print("\tNo new emails")
+        else:
+            for i in diff_msg_ids:
+                typ, msg_data = conn.uid('fetch', i, '(RFC822)')
+                msg_parsed = email.message_from_bytes(msg_data[0][1])
+                for part in msg_parsed.walk():
+                    if part.get_content_charset() is None:
+                        # We cannot know the character set, so return decoded "something"
+                        text = part.get_payload(decode=True)
+                        continue
+
+                    charset = part.get_content_charset()
+
+                    if part.get_content_type() == 'text/plain':
+                        text = part.get_payload(decode=True).decode(str(charset))
+                        break
+                subject = get_subject(msg_parsed['Subject'])
+                print("\tNew email from "+msg_parsed['From'] +
+                      " : "+subject)
+                to_send.append({'server': imap_server,
+                                'id': i,
+                                'from': msg_parsed['From'],
+                                'subject': subject,
+                                'content': text})
+        return to_send
+    finally:
+        try:
+            conn.close()
+        except:
+            pass
+        conn.logout()
+
+
+if __name__ == '__main__':
+    # EDIT BELOW ACCORDING TO YOUR NEEDS
+    imap_servers = [{'server': 'SERVER',
+                     'login': 'LOGIN',
+                     'password': 'PASS',
+                     'inbox': 'INBOX'}
+                   ]
+    save_path = os.path.expanduser('~/.mails_free_api.json')
+    debug = False
+    url = "https://smsapi.free-mobile.fr/sendmsg?user={$user}&pass={$pass}&msg={$msg}"
+    user = 'IDENT'
+    password = 'PASS'
+    # YOU SHOULD NOT HAVE TO EDIT BELOW
+
+    url = url.replace('{$user}', user)
+    url = url.replace('{$pass}', password)
+
+    if os.path.isfile(save_path):
+        with open(save_path, 'r') as fh:
+            msg_ids = json.load(fh)
+
+    to_send = []
+    for imap_server in imap_servers:
+        to_send.extend(get_emails(imap_server['server'],
+                                  imap_server['login'],
+                                  imap_server['password'],
+                                  imap_server['inbox']))
+
+    if debug:
+        print("\nNew emails to send via SMS:")
+        print("\t", to_send)
+
+    i = 1
+    for data in to_send:
+        msg = ('New email from '+data['from']+'\n' +
+               data['subject']+'\n' +
+               str(data['content'][:700])+'...')
+        if send(url, msg):
+            print('Sent '+str(i)+'/'+str(len(to_send)))
+        else:
+            msg_ids[data['imap_server']].remove(data['id'])
+            print('Email '+str(i)+'/'+str(len(to_send))+' : Unable to send the text ' +
+                  'message, remove this email from parsed list.')
+        i += 1
+
+    with open(save_path, 'w') as fh:
+        json.dump(msg_ids, fh)