Update smsquitto daemon
This commit is contained in:
denis
Executable
+197
@@ -0,0 +1,197 @@
|
||||
#
|
||||
# encoding: utf-8
|
||||
import atexit
|
||||
import sys, os
|
||||
import datetime
|
||||
import json
|
||||
import ssl
|
||||
import subprocess
|
||||
import paho.mqtt.client as mqtt
|
||||
import yaml
|
||||
|
||||
def yaml_load(f):
|
||||
with open(f, 'r') as stream:
|
||||
return yaml.load(stream, Loader=yaml.FullLoader)
|
||||
return {}
|
||||
|
||||
class ClientMQTTBase(mqtt.Client):
|
||||
def __init__(self,
|
||||
host='localhost',
|
||||
port=1883,
|
||||
keepalive=60,
|
||||
username=None,
|
||||
password=None,
|
||||
useSSL=False,
|
||||
ca_cert=None,
|
||||
apikey='0192837495867'
|
||||
):
|
||||
|
||||
super(ClientMQTTBase, self).__init__()
|
||||
self.host = host
|
||||
self.port = port
|
||||
self.keepalive = keepalive
|
||||
if username:
|
||||
self.username_pw_set(username, password)
|
||||
|
||||
if useSSL and ca_cert:
|
||||
self.tls_set(ca_certs=ca_cert, tls_version=ssl.PROTOCOL_TLSv1_2)
|
||||
#self.tls_insecure_set(True)
|
||||
|
||||
self.topic = '%s/emit/' % apikey
|
||||
self.subscriptions = [['%s/pub/#' % apikey, 0],]
|
||||
|
||||
def now(self):
|
||||
return int(datetime.datetime.now().timestamp())
|
||||
|
||||
def sub(self):
|
||||
subs = [(topic, qos) for topic, qos in self.subscriptions]
|
||||
return subs
|
||||
|
||||
def unsub(self):
|
||||
return [topic for topic, _ in self.subscriptions ]
|
||||
|
||||
def startMQTT(self):
|
||||
self.connect(self.host, self.port, self.keepalive)
|
||||
self.subscribe(self.sub())
|
||||
|
||||
def stopMQTT(self):
|
||||
self.unsubscribe(self.unsub())
|
||||
self.disconnect()
|
||||
self.exit()
|
||||
|
||||
def publish_message(self, topic, **payload):
|
||||
try:
|
||||
qos = payload.pop('qos', 0)
|
||||
retain = payload.pop('retain', False)
|
||||
message = json.dumps(payload)
|
||||
self.publish(topic, payload=message.encode('utf_8'), qos=qos, retain=retain)
|
||||
except Exception as e:
|
||||
print("ClientMQTTBase::publish_message error:", e, flush=True)
|
||||
|
||||
def on_connect(self, mqttc, obj, flags, rc): # @UnusedVariable
|
||||
#print("ClientMQTTBase on_connect ")
|
||||
pass
|
||||
|
||||
def on_message(self, mqttc, obj, msg):
|
||||
#print("ClientMQTTBase on_message:", msg.topic, msg.qos, msg.payload)
|
||||
pass
|
||||
|
||||
def on_publish(self, mqttc, obj, mid):
|
||||
#print("on_publish:", mid, obj)
|
||||
pass
|
||||
|
||||
def on_subscribe(self, mqttc, obj, mid, granted_qos): # @UnusedVariable
|
||||
#print("ClientMQTTBase on_subscribe:", mid, granted_qos)
|
||||
pass
|
||||
|
||||
def on_unsubscribe(self, mqttc, userdata, mid): # @UnusedVariable
|
||||
#print("ClientMQTTBase on_unsubscribe:", mid, userdata)
|
||||
pass
|
||||
|
||||
def on_disconnect(self, mqttc, userdata, mid): # @UnusedVariable
|
||||
#print("ClientMQTTBase on_disconnect:", mqttc)
|
||||
pass
|
||||
|
||||
def on_log(self, mqttc, obj, level, string): # @UnusedVariable
|
||||
#print("onlog:", level, string)
|
||||
pass
|
||||
|
||||
|
||||
class SmsQuitto(ClientMQTTBase):
|
||||
|
||||
def __init__(self, **mosquitto):
|
||||
super(SmsQuitto, self).__init__(**mosquitto)
|
||||
|
||||
def exit(self):
|
||||
self.gw_publish('stop', msg='SmsQuitto service stop!')
|
||||
|
||||
def fetchSMS(self, mtype='inbox'):
|
||||
p = subprocess.Popen(["termux-sms-list", "-t", mtype], stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
||||
p.wait()
|
||||
msg = p.stdout.read()
|
||||
payload = json.loads(msg.decode("utf-8"))
|
||||
for p in payload:
|
||||
evt = p.pop('type')
|
||||
dte = p.pop('received')
|
||||
time = datetime.datetime.strptime(dte, '%Y-%m-%d %H:%M').timestamp()
|
||||
self.publish_message(self.topic + evt, time=time, **p)
|
||||
|
||||
def fetchLocation(self):
|
||||
p = subprocess.Popen("termux-location", stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
|
||||
p.wait()
|
||||
msg = p.stdout.read()
|
||||
payload = json.loads(msg.decode("utf-8"))
|
||||
self.publish_message(self.topic + 'location', time=self.now(), **payload)
|
||||
|
||||
def notification(self, title, msg):
|
||||
subprocess.run(["termux-notification", "--sound", '--title', title, '--content', msg])
|
||||
#print("SmsQuitto::notification", title, msg, flush=True)
|
||||
|
||||
def sendSMS(self, mobile, text):
|
||||
subprocess.run(["termux-sms-send", "-n", mobile, text])
|
||||
print("SmsQuitto::sendSMS", mobile, text, flush=True)
|
||||
|
||||
def gw_publish(self, evt, **payload):
|
||||
self.publish_message(self.topic + evt, time=self.now(), **payload)
|
||||
#print("SmsQuitto::gw_publish topic: %s payload: %s"% (self.topic+evt, payload), flush=True)
|
||||
|
||||
def on_connect(self, mqttc, obj, flags, rc): # @UnusedVariable
|
||||
self.gw_publish('start', msg='SmsQuitto service started!')
|
||||
self.notification('SmsQuitto', 'Service started!')
|
||||
#print("SmsQuitto::on_connect SmsQuitto service started!", flush=True)
|
||||
|
||||
def on_disconnect(self, mqttc, obj, flags, rc): # @UnusedVariable
|
||||
self.gw_publish('off', msg='SmsQuitto disconnected!')
|
||||
self.notification('SmsQuitto', 'Service disconnected!')
|
||||
#print("SmsQuitto::on_connect SmsQuitto service disconnected!", flush=True)
|
||||
|
||||
def on_message(self, mqttc, obj, msg): # @UnusedVariable
|
||||
try:
|
||||
topic = msg.topic
|
||||
evt = topic.rsplit('/', 1)[1]
|
||||
payload = json.loads(msg.payload.decode("utf-8"))
|
||||
if evt == 'sms':
|
||||
mobile = payload.get('mobile', None)
|
||||
text = '%s\n%s' % (payload.get('subject', ''), payload.get('msg', ''))
|
||||
if mobile:
|
||||
if not isinstance(mobile, list):
|
||||
mobile = [mobile]
|
||||
for mob in mobile:
|
||||
self.sendSMS(mob, text)
|
||||
|
||||
elif evt == 'location':
|
||||
self.fetchLocation()
|
||||
|
||||
elif evt in ['all','inbox','sent','draft','outbox']:
|
||||
self.fetchSMS(evt)
|
||||
|
||||
elif evt == 'notify':
|
||||
title = payload.get('subject', '')
|
||||
text = payload.get('msg', '')
|
||||
self.notification(title, text)
|
||||
|
||||
elif evt == 'kill-service':
|
||||
self.stopMQTT()
|
||||
|
||||
except Exception as e:
|
||||
self.gw_publish('error', msg=str(e))
|
||||
print("SmsQuitto::on_message error:", e)
|
||||
|
||||
def main(path):
|
||||
try:
|
||||
pf = path[0] if path else os.path.dirname(os.path.abspath(__file__))
|
||||
config = yaml_load(os.path.join(pf, 'smsquitto-conf.yaml'))
|
||||
settings = dict(config.get('settings'))
|
||||
|
||||
daemon = SmsQuitto(**settings)
|
||||
atexit.register(daemon.stopMQTT)
|
||||
daemon.startMQTT()
|
||||
daemon.loop_forever()
|
||||
|
||||
except Exception as e:
|
||||
print("smsquitto.main error", e)
|
||||
|
||||
if __name__ == '__main__':
|
||||
main(sys.argv[1:])
|
||||
|
||||
#
|
||||
Reference in New Issue
Block a user