fix login info

buremba · Aug 3, 2024 · e757a20 · e757a20
1 parent 2c7d4c5
commit e757a20
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/universql/server.py b/universql/server.py
@@ -65,7 +65,7 @@ async def login_request(request: Request) -> JSONResponse:
     except OAuthError as e:
         message = e.args[0]
 
-    logger.info(f"[{token}] Created local session for user {credentials.get('user')}")
+    logger.info(f"[{token}] Created local session for user {credentials.get('user')} from {request.client.host}:{request.client.port}")
     return JSONResponse(
         {
             "data":
@@ -74,10 +74,10 @@ async def login_request(request: Request) -> JSONResponse:
                     "masterToken": token,
                     "parameters": parameters,
                     "sessionInfo": {
-                        "databaseName": "MY_CUSTOM_APP",
-                        "schemaName": "PUBLIC",
-                        "warehouseName": "PC_DBT_WH",
-                        "roleName": "ACCOUNTADMIN"
+                        "databaseName": credentials.get('database'),
+                        "schemaName": credentials.get('schema'),
+                        "warehouseName": credentials.get('warehouse'),
+                        "roleName": credentials.get('role')
                     },
                     "idToken": None,
                     "idTokenValidityInSeconds": 0,