This repository has been archived by the owner on Feb 3, 2022. It is now read-only.
-
-
Notifications
You must be signed in to change notification settings - Fork 5
/
main.py
245 lines (206 loc) 路 10.6 KB
/
main.py
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
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
import logging
import os
import os.path
import pushover
import re
import sqlite3
import time
EVENT_LOG_PATH = "/event.log"
EVENT_ID_PATH = "/data/last_event_id.txt"
FONT_COLOR = ["grey", "#ffc311", "#ca414b"]
LOGGING_LEVEL = {
'critical': logging.CRITICAL,
'error': logging.ERROR,
'warn': logging.WARNING,
'warning': logging.WARNING,
'info': logging.INFO,
'debug': logging.DEBUG
}
# Banner
print('''
_, _, , |) _ _ ,_
/ | /|/| / | |/\_----|/\_| | / \_|/\ / \_| |_|/ / |
\/|_/ | |_/\/|_/|_/ |_/ \/|_/ \/ | |/\_/ \/ |_/ |/
|) (| (|
https://github.com/TheCatLady/docker-qnap-pushover
''')
logging_level = LOGGING_LEVEL.get(os.environ['LOG_LEVEL'].lower())
if logging_level is None:
logging_level = logging.WARNING
logging.basicConfig(format="%(levelname)s: %(message)s", level=logging_level)
try:
if "info" in os.environ['NOTIFY_LEVEL'].lower():
NOTIFY_TYPE = 0
elif "warn" in os.environ['NOTIFY_LEVEL'].lower():
NOTIFY_TYPE = 1
else:
NOTIFY_TYPE = 2
except Exception:
NOTIFY_TYPE = 1
finally:
if NOTIFY_TYPE == 0:
logging.info("NOTIFY_LEVEL is set to INFO. " \
"Sending notifications for INFO, WARN, and ERROR system log events (all event types).")
elif NOTIFY_TYPE == 1:
logging.info("NOTIFY_LEVEL is set to WARN. " \
"Sending notifications only for WARN and ERROR system log events; INFO events will not trigger notifications.")
else:
logging.info("NOTIFY_LEVEL is set to ERROR. " \
"Sending notifications only for ERROR system log events; INFO and WARN events will not trigger notifications.")
try:
POLL_INTERVAL = int(os.environ['POLL_INTERVAL'])
except Exception:
POLL_INTERVAL = 10
finally:
logging.info(f"Poll interval is set to {POLL_INTERVAL} seconds.")
try:
INCLUDE = list(filter(str.strip, os.environ['INCLUDE'].lower().split(',')))
except Exception:
INCLUDE = []
finally:
if len(INCLUDE) > 0:
logging.info(f"Only system log events containing the following keywords will trigger notifications: {', '.join(INCLUDE)}")
else:
logging.info("INCLUDE keyword filter is not set.")
try:
EXCLUDE = list(filter(str.strip, os.environ['EXCLUDE'].lower().split(',')))
except Exception:
EXCLUDE = []
finally:
if len(EXCLUDE) > 0:
logging.info(f"System log events containing the following keywords will not trigger notifications: {', '.join(EXCLUDE)}")
else:
logging.info("EXCLUDE keyword filter is not set.")
try:
TESTING_MODE = bool(os.getenv('TESTING_MODE', 'false').lower().strip() in ['true', '1'])
except Exception:
TESTING_MODE = False
finally:
if TESTING_MODE:
logging.warning("Testing mode is enabled. The last 10 system log events will be re-queued for processing on EVERY container start.")
else:
logging.info("Testing mode is not enabled.")
try:
PUSHOVER_TOKEN = os.environ['PUSHOVER_TOKEN'].strip()
PUSHOVER_RECIPIENT = ''.join(os.environ['PUSHOVER_RECIPIENT'].split())
pushover_client = pushover.Client(PUSHOVER_RECIPIENT, api_token=PUSHOVER_TOKEN)
logging.info(f"Using Pushover application API token {PUSHOVER_TOKEN} and recipient user/group key(s) {PUSHOVER_RECIPIENT}.")
if os.path.isfile(EVENT_LOG_PATH):
db_conn = sqlite3.connect(EVENT_LOG_PATH) # event.log is not a text file, but a SQLite database
try:
with open(EVENT_ID_PATH, "r") as f:
last_event_id = int(f.readline()) # read the last-processed event ID from a file if it exists
logging.info(f"Read the last-processed event ID from the data file. ({last_event_id})")
except (FileNotFoundError, ValueError):
last_event_id = 0
if last_event_id == 0: # set the last-processed event ID to the newest event ID if we weren't able to read it from a file
cursor = db_conn.cursor()
cursor.execute(f"SELECT event_id FROM NASLOG_EVENT ORDER BY event_id DESC LIMIT 1;")
last_event_id = cursor.fetchone()[0]
logging.info(f"Setting the last-processed event ID to the newest event ID in the database. ({last_event_id})")
if TESTING_MODE:
last_event_id -= 10 # for testing, re-queue the previous 10 events
logging.info("Testing mode is enabled. Re-queuing last 10 system log events for processing.")
else:
raise Exception(f"Unable to open {EVENT_LOG_PATH}. Was the log file mounted to the container?")
while True:
cursor = db_conn.cursor()
cursor.execute(f"SELECT event_id FROM NASLOG_EVENT ORDER BY event_id DESC LIMIT 1;")
newest_event_id = cursor.fetchone()[0]
if newest_event_id > last_event_id:
new_event_count = newest_event_id - last_event_id
logging.info(f"{new_event_count} new system log events detected in {EVENT_LOG_PATH}.")
try:
for i in range(new_event_count):
cursor = db_conn.cursor()
cursor.execute(f"SELECT event_type, event_desc, event_date, event_time, event_user, event_ip FROM NASLOG_EVENT WHERE event_id={last_event_id + 1 + i};")
event = cursor.fetchone()
event_type = event[0]
event_desc = event[1].strip()
event_datetime = f"{event[2]} {event[3]}"
event_user = event[4]
event_ip = event[5]
if event_type >= NOTIFY_TYPE:
hasIncluded = False
hasExcluded = False
if len(INCLUDE) == 0:
hasIncluded = True
else:
for keyword in INCLUDE:
if keyword in event_desc.lower():
hasIncluded = True
break
if hasIncluded == False:
logging.debug(f"Skipping system log event because it does not contain at least one INCLUDE keyword: {INCLUDE}")
for keyword in EXCLUDE:
if keyword in event_desc.lower():
hasExcluded = True
logging.debug(f"Skipping system log event because it contains an EXCLUDE keyword: {keyword}")
break
if hasIncluded and not hasExcluded:
if ']' in event_desc:
title = event_desc[1 : event_desc.find(']')]
message = event_desc[event_desc.find(']') + 2 :].rstrip(".,;")
else:
title = "System"
message = event_desc
if ". " in message:
first_line = True
prev = ""
quotes = 0
for s in message.split('. '):
quotes += len(re.findall('"', s))
if quotes % 2 == 0:
if first_line:
message = f"<font color=\"{FONT_COLOR[event_type]}\"><b>{prev}{s}</b></font><small>";
first_line = False
else:
message += f"<br/>{prev}{s}"
prev = ""
quotes = 0
else:
prev += f"{s}. "
else:
message = f"<font color=\"{FONT_COLOR[event_type]}\"><b>{message}</b></font><small>"
message_details = ""
if event_user != "System":
message_details += f"<b>{event_user}</b><small><br/>User"
if event_ip != "127.0.0.1":
if message_details != "":
message_details += "<br/> \n</small>"
message_details += f"<b>{event_ip}</b><small><br/>Source IP"
if message_details != "":
message += f"<br/> \n{message_details}</small></small>"
else:
message += "</small>"
timestamp = int(time.mktime(time.strptime(event_datetime, "%Y-%m-%d %H:%M:%S")))
priority = event_type - 1
pushover_answer = pushover_client.send_message(message, html=1, priority=priority, timestamp=timestamp, title=title).answer
logging.debug(f"Pushover API response: {pushover_answer}")
if pushover_answer["status"] != 1:
time.sleep(5) # wait an extra 5 seconds
raise Exception("Unable to connect to Pushover API.")
last_event_id = newest_event_id
logging.info(f"Successfully processed {new_event_count} system log events.")
except pushover.RequestError as e:
i += 1
last_event_id += i
logging.warning(f"Message rejected by Pushover API because {', '.join(e.errors)}. " \
f"Only {i} of {new_event_count} new system log events were processed successfully. " \
f"Skipping the current event and re-queuing the remaining {new_event_count - i} unprocessed events.")
except Exception as e:
last_event_id += i
logging.warning(f"{e}. " \
f"Only {i} of {new_event_count} new system log events were processed successfully. " \
f"Re-queuing the remaining {new_event_count - i} unprocessed events.")
else:
logging.info(f"No new system log events detected in {EVENT_LOG_PATH}.")
with open(EVENT_ID_PATH, "w+") as f:
f.write(str(last_event_id))
f.truncate()
logging.info(f"Wrote the last-processed event ID to the data file. ({last_event_id})")
time.sleep(POLL_INTERVAL)
except (pushover.InitError, pushover.UserError):
logging.critical(f"Pushover application API token and recipient must be set.")
except Exception as e:
logging.critical(f"{e}.")