filter.py

import csv
import os

# Runs a bash script that scans the network's BSSID, PWR, ESSID and saves it to a csv file
os.system("chmod +x convert.sh")
os.system("chmod +x scanner.sh")
os.system("./scanner.sh")
os.system("./convert.sh")

# Opens the saved csv and parses the info and saved as lists
with open('isolated.csv') as file1:
    reader = csv.reader(file1, delimiter=',')
    temp = []
    full_list = []
    for row in reader:
        temp.append(row)
    for row in temp:
        if row[1] == ',' or row[1] == 'Power' or row[1] == '' or row[1] == 'MAC' or row[2] == '':
            continue
        elif int(row[1]) <= -70 or int(row[1]) == -1:
            continue
        else:
            full_list.append(row)
os.system("rm isolated.csv")

# Our trusted list is parsed and saved to a Python dictionary
with open('baseline.csv') as file2:
    reader2 = csv.reader(file2)
    base_dict = {}
    for row in reader2:
        try:
            base_dict[row[2]] = {'BSSID': row[0], 'PWR': row[1]}
        except:
            continue

# Compares the scanned list to our trusted list and outputs safe if the area is secure, vice versa
safe = True
for list in full_list:
    print(list)
    if list[2] in base_dict.keys():
        # Check BSSID, if similar, move on
        if list[0] == base_dict[list[2]]['BSSID']:
            # Check pwr
            lower = base_dict[list[2]]['PWR']
            true_lower = int(lower) + 5
            upper = base_dict[list[2]]['PWR']
            true_upper = int(upper) - 5
            if (int(list[1]) <= true_lower) and (int(list[1]) >= true_upper):
                safe = True
            else:
                safe = False
        else:
            safe = False
    else:
        safe = False

if safe:
    print("Safe, No Threat Found. Rescanning in 10 mins")
    os.system ("notify --text 'The network should be secure' ")
else:
    print("ERROR UNSAFE, Mobile notification sent")
os.system ("notify --text 'The network might not be secure' ")