mainsequence 2.0.2__py3-none-any.whl → 2.0.3__py3-none-any.whl

mainsequence/cli/api.py

@@ -155,3 +155,32 @@ def add_deploy_key(project_id: int | str, key_title: str, public_key: str) -> No
                {"key_title": key_title, "public_key": public_key})
     except Exception:
         pass
+
+def get_project_token(project_id: int | str) -> str:
+    """
+    Fetch the project's token using the current access token.
+    If the access token is expired or missing, authed() will refresh once.
+    If refresh also fails, NotLoggedIn is raised so the caller can prompt re-login.
+    """
+    r = authed("GET", f"/orm/api/pods/projects/{project_id}/get_project_token/")
+    if not r.ok:
+        # authed() already tried refresh on 401;
+        # at this point treat as API error with server message.
+        msg = r.text
+        try:
+            if r.headers.get("content-type","").startswith("application/json"):
+                data = r.json()
+                msg = data.get("detail") or data.get("message") or msg
+        except Exception:
+            pass
+        raise ApiError(f"Project token fetch failed ({r.status_code}). {msg}")
+
+    token = None
+    if r.headers.get("content-type","").startswith("application/json"):
+        data = r.json()
+        token=data["development"]
+    if not token:
+        token = (r.text or "").strip()
+    if not token:
+        raise ApiError("Project token response did not include a token.")
+    return token[0]

mainsequence/cli/cli.py

