uptime-monitor/main.py

134 lines
3.9 KiB
Python

import cv2
import numpy as np
import urllib.request
from datetime import datetime
import time
import os
recent_blog=[]
recent_cloud=[]
recent_code=[]
recent_git=[]
blog=[]
cloud=[]
code=[]
git=[]
width=720
height=50
red=(0x00,0x00,0xff)
orange=(0x00,0x66,0xff)
yellow=(0x00,0xcc,0xff)
light_green=(0x33,0xff,0xcc)
green=(0x00,0xcc,0x00)
imgpath="/tmp/uptime-graph/"
def update_history(history,recent):
average=0
for i in recent:
average+=i
average/=len(recent)
if len(history)>=144:
del(history[0])
history.append(average)
def fill(img,pos,color):
for i in range(5):
for j in range(height):
img[j,i+5*pos,:]=[color[0]%256,color[1]%256,color[2]%256]
def update_graph(filename,history):
img=np.ones([height,width,3],dtype=np.uint8)
for i,delay in enumerate(history):
if delay>=5000:
fill(img,i,red)
elif delay>=2000:
fill(img,i,orange)
elif delay>=1000:
fill(img,i,yellow)
elif delay>=500:
fill(img,i,light_green)
else:
fill(img,i,green)
cv2.imwrite(imgpath+filename, img)
def testConnect(url):
try:
time1=datetime.now().timestamp()
status_code=urllib.request.urlopen(url,timeout=5).getcode()
time2=datetime.now().timestamp()
delay=int((time2-time1)*1000)
except:
status_code=404
if status_code/100>=4:
delay=10000
return (status_code,delay)
if __name__=="__main__":
if not os.path.exists(imgpath):#如果路径不存在
os.makedirs(imgpath)
while True:
time.sleep(0.5)
if datetime.now().second==0:
time.sleep(1)
(status_code,delay)=testConnect("https://blog.clf3.org")
print(status_code,end=" ")
if status_code/100<=3:
print("connected",delay,"ms")
else:
print("error")
if len(recent_blog)<10:
recent_blog.append(delay)
else:
update_history(blog,recent_blog)
update_graph("blog.png",blog)
recent_blog=[]
recent_blog.append(delay)
elif datetime.now().second==10:
time.sleep(1)
(status_code,delay)=testConnect("https://cloud.clf3.org")
print(status_code,end=" ")
if status_code/100<=3:
print("connected",delay,"ms")
else:
print("error")
if len(recent_cloud)<10:
recent_cloud.append(delay)
else:
update_history(cloud,recent_cloud)
update_graph("cloud.png",cloud)
recent_cloud=[]
recent_cloud.append(delay)
elif datetime.now().second==20:
# if True:
time.sleep(1)
(status_code,delay)=testConnect("https://code.clf3.org")
print(status_code,end=" ")
if status_code/100<=3:
print("connected",delay,"ms")
else:
print("error")
if len(recent_code)<10:
recent_code.append(delay)
else:
update_history(code,recent_code)
update_graph("code.png",code)
recent_code=[]
recent_code.append(delay)
elif datetime.now().second==30:
# if True:
time.sleep(1)
(status_code,delay)=testConnect("https://git.clf3.org")
print(status_code,end=" ")
if status_code/100<=3:
print("connected",delay,"ms")
else:
print("error")
if len(recent_git)<10:
recent_git.append(delay)
else:
update_history(git,recent_git)
update_graph("git.png",git)
recent_git=[]
recent_git.append(delay)