[IMP] Enable SSL/TLS support

pywebdriver · Mar 4, 2020 · 003bacd · 003bacd
1 parent 5ddafdf
commit 003bacd
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/pywebdriverd b/pywebdriverd
@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+import os
 
 from pywebdriver import app, config, drivers
 
@@ -9,7 +10,28 @@ def main():
     if config.getboolean('application', 'print_status_start'):
         if 'escpos' in drivers:
             drivers['escpos'].push_task('printstatus')
-    app.run(host=host, port=port, debug=debug)
+    flask_args = dict(
+        host=host,
+        port=port,
+        debug=debug,
+        processes=0,
+        threaded=True
+    )
+    if config.has_option('flask', 'sslcert'):
+        sslcert = config.get('flask', 'sslcert')
+        if sslcert:
+            if not config.has_option('flask', 'sslkey'):
+                print("If you want SSL, you must also provide the sslkey")
+                sys.exit(-1)
+            sslkey = config.get('flask', 'sslkey')
+            if not os.path.exists(sslcert):
+                print("SSL cert not found at", sslcert)
+                sys.exit(-1)
+            if not os.path.exists(sslkey):
+                print("SSL key not found at", sslkey)
+                sys.exit(-1)
+            flask_args['ssl_context'] = (sslcert, sslkey)
+    app.run(**flask_args)
 
 # Run application
 if __name__ == '__main__':