make token available through CLI

zndraw/app.py

@@ -24,7 +24,7 @@
 def index():
     """Render the main ZnDraw page."""
     if "uuid" not in session:
-        session["uuid"] = str(uuid.uuid4())
+        session["uuid"] = app.config.get("uuid", str(uuid.uuid4()))
 
         kwargs = {
             "url": request.url_root,

zndraw/cli.py

@@ -57,6 +57,10 @@ def main(
         hidden=True,
         help="Set the html attribute upgrade-insecure-requests. If you are running ZnDraw behind a reverse proxy and encounter issues with insecure requests, you might want to set this to true.",
     ),
+    use_token: bool = typer.Option(
+        False,
+        help="Use a token to authenticate the ZnDraw server.  This is useful if you are running ZnDraw as a server application.",
+    ),
 ):
     """Start the ZnDraw server.
 
@@ -77,4 +81,5 @@ def main(
         compute_bonds=compute_bonds,
         multiprocessing=multiprocessing,
         upgrade_insecure_requests=upgrade_insecure_requests,
+        use_token=use_token,
     )

zndraw/view.py

@@ -47,13 +47,16 @@ def view(
     step: int,
     compute_bonds: bool,
     multiprocessing: bool,
+    use_token: bool = False,
     upgrade_insecure_requests: bool = False,
 ):
     if filename is not None:
         app.config["filename"] = filename
         app.config["start"] = start
         app.config["stop"] = stop
         app.config["step"] = step
+    if not use_token:
+        app.config["uuid"] = "default"
     app.config["upgrade_insecure_requests"] = upgrade_insecure_requests
     app.config["compute_bonds"] = compute_bonds
     app.config["multiprocessing"] = multiprocessing

zndraw/zndraw.py

@@ -61,7 +61,7 @@ class ZnDraw(collections.abc.MutableSequence):
     )
     file: FileIO = None
     wait: bool = False
-    token: str = None
+    token: str = "default"
 
     display_new: bool = True
     _retries: int = 5