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
|
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, alarm_id=False, routine_delay=5):
self.delay = delay
self.alarm_id = alarm_id
self.last_routine_delay = routine_delay
self.active_alarm = False
self.last_routine = 0
def start(self):
while True:
self.fetch()
sleep(self.delay)
def fetch(self):
res = requests.get(self.URL, headers=self.HEADERS, timeout=1)
now = datetime.now()
ts = now.strftime("%Y-%m-%d %H:%M:%S")
if res.content:
click.secho(f"{ts} ", nl=False)
try:
data = res.json()
self.active_alarm = True
alarm_id = data["id"]
cities = data["data"]
click.secho(f"{', '.join(cities)} ", fg="red", nl=not self.alarm_id)
if self.alarm_id:
click.secho(f"({alarm_id})")
except:
click.secho(f"Error parsing JSON {res.content}", fg="yellow", bold=True)
else:
if self.active_alarm or (
time() - self.last_routine > self.last_routine_delay
):
click.secho(f"{ts} ", nl=False)
click.secho(f"No active alarms", fg="green")
self.last_routine = time()
self.active_alarm = False
@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 ID")
def alarm(delay=1, routine_delay=60 * 5, alarm_id=False):
Alarm(
delay=delay,
routine_delay=routine_delay,
alarm_id=alarm_id,
).start()
if __name__ == "__main__":
alarm() # pylint: disable=no-value-for-parameter
|