@@ -26,6 +26,7 @@ from .api import (
     login as api_login,
     repo_name_from_git_url,
     safe_slug,
+get_project_token
 )
 from .ssh_utils import (
     ensure_key_for_repo,
@@ -287,6 +288,7 @@ def project_set_up_locally(
     cfg_obj = cfg.get_config()
     base = base_dir or cfg_obj["mainsequence_path"]
 
+
     org_slug = _org_slug_from_profile()
 
     items = get_projects()
@@ -346,8 +348,33 @@ def project_set_up_locally(
     else:
         if env_text and not env_text.endswith("\n"): env_text += "\n"
         env_text += f"VFB_PROJECT_PATH={str(target_dir)}\n"
+
+    try:
+        project_token = get_project_token(project_id)
+    except NotLoggedIn:
+        typer.secho("Session expired or refresh failed. Run: mainsequence login <email>", fg=typer.colors.RED)
+        raise typer.Exit(1)
+    except ApiError as e:
+        typer.secho(f"Could not fetch project token: {e}", fg=typer.colors.RED)
+        raise typer.Exit(1)
+
+    lines = env_text.splitlines()
+    if any(line.startswith("MAINSEQUENCE_TOKEN=") for line in lines):
+        lines = [
+            (f"MAINSEQUENCE_TOKEN={project_token}" if line.startswith("MAINSEQUENCE_TOKEN=") else line)
+            for line in lines
+        ]
+        env_text = "\n".join(lines)
+    else:
+        if env_text and not env_text.endswith("\n"): env_text += "\n"
+        env_text += f"MAINSEQUENCE_TOKEN={project_token}\n"
+
+    # write final .env with both vars present
     (target_dir / ".env").write_text(env_text, encoding="utf-8")
 
+
+
+
     cfg.set_link(project_id, str(target_dir))
 
     typer.secho(f"Local folder: {target_dir}", fg=typer.colors.GREEN)

mainsequence/cli/config.py

@@ -75,4 +75,8 @@ def set_env_access(access: str) -> None:
 def backend_url() -> str:
     cfg = get_config()
     url = (cfg.get("backend_url") or DEFAULTS["backend_url"]).rstrip("/")
+
+    if os.environ.get("MAIN_SEQUENCE_BACKEND_URL") is not None:
+        url=os.environ.get("MAIN_SEQUENCE_BACKEND_URL")
+
     return url

{mainsequence-2.0.2.dist-info → mainsequence-2.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 2.0.2
+Version: 2.0.3
 Summary: Main Sequence SDK
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-2.0.2.dist-info → mainsequence-2.0.3.dist-info}/RECORD

@@ -2,9 +2,9 @@ mainsequence/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mainsequence/__main__.py,sha256=1_gCQd-MlXQxAubgVdTDgmP2117UQmj5DvTI_PASZis,162
 mainsequence/logconf.py,sha256=OLkhALon2uIDZSeGOwX3DneapkVqlwDjqJm96TRAhgA,10057
 mainsequence/cli/__init__.py,sha256=TNxXsTPgb52OhakIda9wTRh91cqoBqgQRx5TxjzQQFU,21
-mainsequence/cli/api.py,sha256=swUAVjJCr2xannssLcg6BrwbToPFNgqu1G0_kyVTLck,6155
-mainsequence/cli/cli.py,sha256=fVc5k8g96LYDrah5334vDXeSGH8_lf6unMH0WSFtG34,16435
-mainsequence/cli/config.py,sha256=HVTGfXaxlYE47VShvYuIo40Wdkxaf3xYM_y4FsI0mGg,2583
+mainsequence/cli/api.py,sha256=PocYESdJvywSo7WyZnnJZmPR_Q8TKe-5jaNazyMhX9I,7313
+mainsequence/cli/cli.py,sha256=CPG-Q_8b2KFGWGgGbfZzSx8qjauVrgBzg5UMUyQrM00,17308
+mainsequence/cli/config.py,sha256=UAi0kfxro2oJhzaRdd2D3F72eGpqkYaWjSsCzBrb2LY,2705
 mainsequence/cli/ssh_utils.py,sha256=95CmZk2nIoJhNTCmifMEsE9QXJrXY3zvfxLQp3QpdiM,5088
 mainsequence/client/__init__.py,sha256=CDlJtsh0L9Z29OVMBnZsQajKBa193Yy5TkqLoilkbZM,869
 mainsequence/client/base.py,sha256=j6_iXmt6vPCS7INSOItTpM6ZbAmITAuPpMFNB9ZQTeE,15000
@@ -103,9 +103,9 @@ mainsequence/virtualfundbuilder/resource_factory/app_factory.py,sha256=XSZo9ImdT
 mainsequence/virtualfundbuilder/resource_factory/base_factory.py,sha256=jPXdK2WCaNw3r6kP3sZGIL7M4ygfIMs8ek3Yq4QYQZg,9434
 mainsequence/virtualfundbuilder/resource_factory/rebalance_factory.py,sha256=ysEeJrlbxrMPA7wFw7KDtuCTzXYkdfYZuxUFpPPY7vE,3732
 mainsequence/virtualfundbuilder/resource_factory/signal_factory.py,sha256=ywa7vxxLlQopuRwwRKyj866ftgaj8uKVoiPQ9YJ2IIo,7198
-mainsequence-2.0.2.dist-info/licenses/LICENSE,sha256=fXoCKgEuZXmP84_QDXpvO18QHze_6AAhd-zvZBUjJxQ,4524
-mainsequence-2.0.2.dist-info/METADATA,sha256=atyVDb6JvsyadKJYCHpSJ5grgkx5dhOI4SI-N7zcNc8,8028
-mainsequence-2.0.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mainsequence-2.0.2.dist-info/entry_points.txt,sha256=2J8TprrUndh7AttNTlXAaxgGtkXFUAHgXs-M7DCj5MU,58
-mainsequence-2.0.2.dist-info/top_level.txt,sha256=uSLD9rXMDMN0cc1x0p808bwyQMoRmYY2pdQZEWLajX8,13
-mainsequence-2.0.2.dist-info/RECORD,,
+mainsequence-2.0.3.dist-info/licenses/LICENSE,sha256=fXoCKgEuZXmP84_QDXpvO18QHze_6AAhd-zvZBUjJxQ,4524
+mainsequence-2.0.3.dist-info/METADATA,sha256=NKlRdCIFUq6JmHdtsEktvayTEKAWrhNVnWdQ4A_-JAk,8028
+mainsequence-2.0.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mainsequence-2.0.3.dist-info/entry_points.txt,sha256=2J8TprrUndh7AttNTlXAaxgGtkXFUAHgXs-M7DCj5MU,58
+mainsequence-2.0.3.dist-info/top_level.txt,sha256=uSLD9rXMDMN0cc1x0p808bwyQMoRmYY2pdQZEWLajX8,13
+mainsequence-2.0.3.dist-info/RECORD,,