-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
84 lines (73 loc) · 2.8 KB
/
server.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
import worker
import requests
import time
import configparser
import sys
import json
ERROR_WAIT_TIME = 500 # seconds
COOLDOWN_WAIT_TIME = 250 # seconds
def read_config(file_path: str):
"""
Reads a config file and returns a dictionary of the config
:param: file_path: str
:return: dict
"""
config = configparser.ConfigParser()
config.read(file_path)
return config
def send_heartbeat(status: str):
"""
Sends a heartbeat to the server
:param: status: str
"""
config = read_config("config.ini")
base_url = config.get("queue", "base_url")
password = config.get("queue", "worker_password")
name = config.get("queue", "worker_name")
headers = {'X-AUTHENTICATION': password}
requests.post(f"{base_url}/api/worker/heartbeat", headers=headers, data={"status": status, "name": name})
def wait_and_check_for_quit(seconds: int):
try:
for _ in range(seconds):
time.sleep(1)
except KeyboardInterrupt:
print("Keyboard interrupt detected. Sending offline heartbeat...")
send_heartbeat("Offline")
exit()
def main():
if len(sys.argv) > 1:
if sys.argv[1] == "--offline_msg":
send_heartbeat(sys.argv[2])
exit()
config = read_config("config.ini")
base_url = config.get("queue", "base_url")
password = config.get("queue", "worker_password")
send_heartbeat("Offline")
try:
while True:
headers = {'X-AUTHENTICATION': password}
next_video = requests.get(f"{base_url}/api/worker/next", headers=headers)
if next_video.status_code == 200:
print("Found video to archive. Starting...")
next_video_data = json.loads(next_video.text)
send_heartbeat("Archiving " + next_video_data["next_video"])
mode = next_video_data["mode"]
worker.execute_server_worker(next_video_data["next_video"], mode)
elif next_video.status_code == 401:
print("Invalid credentials. The password may be incorrect")
send_heartbeat("Invalid credentials. The password may be incorrect")
wait_and_check_for_quit(ERROR_WAIT_TIME)
else:
print("No videos to archive at this time. Cooling down...")
send_heartbeat("Idle. Waiting for work...")
wait_and_check_for_quit(COOLDOWN_WAIT_TIME)
except Exception as e:
if str(e) == "KeyboardInterrupt":
print("Keyboard interrupt detected. Sending offline heartbeat...")
send_heartbeat("Offline")
else:
print("An error occurred. Sending offline heartbeat...")
send_heartbeat("Offline - An error occured " + str(e))
print(e)
if __name__ == '__main__':
main()