src/containerspawner/state.py

@@ -44,11 +44,15 @@ class StateManager:
             name=container_name,
             network=CONTAINER_NETWORK,
             volumes={container_name: {"bind": CONTAINER_PERSIST}},
+            restart_policy={"Name": "always", "MaximumRetryCount": 5}
             detach=True,
             auto_remove=True
         )
 
         # Wait for container to start.
+        while container.status != "running":
+            time.sleep(1)
+
         time.sleep(3)
 
         # Disable auth.
@@ -56,3 +60,6 @@ class StateManager:
 
         # Install extensions.
         container.exec_run("code-server --install-extension ms-python.python")
+
+        # Restart container.
+        container.restart()