chore: prevent multiple instances (#386)

* chore: prevent multiple instances * fix: linting --------- Co-authored-by: Frederik Hvilshøj <[email protected]>
encord-team · May 9, 2023 · 41d02e0 · 41d02e0
1 parent 244a6cc
commit 41d02e0
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/encord_active/cli/utils/streamlit.py b/src/encord_active/cli/utils/streamlit.py
@@ -3,6 +3,7 @@
 from os import environ
 from pathlib import Path
 
+import rich
 from streamlit.web import cli as stcli
 
 from encord_active.app.app_config import app_config
@@ -29,7 +30,22 @@ def ensure_safe_project(path: Path):
         ensure_prisma_db(project_file_structure.prisma_db)
 
 
+def is_port_in_use(port: int) -> bool:
+    import socket
+
+    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
+        return s.connect_ex(("localhost", port)) == 0
+
+
 def launch_streamlit_app(target: Path):
+    if is_port_in_use(8501):
+        import typer
+
+        rich.print("[red]Port already in use...")
+        raise typer.Exit()
+    else:
+        rich.print("[yellow]Bear with us, this might take a short while...")
+
     if did_schema_change():
         generate_prisma_client()
     ensure_safe_project(target)

diff --git a/src/encord_active/server/main.py b/src/encord_active/server/main.py
@@ -38,7 +38,7 @@
 
 app = FastAPI()
 
-origins = ["http://localhost:5173", "http://localhost:8501", settings.ALLOWED_ORIGIN]
+origins = [settings.ALLOWED_ORIGIN]
 
 app.add_middleware(
     CORSMiddleware,