61 lines
1.7 KiB
Python
61 lines
1.7 KiB
Python
import requests
|
|
|
|
from app import app
|
|
from app.extensions import db
|
|
from app.models import AlarmState, Alarm, Proxy
|
|
|
|
|
|
def set_http_alarm(proxy_id: int, state: AlarmState, text: str):
|
|
alarm = Alarm.query.filter(
|
|
Alarm.proxy_id == proxy_id,
|
|
Alarm.alarm_type == "http-status"
|
|
).first()
|
|
if alarm is None:
|
|
alarm = Alarm()
|
|
alarm.proxy_id = proxy_id
|
|
alarm.alarm_type = "http-status"
|
|
alarm.target = "proxy"
|
|
db.session.add(alarm)
|
|
alarm.update_state(state, text)
|
|
|
|
|
|
def check_http():
|
|
proxies = Proxy.query.filter(
|
|
Proxy.destroyed == None
|
|
)
|
|
for proxy in proxies:
|
|
try:
|
|
if proxy.url is None:
|
|
continue
|
|
r = requests.get(proxy.url,
|
|
allow_redirects=False,
|
|
timeout=5)
|
|
r.raise_for_status()
|
|
if r.is_redirect:
|
|
set_http_alarm(
|
|
proxy.id,
|
|
AlarmState.CRITICAL,
|
|
f"{r.status_code} {r.reason}"
|
|
)
|
|
else:
|
|
set_http_alarm(
|
|
proxy.id,
|
|
AlarmState.OK,
|
|
f"{r.status_code} {r.reason}"
|
|
)
|
|
except (requests.ConnectionError, requests.Timeout):
|
|
set_http_alarm(
|
|
proxy.id,
|
|
AlarmState.CRITICAL,
|
|
f"Connection failure")
|
|
except requests.HTTPError:
|
|
set_http_alarm(
|
|
proxy.id,
|
|
AlarmState.CRITICAL,
|
|
f"{r.status_code} {r.reason}"
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
with app.app_context():
|
|
check_http()
|