4 rasberry pi wifi upload script

.env.example

@@ -0,0 +1,4 @@
+WIFI_PASSWORD =
+DISCORD_WEBHOOK = 
+DEVICE_ID = 
+NETWORK_NAME = 

.flake8

@@ -1,6 +1,6 @@
 [flake8]
 
-max-line-length = 140
+max-line-length = 180
 
 # Exclude certain file patterns from checking
 exclude = 

.gitignore

@@ -30,4 +30,5 @@ Thumbs.db
 
 data/
 .ecu_data
-512311_xk3jq6ao_config.json
+512311_xk3jq6ao_config.json
+512311__config.json

box_client/box_client.py → data_uploader/box_client/box_client.py

@@ -7,28 +7,27 @@
 from cryptography.hazmat.primitives.serialization import load_pem_private_key
 from cryptography.hazmat.backends import default_backend
 
-TOKEN_URL = "https://api.box.com/oauth2/token"
-UPLOAD_URL = "https://upload.box.com/api/2.0/files/content"
-USER_INFO_URL = "https://api.box.com/2.0/users/me"
-
-config = json.load(
-    open('512311_xk3jq6ao_config.json'))
-
-key_id = config['boxAppSettings']['appAuth']['publicKeyID']
+TOKEN_URL = os.getenv("TOKEN_URL")
+UPLOAD_URL = os.getenv("UPLOAD_URL")
+USER_INFO_URL = os.getenv("USER_INFO_URL")
 
 
 class Client:
 
-    def __init__(self, client_id: str, client_secret: str, file_path: str, folder_id: int):
-        self.client_id = config['boxAppSettings']['clientID']
-        self.client_secret = config['boxAppSettings']['clientSecret']
-        self.file_path = file_path
+    def __init__(self, client_id: str, client_secret: str, enterprise_id: str, key_id: str, private_key: str, password: str, folder_path: str, folder_id: int):
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.enterprise_id = enterprise_id
+        self.key_id = key_id
+        self.private_key = private_key
+        self.password = password
+        self.folder_path = folder_path
         self.folder_id = folder_id
 
     def retrieve_access_token(self) -> str:
