healthchecker/notify.py at master · smd00/healthchecker · 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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
import smtplib
from datetime import datetime

'''
Load .env
'''
import os

# from dotenv import load_dotenv
# load_dotenv(dotenv_path='.env')

'''
Load args
'''
import sys
arg = sys.argv[1]
# print(arg)

'''
Read file
'''
logFilePath = arg
logFile = open(logFilePath, 'r+')
logFileRead = logFile.read()

# import subprocess
# subprocess.run(["pm2", "ls"])

print("logFilePath: " + logFilePath)

'''
Email content vars
'''
import socket
hostname = socket.gethostname()

now = datetime.now()
now_string = now.strftime('%d/%m/%Y %H:%M:%S')
service = os.environ.get('SMDHC_CLIENT_NAME', '')
env = os.environ.get('SMDHC_CLIENT_ENV', '')
subject = "Health Check: " + env + " " + service + " (" + hostname + ")"

'''
Email vars (smdhc)
'''
smdhc_smtp_server = os.environ.get('SMDHC_MAIL_HOST', '')
smdhc_port = os.environ.get('SMDHC_MAIL_PORT', '')
smdhc_login = os.environ.get('SMDHC_MAIL_USER', '')
smdhc_password = os.environ.get('SMDHC_MAIL_PWD', '')

smdhc_sender = os.environ.get('SMDHC_MAIL_SENDER', '')
smdhc_receiver = os.environ.get('SMDHC_MAIL_RECEIVER', '')

'''
Email vars (existing env)
'''
smtp_server = os.environ.get('SMTP_ADDRESS', smdhc_smtp_server)
port = os.environ.get('SMTP_PORT', smdhc_port)
login = os.environ.get('SMTP_USERNAME', smdhc_login)
password = os.environ.get('SMTP_PASSWORD', smdhc_password)

sender = os.environ.get('SYSTEM_MAIL_FROM', smdhc_sender)
receiver = os.environ.get('SYSTEM_MAIL_TO', smdhc_receiver)


formatMessage = """\
Subject: {}
To: {}
From: {}


System status summary:

Date: {}
Hostname: {}
Log: {}

{}""".format(subject, receiver, sender, now_string, hostname, logFilePath, logFileRead)

if service == "WALLETD":
    message = formatMessage.decode('ascii', 'ignore').encode('ascii')
elif service == "TBOT":
    message = formatMessage.encode("ascii", "ignore")
else:
    import sys
    reload(sys)
    sys.setdefaultencoding('utf8')
    message = formatMessage.encode('utf-8')

print("smtp_server: " + smtp_server)
# print("port: " + port)
# print("message: " + message)

'''
Send email
'''
try:
    #server = smtplib.SMTP_SSL(smtp_server, port)
    server = smtplib.SMTP(smtp_server, port)
    server.starttls()
    server.login(login, password)
    server.sendmail(sender, receiver, message)
    server.quit()

    print('Email sent to {} at {}'.format(receiver, now_string))
except smtplib.SMTPServerDisconnected:
    print('Failed to connect to the server. Wrong user/password?')
except smtplib.SMTPException as e:
    print('SMTP error occurred: ' + str(e))