change default cwd depending on the user if cwd is not provided

Author: mishushakov

template/server/contexts.py

@@ -11,6 +11,13 @@
 
 logger = logging.Logger(__name__)
 
+def get_user_cwd(user: str, cwd: Optional[str]) -> str:
+    if not cwd:
+        if user == "root":
+            return "/root"
+        return "/home/user"
+    return cwd
+
 def normalize_language(language: Optional[str]) -> str:
     if not language:
         return "python"

template/server/main.py

@@ -13,7 +13,7 @@
 from api.models.create_context import CreateContext
 from api.models.execution_request import ExecutionRequest
 from consts import JUPYTER_BASE_URL
-from contexts import create_context, normalize_language
+from contexts import create_context, normalize_language, get_user_cwd
 from messaging import ContextWebSocket
 from stream import StreamingListJsonResponse
 from utils.locks import LockedMap
@@ -126,8 +126,8 @@ async def post_contexts(request: CreateContext) -> Context:
     logger.info(f"Creating a new context")
 
     language = normalize_language(request.language)
-    cwd = request.cwd or "/home/user"
     user = request.user or "user"
+    cwd = get_user_cwd(user, request.cwd)
 
     try:
         return await create_context(client, websockets, language, cwd, user)