themo_monitors/run_monitors.py at master · marinetech/themo_monitors · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
from pymongo import MongoClient
import smtplib
from lib.send_sms import *
from monitors.no_communication_with_buoy import *
from monitors.battery_is_low import *
from monitors.buoy_runaway_shallow import *
# from monitors.buoy_runaway import *


# monitors = ["no_communication_with_buoy", "battery_is_low"]
monitors = ["check_when_was_last_ais", "check_runaway"]
# some monitors will use sms alerts
require_sms = ["check_runaway"]


def send_notification(alert_obj):

    receiver = ", ".join(alert_obj["receiver"])
    body = alert_obj["body"]
    subject = alert_obj["subject"]
    sender = 'themo@univ.haifa.ac.il'

    message = "\r\n".join([
        "From: themo@univ.haifa.ac.il",
        "To: %s" % receiver,
        "Subject: Themo alert:  %s" % subject,
        "",
        "%s" % body,
        "",
        "",
        """
        """
    ])

    try:
       smtpObj = smtplib.SMTP('mr1res.haifa.ac.il', 25)
       smtpObj.sendmail(sender, receiver, message)
       print("-I- Successfully sent email")
    except SMTPException:
       print("-E- Error: unable to send email")

def init_db():
    print("-I- Connecting to MOngo DB...")

    global client; client = MongoClient()
    global db; db = client.themo

if __name__ == "__main__":
    init_db()
    for monitor in monitors:
        print("\n-I- running '{0}'".format(monitor))
        alert_obj = globals()[monitor](db)
        if alert_obj:
            print("-I- sending alert mail")
            send_notification(alert_obj)
            if monitor in require_sms:
                print("-I- sending text message")
                sendsms(alert_obj["body"])