Wrap export in try: finally:

export/main.py

@@ -12,39 +12,45 @@ chrome_options = Options()
 chrome_options.add_argument('--headless')
 chrome_options.add_argument('start-maximized')
 chrome_options.add_argument('--no-sandbox')
+
 driver = webdriver.Chrome(service=ser, options=chrome_options)
-driver.get('https://sensors.dns.t0.vc')
+
-time.sleep(3)
+try:
-
+    driver.get('https://sensors.dns.t0.vc')
-driver.execute_script("return document.getElementsByClassName('menu')[0].remove();")
+    time.sleep(3)
-
+
-graphs = [
+    driver.execute_script("return document.getElementsByClassName('menu')[0].remove();")
-    'Solar_Power',
+
-    'Living_Room_Air',
+    graphs = [
-    'Outside_Temperature',
+        'Solar_Power',
-    'Bedroom_Temperature',
+        'Living_Room_Air',
-    'Nook_Temperature',
+        'Outside_Temperature',
-    'Misc_Temperature',
+        'Bedroom_Temperature',
-    'Nook_Thermostat',
+        'Nook_Temperature',
-    'Gas_Usage',
+        'Misc_Temperature',
-    'Water_Usage',
+        'Nook_Thermostat',
-    'Living_Room_Lux',
+        'Gas_Usage',
-]
+        'Water_Usage',
-
+        'Living_Room_Lux',
-for graph in graphs:
+    ]
-    print('Capturing', graph, 'graph...')
+
-
+    for graph in graphs:
-    try:
+        print('Capturing', graph, 'graph...')
-        element = driver.find_element(By.ID, graph)
+
-    except NoSuchElementException:
+        try:
-        print('Graph not found, skipping.')
+            element = driver.find_element(By.ID, graph)
-        continue
+        except NoSuchElementException:
-
+            print('Graph not found, skipping.')
-    driver.execute_script('arguments[0].scrollIntoView({block: "center"});', element)
+            continue
-
+
-    with open('/home/tanner/sensors/export/data/{}.png'.format(graph), 'wb') as f:
+        driver.execute_script('arguments[0].scrollIntoView({block: "center"});', element)
-        f.write(element.screenshot_as_png)
+
-
+        with open('/home/tanner/sensors/export/data/{}.png'.format(graph), 'wb') as f:
-driver.close()
+            f.write(element.screenshot_as_png)
+
+finally:
+    driver.close()
+    driver.quit()
+
 print('done.')