Added som test for querying the db.

This commit is contained in:
2017-02-10 14:18:03 +01:00
parent cef5a369d3
commit 7e0f60996d

View File

@@ -1,44 +1,66 @@
#!/usr/bin/python
#!/usr/bin/env python3
import subprocess
from time import time, sleep
import sqlite3
def getAddr():
c.execute('SELECT adr FROM clients')
for adr in c.fetchall():
print(adr[0])
def getTimes():
c.execute('SELECT c.name, l.timesince FROM lastonline AS l JOIN clients AS c WHERE l.clientadr=c.adr')
print(c.fetchall())
def checkNameExistance():
c.execute('SELECT name, addr FROM macAddr')
test = c.fetchall()
for name, addr in test:
print(name)
# Define the users to lookup
users = ['Elias', 'Kevin', 'Nora', 'Inge', 'Bazzinga']
addr = ['38:ca:da:eb:3f:da', '2c:33:61:aa:e6:a9', 'cc:29:f5:b8:2d:a2', 'ac:5f:3e:28:2a:c0', 'f0:79:59:70:a4:a6']
addr = { "elias": '38:ca:da:eb:3f:da', "kevin": '2c:33:61:aa:e6:a9', "nora": 'cc:29:f5:b8:2d:a2',
"inge": 'ac:5f:3e:28:2a:c0', "bazzinga": 'f0:79:59:70:a4:a6' }
conn = sqlite3.connect('home.db')
c = conn.cursor()
getAddr()
getTimes()
# checkNameExistance()
# def whosHome():
# # Get the output of the command 'arp-scan -l'
# arpOutput = subprocess.check_output("sudo arp-scan -l", shell=True).split('\n')
# # Strip away first three lines and last 5 lines
# arpOutput = arpOutput[2:-4]
# # Open file times.txt and read lines to 'logFile'
# with open('/home/kevin/homeCheck/times.txt', 'r') as file:
# logFile = file.readlines()
# i = 0
# # Go through each element in list 'addr'
# for mac in addr:
# # Then iterate through each line in arpOutput
# for line in arpOutput:
# line_mac = str(line.split('\t')[1])
# # For each line we check after a matching mac addr
# if (mac in line_mac):
# logFile[i] = str(users[i]) + ':' + str(time()) + '\n'
# # print mac
# # print users[i]
# # print str(i) + '\n'
# # print logFile
# i+=1
# # Write changes to file
# with open('/home/kevin/homeCheck/times.txt', 'w') as file:
# file.writelines(logFile)
# print logFile
def whosHome():
# Get the output of the command 'arp-scan -l'
arpOutput = subprocess.check_output("sudo arp-scan -l", shell=True).split('\n')
# Strip away first three lines and last 5 lines
arpOutput = arpOutput[2:-4]
# Open file times.txt and read lines to 'logFile'
with open('/home/kevin/homeCheck/times.txt', 'r') as file:
logFile = file.readlines()
i = 0
# Go through each element in list 'addr'
for mac in addr:
# Then iterate through each line in arpOutput
for line in arpOutput:
line_mac = str(line.split('\t')[1])
# For each line we check after a matching mac addr
if (mac in line_mac):
logFile[i] = str(users[i]) + ':' + str(time()) + '\n'
# print mac
# print users[i]
# print str(i) + '\n'
# print logFile
i+=1
# Write changes to file
with open('/home/kevin/homeCheck/times.txt', 'w') as file:
file.writelines(logFile)
print logFile
whosHome()
# whosHome()