mirror of
https://github.com/KevinMidboe/homeChecker.git
synced 2025-10-29 17:40:26 +00:00
81 lines
2.1 KiB
Python
Executable File
81 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import sqlite3
|
|
from subprocess import check_output, CalledProcessError
|
|
from time import time
|
|
from re import findall
|
|
from sys import argv
|
|
|
|
def getOnlineClients():
|
|
try:
|
|
arpOutput = check_output("sudo arp-scan -l", shell=True)
|
|
arpOutput = arpOutput.decode()
|
|
|
|
macAdr = findall('(([0-9A-Fa-f]{2}[:-]){5}([0-9A-Fa-f]{2}))', arpOutput)
|
|
return [i[0] for i in macAdr]
|
|
|
|
except CalledProcessError:
|
|
print("Not able to run 'arp-scan -l' on this machine.")
|
|
exit(0)
|
|
|
|
def getAddr():
|
|
c.execute('SELECT adr FROM clients')
|
|
|
|
return [i[0] for i in c.fetchall()]
|
|
|
|
def getTimes():
|
|
conn = sqlite3.connect('home.db')
|
|
c = conn.cursor()
|
|
|
|
c.execute('SELECT c.name, l.timesince FROM lastonline AS l JOIN clients AS c WHERE l.clientadr=c.adr')
|
|
|
|
returnList = []
|
|
for name, time in c.fetchall():
|
|
returnList.append({"name": name, "time": convertTime(time)})
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
return returnList
|
|
|
|
def convertTime(seconds):
|
|
delta = time() - seconds
|
|
if delta >= 86400:
|
|
return str(delta//86400) + ' days'
|
|
elif delta >= 3600:
|
|
if delta//3600 < 10:
|
|
parent = str(delta//3600)
|
|
child = str((delta - (3600 * (delta//3600)))//60)
|
|
if len(child) == 1:
|
|
child = '0' + child
|
|
return parent + ':' + child + ' hours'
|
|
else:
|
|
return str(delta//3600) + ' hours'
|
|
elif delta >= 60:
|
|
return str(delta//60) + ' minutes'
|
|
else:
|
|
return str(delta) + ' seconds'
|
|
|
|
|
|
def updateTimes():
|
|
curTime = time()
|
|
conn = sqlite3.connect('home.db')
|
|
c = conn.cursor()
|
|
|
|
online = list(set(getOnlineClients()) & set(getAddr()))
|
|
|
|
for adr in online:
|
|
c.execute('UPDATE lastonline SET timesince='+ str(curTime) +' WHERE clientadr="cc:29:f5:b8:2d:a2"')
|
|
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
if __name__ == '__main__':
|
|
if argv[-1] == 'add':
|
|
print(updateTimes())
|
|
elif argv[-1] == 'get':
|
|
print(getTimes())
|
|
else:
|
|
print("Add args 'add' or 'get'")
|
|
|
|
|