CloudBot/plugins/ping.py

# ping plugin by neersighted
from util import hook
import subprocess
import re

PING_REGEX = "rtt min/avg/max/mdev = (\d+.\d+)/(\d+.\d+)/(\d+.\d+)/(\d+.\d+)"


@hook.command
def ping(inp, reply=None):
    ".ping <host> [count] -- Pings <host> [count] times."

    args = inp.split(' ')
    host = args[0]

    if len(args) > 1:
        count = int(args[1])
        if count > 20:
            count = 20
    else:
        count = 5

    count = str(count)

    host = re.sub(r'([^\s\w\.])+', '', host)

    reply("Attempting to ping %s %s times..." % (host, count))

    pingcmd = subprocess.check_output(["ping", "-c", count, host])
    if 'request timed out' in pingcmd or 'unknown host' in pingcmd:
        return "error: could not ping host"
    else:
        m = re.search(PING_REGEX, pingcmd)
        return "min: %sms, max: %sms, average: %sms, range: %sms, count: %s" \
        % (m.group(1), m.group(3), m.group(2), m.group(4), count)