Implementing platform-agnostic packet loss and ssid detection

This commit is contained in:
Will Bradley 2017-03-08 01:50:06 -08:00
parent dee07c0e92
commit 9f20431887

View File

@ -44,6 +44,8 @@ import re
import signal import signal
# File paths # File paths
import os import os
# OS naming
import platform
# Argument parsing # Argument parsing
import argparse import argparse
# For exit # For exit
@ -52,7 +54,8 @@ import sys
# For IP addresses # For IP addresses
import socket, struct import socket, struct
# For API
import json
# #
# Main Class # Main Class
@ -61,19 +64,10 @@ import socket, struct
class Pinger: class Pinger:
def ping(self, target, log=None, widget=None, data=None): def ping(self, target, log=None, widget=None, data=None):
ping = subprocess.Popen( pingout = Util.doPing(target)
["ping", "-c", "1", target], ssidout = Util.getSsid()
stdout = subprocess.PIPE,
stderr = subprocess.PIPE
)
out, error = ping.communicate()
m = re.search('time=(.*) ms', out)
if error or m == None:
print "PING FAIL"
else:
latency = "%.2f" % float(m.group(1))
print latency+" ms"
print json.dumps({'ssid': ssidout, 'loss': int(pingout)})
def __init__(self): def __init__(self):
# Print welcome message # Print welcome message
@ -82,7 +76,45 @@ class Pinger:
self.ping("4.2.2.2") self.ping("4.2.2.2")
# Print started message # Print started message
print "Started." print "Finished."
#
# Utility class for platform-agnosticism and reusability
#
class Util:
@staticmethod
def getSsid():
if platform.system() == "Linux":
ssid = subprocess.Popen(["nmcli","-t","-f","active,ssid","dev","wifi"],
stdout = subprocess.PIPE,
stderr = subprocess.PIPE
) # | egrep '^yes(.*)' | cut -d\' -f2
out, error = ssid.communicate()
m = re.search(r'yes:(.*)', out)
result = m.group(1)
elif platform.system() == "Macintosh":
foo
return result
@staticmethod
def doPing(target):
if platform.system() == "Linux":
ping = subprocess.Popen(
["ping", "-c", "5", target],
stdout = subprocess.PIPE,
stderr = subprocess.PIPE
)
out, error = ping.communicate()
#print out
m = re.search(r' ([0-9]+)% packet loss', out, re.MULTILINE) # (^rtt .*$)|
# print m.groups()
if error or m.group(1) == 100:
pingout = -1
else:
pingout = m.group(1)
elif platform.system() == "Macintosh":
foo
return pingout
# #
# Runtime # Runtime