Import

app/sms.py

@@ -0,0 +1,180 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import sys, os
+import xmltodict
+import requests
+import json
+import time
+from pushbullet import Pushbullet
+
+pb = Pushbullet(os.environ['PB_APIKEY'])
+
+pin = os.environ['PIN']
+
+PIN_ENTER_TEMPLATE = '''<request>
+    <OperateType>0</OperateType>
+    <CurrentPin>''' + PIN + '''</CurrentPin>
+    <NewPin></NewPin>
+    <PukCode></PukCode>
+    </request>'''
+
+SMS_LIST_TEMPLATE = '''<request>
+    <PageIndex>1</PageIndex>
+    <ReadCount>20</ReadCount>
+    <BoxType>1</BoxType>
+    <SortType>0</SortType>
+    <Ascending>0</Ascending>
+    <UnreadPreferred>0</UnreadPreferred>
+    </request>'''
+
+SMS_DEL_TEMPLATE = '<request><Index>{index}</Index></request>'
+
+"""SMS_SEND_TEMPLATE = '''<request>
+    <Index>-1</Index>
+    <Phones><Phone>{phone}</Phone></Phones>
+    <Sca></Sca>
+    <Content>{content}</Content>
+    <Length>{length}</Length>
+    <Reserved>1</Reserved>
+    <Date>{timestamp}</Date>
+    </request>'''
+    """
+    
+__location__ = os.path.realpath(os.path.join(os.getcwd(), os.path.dirname(__file__)))
+
+def isHilink(device_ip):
+    try:
+        r = requests.get(url='http://' + device_ip + '/api/device/information', timeout=(2.0,2.0))
+    except requests.exceptions.RequestException as e:
+        return False;
+
+    if r.status_code != 200:
+        return False
+    return True
+    
+def getHeaders(device_ip):
+    token = None
+    sessionID = None
+    try:
+        r = requests.get(url='http://' + device_ip + '/api/webserver/SesTokInfo')
+    except requests.exceptions.RequestException as e:
+        return (token, sessionID)
+    try:        
+        d = xmltodict.parse(r.text, xml_attribs=True)
+        if 'response' in d and 'TokInfo' in d['response']:
+            token = d['response']['TokInfo']
+        d = xmltodict.parse(r.text, xml_attribs=True)
+        if 'response' in d and 'SesInfo' in d['response']:
+            sessionID = d['response']['SesInfo']
+        headers = {'__RequestVerificationToken': token, 'Cookie': sessionID}
+    except:
+        pass
+    return headers
+    
+
+def getSMS(device_ip, headers):
+    r = requests.post(url = 'http://' + device_ip + '/api/sms/sms-list', data = SMS_LIST_TEMPLATE, headers = headers)
+    d = xmltodict.parse(r.text, xml_attribs=True)
+    numMessages = int(d['response']['Count'])
+    messagesR = d['response']['Messages']['Message']
+    if numMessages == 1:
+        temp = messagesR
+        messagesR = [temp]
+    messages = getContent(messagesR, numMessages)
+    return messages, messagesR
+
+def statusPin(device_ip, headers):
+    r = requests.get(url = 'http://' + device_ip + '/api/pin/status', headers = headers)
+    d = xmltodict.parse(r.text, xml_attribs=True)
+    state = int(d['response']['SimState'])
+    return state
+
+def enterPin(pin, device_ip, headers):
+    r = requests.post(url = 'http://' + device_ip + '/api/pin/operate', data = PIN_ENTER_TEMPLATE, headers = headers)
+    d = xmltodict.parse(r.text, xml_attribs=True)
+    print(d['response'])
+
+def getContent(data, numMessages):
+    messages = []
+    for i in range(numMessages):
+        message = data[i]
+        number = message['Phone']
+        content = message['Content']
+        date = message['Date']
+        messages.append('Message from ' + number + ' recieved ' + date + ' : ' + str(content))
+    return messages
+
+def delMessage(device_ip, headers, ind):
+    r = requests.post(url = 'http://' + device_ip + '/api/sms/delete-sms', data = SMS_DEL_TEMPLATE.format(index=ind), headers = headers)
+    d = xmltodict.parse(r.text, xml_attribs=True)
+    print(d['response'])
+
+def getUnread(device_ip, headers):
+    r = requests.get(url = 'http://' + device_ip + '/api/monitoring/check-notifications', headers = headers)
+    d = xmltodict.parse(r.text, xml_attribs=True)
+    unread = int(d['response']['UnreadMessage'])
+    return unread
+
+if __name__ == "__main__":
+    while True:
+        device_ip = os.environ['GW_IP']
+        if not isHilink(device_ip):
+            print("Can't find a Huawei HiLink device on " + device_ip + ", exit")
+            print('')
+            sys.exit(-1)
+    
+        headers = getHeaders(device_ip)
+        # Require to enter the PIN to unlock device
+        if statusPin(device_ip, headers) == 260:
+            print('# Configure PIN')
+            enterPin(os.environ['PIN'], device_ip, headers)
+        
+        unread = getUnread(device_ip, headers)
+        print('# Unread message(s) : %d' % unread)
+        if unread != 0:
+            messages, messagesR = getSMS(device_ip, headers)
+    
+            #Read
+            if not os.path.exists(os.path.join(__location__,'data',)):
+                os.makedirs(os.path.join(__location__,'data'))
+            try:
+                f1 = open(os.path.join(__location__,'data',"sms.json"), 'r')
+                data = json.load(f1)
+                f1.close()
+            except:
+                data = []
+        
+            f3 = open(os.path.join(__location__,'data',"sms.txt"), 'a')
+    
+            # Ok, fonctionnel
+            for i in range(len(messages)):
+                # Json
+                print('# Log JSON')
+                data.append(messagesR[i])
+    
+                # Text
+                f3.write(messages[i] + '\n')
+                # Pushbullet
+                print('# Notif pushbullet')
+                pb.push_note("SMS", "From: %s\nDate: %s\n%s" % (messagesR[i]['Phone'], messagesR[i]['Date'], messagesR[i]['Content']))
+    
+                # Log HTTP
+                print('# Notif HTTP')
+                datam = {'sender' : messagesR[i]['Phone'], 'date': messagesR[i]['Date'], 'content': messagesR[i]['Content']}
+                r = requests.post(url="https://sms.cabillot.eu", headers={'Content-Type': 'application/json' }, json=datam)
+                #print(r.status_code)
+    
+            #Save
+            f2 = open(os.path.join(__location__,'data',"sms.json"), 'w')
+            json.dump(data, f2)
+            f2.close()
+    
+            f3.close()
+    
+            #delete from device
+            for i in range(len(messagesR)):
+                headers = getHeaders(device_ip)
+                delMessage(device_ip, headers, messagesR[i]['Index'])
+
+        time.sleep(30)