Allow login credentials to be saved in user folder

spectur_live_view.py

@@ -9,6 +9,7 @@ import pytz
 from io import BytesIO
 from datetime import datetime
 import requests
+import yaml
 import numpy as np
 from PIL import Image
 from selenium import webdriver
@@ -52,25 +53,35 @@ def login(driver):
 
     driver.get(URL)
 
-    # Login automatically if details provided
+    # Get login fields
-    if (username and password):
+    usr = driver.find_element_by_id('UserName')
-        # Get login fields
+    pss = driver.find_element_by_id('Password')
-        usr = driver.find_element_by_id('UserName')
+    btn = driver.find_element_by_id('submitlogin')
-        pss = driver.find_element_by_id('Password')
-        btn = driver.find_element_by_id('submitlogin')
 
-        # Input login details
+    if (username and password):
+        # Login automatically if details provided
         usr.send_keys(username)
         pss.send_keys(password)
         btn.click()
     else:
-        # Wait for user to log in manually
+        # Try to get credientials saved locally
-        while True:
+        try:
-            try:
+            credential_path = os.path.join(os.path.expanduser('~'), '.spectur')
-                driver.find_element_by_id('UserName')
+            with open(credential_path, 'r') as f:
-                time.sleep(1)
+                credentials = yaml.safe_load(f.read())
-            except NoSuchElementException:
+
-                break
+            # Input login details
+            usr.send_keys(credentials['USERNAME'])
+            pss.send_keys(credentials['PASSWORD'])
+            btn.click()
+        except FileNotFoundError:
+            # Wait for user to log in manually
+            while True:
+                try:
+                    driver.find_element_by_id('UserName')
+                    time.sleep(1)
+                except NoSuchElementException:
+                    break
 
     # Extract cookies
     session = requests.Session()