57 lines
1.4 KiB
Python
57 lines
1.4 KiB
Python
import subprocess
|
|
import re
|
|
import subprocess
|
|
from datetime import datetime, timedelta
|
|
|
|
import psutil
|
|
from peewee import *
|
|
from playhouse.shortcuts import model_to_dict
|
|
|
|
import config
|
|
from database import BaseModel
|
|
from plugins.plugin import Plugin
|
|
|
|
|
|
class Ping(BaseModel):
|
|
time = DateTimeField(index=True, default=datetime.utcnow)
|
|
host = TextField(null=False)
|
|
ping = FloatField(null=True) # null = timeout or error
|
|
|
|
|
|
class PingPlugin(Plugin):
|
|
models = [Ping]
|
|
|
|
def __init__(self):
|
|
self.__timeout = 20
|
|
|
|
def get_interval(self):
|
|
return config.PING_INTERVAL
|
|
|
|
def do_ping(self, host):
|
|
command = ['ping', '-c', '1', '-W', str(self.__timeout), host]
|
|
proc = subprocess.Popen(command, stdout=subprocess.PIPE)
|
|
ping = None
|
|
|
|
try:
|
|
proc.wait(60)
|
|
stdout = proc.stdout.read().decode()
|
|
|
|
match = re.search(r'time=([\d\.]+) ms', stdout)
|
|
if match is not None:
|
|
ping = float(match.group(1))
|
|
|
|
except subprocess.TimeoutExpired:
|
|
proc.kill()
|
|
|
|
entry = Ping()
|
|
entry.host = host
|
|
entry.ping = ping
|
|
entry.save()
|
|
|
|
def execute(self):
|
|
for host in config.PING_HOSTS:
|
|
self.do_ping(host)
|
|
|
|
def cleanup(self):
|
|
limit = datetime.utcnow() - timedelta(days=config.PING_RETAIN_DAYS)
|
|
return Ping.delete().where(Ping.time < limit).execute() |