-
Notifications
You must be signed in to change notification settings - Fork 124
/
netprobe_speedtest.py
61 lines (39 loc) · 1.57 KB
/
netprobe_speedtest.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
# Netprobe Service
import time
from helpers.network_helper import Netprobe_Speedtest
from helpers.http_helper import *
from helpers.redis_helper import *
from config import Config_Netprobe
from datetime import datetime
from helpers.logging_helper import *
if __name__ == '__main__':
# Global Variables
speedtest_enabled = Config_Netprobe.speedtest_enabled
speedtest_interval = Config_Netprobe.speedtest_interval
collector = Netprobe_Speedtest()
# Logging Config
logger = setup_logging("logs/speedtest.log")
if speedtest_enabled == True:
while True:
try:
stats = collector.collect()
current_time = datetime.now()
except Exception as e:
print("Error running speedtest")
logger.error("Error running speedtest")
logger.error(e)
time.sleep(speedtest_interval) # Pause before retrying
continue
# Connect to Redis
try:
cache = RedisConnect()
# Save Data to Redis
cache_interval = speedtest_interval*2 # Set the redis cache 2x longer than the speedtest interval
cache.redis_write('speedtest',json.dumps(stats),cache_interval)
logger.info(f"Stats successfully written to Redis for Speed Test")
except Exception as e:
logger.error("Could not connect to Redis")
logger.error(e)
time.sleep(speedtest_interval)
else:
exit()