-        appAuth = config['boxAppSettings']['appAuth']
-        private_key = appAuth['privateKey']
-        password = appAuth['passphrase']
+        key_id = self.key_id
+        private_key = self.private_key
+        password = self.password
 
         key = load_pem_private_key(
             data=private_key.encode('utf8'),
@@ -37,8 +36,8 @@ def retrieve_access_token(self) -> str:
         )
 
         claims = {
-            'iss': config['boxAppSettings']['clientID'],
-            'sub': config['enterpriseID'],
+            'iss': self.client_id,
+            'sub': self.enterprise_id,
             'box_sub_type': 'enterprise',
             'aud': TOKEN_URL,
             'jti': secrets.token_hex(64),
@@ -77,37 +76,39 @@ def retrieve_access_token(self) -> str:
             print(response.text)
             return None
 
-    def send_files(self) -> bool:
+    def send_files(self, filenames: list) -> bool:
         access_token = self.retrieve_access_token()
-        file_name = os.path.basename(self.file_path)
 
-        headers = {
-            "Authorization": f"Bearer {access_token}",
-        }
+        for filename in filenames:
 
-        with open(self.file_path, 'rb') as file_to_upload:
+            file_name = os.path.basename(filename)
 
-            files = {
-                'file': (file_name, file_to_upload),
-                'attributes': (None, json.dumps({'parent': {'id': '217403389478'}})),
+            headers = {
+                "Authorization": f"Bearer {access_token}",
             }
 
-            # print(json.dumps({'parent': {'id': '217403389478'}}))
+            with open(filename, 'rb') as file_to_upload:
 
-            try:
-                response = requests.post(
-                    UPLOAD_URL, headers=headers, files=files)
+                files = {
+                    'file': (filename, file_to_upload),
+                    'attributes': (None, json.dumps({'parent': {'id': str(self.folder_id)}, 'name': file_name}), 'application/json'),
+                }
 
-                if response.status_code == 201:
-                    return True
-                else:
-                    print(f"Error: {response.status_code}")
-                    print(response.json())
+                try:
+                    response = requests.post(UPLOAD_URL, headers=headers, files=files)
+
+                    if response.status_code == 201:
+                        continue
+                    else:
+                        print(f"Error: {response.status_code}")
+                        print(response.json())
+                        return False
+
+                except requests.exceptions.RequestException as error:
+                    print(f'Request Exception: {error}')
                     return False
 
-            except requests.exceptions.RequestException as error:
-                print(f'Request Exception: {error}')
-                return False
+        return True
 
     def get_user_info(self):
 
@@ -127,3 +128,18 @@ def get_user_info(self):
         except requests.exceptions.RequestException as error:
             print(f'Request Exception: {error}')
             return False
+
+    def discover_files(self) -> list:
+
+        list_of_files = []
+
+        if os.path.exists(self.folder_path) and os.path.isdir(self.folder_path):
+            files = os.listdir(self.folder_path)
+
+            for file in files:
+                list_of_files.append("C:\\Users\\sajip\\OneDrive\\Desktop\\" + file)
+                print(file)
+
+            return list_of_files
+
+        return None

data_uploader/discord_client/messages.py

@@ -0,0 +1,11 @@
+from enum import Enum
+
+
+class Messages(str, Enum):
+    MONGO_SUCCESS_MESSAGE = "Docuemnts inserted Successfully"
+    MONGO_ERROR_MESSAGE = "An error occurred while trying to connect to MongoDB: {e}"
+    MONGO_CLOSE_MESSAGE = "MongoDB connection closed."
+    BOX_SUCCESS_MESSAGE = "Files uploaded successfully"
+    BOX_ERROR_MESSAGE = "Error occurred while trying to upload files to Box"
+    WIFI_SUCCESS_MESSAGE = "Connected to network successfully"
+    WIFI_ERROR_MESSAGE = "Error occurred while trying to connect to network"

data_uploader/wifi_client/wifi_client.py

@@ -0,0 +1,72 @@
+import os
+import subprocess
+import re
+import traceback
+
+'''
+This script removes all saved networks to ensure that the specific
+network passed is chosen when restarting the wifi adapter
+
+We could try testing if appending to the wpa_supplicant.conf file
+works so other saved wifi networks can remain
+'''
+
+
+class Client:
+    def __init__(self, network_name: str, network_password: str):
+        self.network_name = network_name
+        self.network_password = network_password
+
+    def scan_for_networks(self) -> list:
+        try:
+            devices = subprocess.check_output(['sudo', 'iwlist', 'wlan0', 'scan'])
+            network_names = devices.decode('utf-8')
+            network_names = re.findall(r'ESSID:"(.*?)"', devices)
+
+            return network_names
+
+        except subprocess.CalledProcessError as err:
+            traceback.print_exc(err.returncode)
+            return []
+
+    def find_network(self) -> bool:
+        found = False
+
+        for network_ssid in self.scan_for_networks():
+            print(self.network_name, network_ssid)
+            if network_ssid == self.network_name:
+                print("Found home network")
+                found = True
+                break
+
+        return found
+
+    def connect_to_network(self) -> bool:
+        if self.find_network():
+            try:
+                network = subprocess.check_output(['wpa_passphrase', self.network_name, self.network_password], stderr=subprocess.STDOUT)
+                with open("/etc/wpa_supplicant/wpa_supplicant.conf", "w+") as fp:
+                    fp.write("ctrl_interface=DIR=/var/run/wpa_supplicant GROUP=netdev\nupdate_config=1\ncountry=US\n")
+                    fp.write(network.decode("utf8"))
+
+                subprocess.check_output(["wpa_cli", "-i", "wlan0", "reconfigure"])
+                return True
+
+            except subprocess.CalledProcessError as err:
+                traceback.print_exc(err.returncode)
+                return False
+        else:
+            print("Could not find network")
+            return False
+
+
+def main():
+    name = os.getenv('NETWORK_NAME')
+    password = os.getenv('NETWORK_PASSWORD')
+
+    client = Client(name, password)
+    print(client.connect_to_network())
+
+
+if __name__ == "__main__":
+    main()

handler.py

@@ -1,43 +1,61 @@
-from box_client.box_client import Client as BoxClient
-from discord_client.discord_client import Client as DiscordClient
-from mongo_client.mongo_client import Client as MongoClient
-from wifi_client.wifi_client import Client as WifiClient
+from data_uploader.box_client.box_client import Client as BoxClient
+from data_uploader.discord_client.discord_client import Client as DiscordClient
+from data_uploader.discord_client.messages import Messages as discord_messages
+from data_uploader.mongo_client.mongo_client import Client as MongoClient
+from data_uploader.wifi_client.wifi_client import Client as WifiClient
+
 
 import os
 from dotenv import load_dotenv
+import json
+
+
+config = json.load(
+    open('512311_xk3jq6ao_config.json')
+)
 
 
 class Handler:
 
     def handler():
 
         load_dotenv()
-        WIFI_PASSWORD = os.getenv('WIFI_PASSWORD')
+        NETWORK_NAME = os.getenv('NETWORK_NAME')
+        WIFI_PASSWORD = os.getenv('NETWORK_PASSWORD')
         WEBHOOK_URL = os.getenv('DISCORD_WEBHOOK')
 
-        wifi_client = WifiClient('device_id', 'home_network', WIFI_PASSWORD)
+        wifi_client = WifiClient(NETWORK_NAME, WIFI_PASSWORD)
         discord_client = DiscordClient(WEBHOOK_URL)
-        box_client = BoxClient()
+        box_client = BoxClient(config['boxAppSettings']['clientID'], config['boxAppSettings']['clientSecret'], config['enterpriseID'], config['appAuth']['publicKeyID'],
+                               config['appAuth']['privateKey'], config['boxAppSettings']['appAuth']['passphrase'], config['file_path'], config['folder_id'])
         mongo_client = MongoClient('cluster0', 'dfr_sensor_data')
 
-        wifi_networks = wifi_client.get_wifi_networks()
+        files_for_upload = []
+
+        if wifi_client.connect_to_network():
 
-        if 'NETGEAR76' in wifi_networks:
+            discord_client.post_message(discord_messages.WIFI_SUCCESS_MESSAGE)
 
             try:
-                wifi_client.connect()
-                box_client.send_files()
-                discord_client.post_message("File uploaded to Box")
+                files_for_upload = box_client.discover_files()
+                box_client.send_files(files_for_upload)
+
+                discord_client.post_message(discord_messages.BOX_SUCCESS_MESSAGE)
+
                 mongo_client.check_connection()
-                mongo_client.insert_documents()
+                mongo_client.insert_documents(files_for_upload)
+                discord_client.post_message(discord_messages.MONGO_SUCCESS_MESSAGE)
                 mongo_client.close_connection()
+                discord_client.post_message(discord_messages.MONGO_CLOSE_MESSAGE)
 
                 discord_client.post_message("Documents inserted into MongoDB")
             except Exception as e:
                 print.traceback(e)
                 discord_client.post_message(f"An error occurred: {e}")
                 return
 
+        discord_client.post_message(discord_messages.WIFI_ERROR_MESSAGE)
+
         return None
 
 

tests/conftest.py

@@ -2,17 +2,26 @@
 from mongo_client.mongo_client import Client as Mongo_Client
 import pytest
 import os
+import json
 
 uri = f"mongodb+srv://noel:${os.getenv('MONGO_PASSWORD')}@cluster0.gw7z3sn.mongodb.net/?retryWrites=true&w=majority"
 
+config = json.load(
+    open('512311_xk3jq6ao_config.json')
+    )
+
 
 @pytest.fixture
 def box_client():
     return Box_Client(
-        client_id=os.getenv("CLIENT_ID"),
-        client_secret=os.getenv("CLIENT_SECRET"),
-        file_path="/Users/noeljohnson/Documents/file.txt",
-        folder_id="217403389478"
+        client_id=config['boxAppSettings']['clientID'],
+        client_secret=config['boxAppSettings']['clientSecret'],
+        enterprise_id=config['enterpriseID'],
+        key_id=config['boxAppSettings']['appAuth']['publicKeyID'],
+        private_key=config['boxAppSettings']['appAuth']['privateKey'],
+        password=config['boxAppSettings']['appAuth']['passphrase'],
+        folder_path=config['folder_path'],
+        folder_id=config['folder_id']
     )