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
|
import click
import requests
from datetime import datetime
from time import sleep, time
class Alarm(object):
URL = "https://www.oref.org.il/WarningMessages/alert/alerts.json"
HEADERS = {
"Referer": "https://www.oref.org.il/11226-he/pakar.aspx",
"X-Requested-With": "XMLHttpRequest",
"User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_13_6) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/75.0.3770.100 Safari/537.36",
}
def __init__(
self,
delay=1,
routine_delay=60 * 5,
alarm_id=False,
repeat_alarms=False,
quiet=False,
):
self.delay = delay
self.last_routine_delay = routine_delay
self.alarm_id = alarm_id
self.repeat_alarms = repeat_alarms
self.quiet = quiet
self.current_alarms = []
self.last_routine_output = 0
def start(self):
while True:
try:
cities, alarm_id = self.fetch()
self.update(cities, alarm_id)
except Exception as e: # pylint: disable=broad-except
self.output_error(f"Exception: {e}")
finally:
sleep(self.delay)
def fetch(self):
try:
res = requests.get(self.URL, headers=self.HEADERS, timeout=1)
except requests.Timeout as e:
raise Exception("HTTP request timed out") from e
if not res.content:
# empty content means no alarms
return [], None
try:
data = res.json()
alarm_id = data["id"]
cities = data["data"]
return cities, alarm_id
except ValueError as ve:
raise Exception(f"Error parsing JSON: {res.content}") from ve
except KeyError as ke:
raise Exception(f"Missing keys in JSON data: {data}") from ke
def update(self, cities, alarm_id):
if cities:
self.update_alarm(cities, alarm_id)
else:
self.update_routine()
def update_alarm(self, cities, alarm_id):
if self.repeat_alarms or set(cities) != set(self.current_alarms):
self.output_alarms(cities, alarm_id)
self.current_alarms = cities
def update_routine(self):
now = time()
if (
self.current_alarms
or now - self.last_routine_output > self.last_routine_delay
):
self.output_routine()
self.last_routine_output = now
self.current_alarms = []
def output_leading_timestamp(self):
now = datetime.now()
ts = now.strftime("%Y-%m-%d %H:%M:%S")
click.secho(f"{ts} ", nl=False)
def output_error(self, err):
if not self.quiet:
self.output_leading_timestamp()
click.secho(err, fg="yellow", bold=True)
def output_routine(self):
if not self.quiet:
self.output_leading_timestamp()
click.secho("No active alarms", fg="green")
def output_alarms(self, cities, alarm_id):
self.output_leading_timestamp()
cities_str = ", ".join(cities)
click.secho(f"{cities_str} ", fg="red", nl=not self.alarm_id)
if self.alarm_id:
click.secho(f"({alarm_id})")
@click.command()
@click.option("--delay", default=1, help="Polling delay in seconds")
@click.option(
"--routine-delay", default=60 * 5, help="Routine message delay in seconds"
)
@click.option("--alarm-id", is_flag=True, help="Print alarm IDs")
@click.option("--repeat-alarms", is_flag=True, help="Do not suppress ongoing alarms")
@click.option("--quiet", is_flag=True, help="Print only active alarms")
def alarm(**kwargs):
Alarm(**kwargs).start()
if __name__ == "__main__":
alarm()
|
