mainsequence 2.0.0__tar.gz

mainsequence-2.0.0/LICENSE

@@ -0,0 +1,40 @@
+MainSequence GmbH SDK License Agreement
+This License Agreement (the "License") governs the use, modification, and distribution of the software provided by MainSequence GmbH (the "Licensor"). The software (the "Software") is provided to you under the terms of this License. By using the Software, you agree to the terms of this License.
+
+TERMS AND CONDITIONS
+0. Definitions
+"Personal Use": Use by an individual for personal purposes that are not connected to any business, organization, or commercial activity.
+"Internal Use": Use within a business, organization, or other entity, provided it is not made accessible to third parties or used for commercial purposes.
+"Commercial Use": Use of the Software in exchange for monetary or other compensation, including hosting, offering Software as a service, selling the Software, or using it in a product or service for sale.
+"License Agreement": The legally binding agreement between the Licensor and the licensee (you), subject to the terms outlined in this License.
+1. Grant of License
+1.1 Personal and Internal Use
+The Licensor grants you a limited, non-exclusive, non-transferable, revocable license to use and modify the Software for personal or internal use only, provided that such use is strictly subject to this License Agreement and continues only while the License Agreement remains in effect.
+Upon termination of this License Agreement, all rights to use the Software for personal or internal purposes shall be immediately revoked, and you must cease all use of the Software.
+1.2 Modification
+You are permitted to modify the Software solely for your own personal or internal use, subject to the restrictions outlined in this License Agreement.
+You are not permitted to distribute, sublicense, or otherwise transfer modified or unmodified versions of the Software to any third party.
+1.3 Prohibited Redistribution
+You may not redistribute, sublicense, sell, lease, rent, or otherwise transfer the Software, whether in its original form or as modified by you, to any third party.
+Any attempt to distribute or transfer the Software in any way, without explicit permission from MainSequence GmbH, is strictly prohibited.
+2. Prohibition of Commercial Use
+The Software may not be used for any commercial purposes without obtaining a separate commercial license from MainSequence GmbH.
+Examples of prohibited commercial use include, but are not limited to:
+Hosting or offering the Software as a service to others, either modified or unmodified.
+Using the Software as part of a commercial product or service provided to customers for a fee.
+Using the Software in any production environment that generates income, directly or indirectly, from its use.
+3. Termination
+This License Agreement will automatically terminate if you fail to comply with any of its terms.
+Upon termination of this License, you must immediately cease all use of the Software, destroy all copies (modified or unmodified), and remove the Software from any devices or systems on which it is installed.
+MainSequence GmbH reserves the right to terminate this License at its discretion for any violation of its terms or for any other reason.
+4. Commercial License
+If you wish to use the Software for commercial purposes, you must contact MainSequence GmbH to negotiate and obtain a separate commercial license. The terms of the commercial license, including any fees, will be negotiated separately from this License Agreement.
+Without a valid commercial license, you are strictly prohibited from using the Software for any commercial activity.
+5. Disclaimer of Warranty
+THE SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NONINFRINGEMENT. IN NO EVENT SHALL MAINSEQUENCE GMBH OR THE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES, OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT, OR OTHERWISE, ARISING FROM, OUT OF, OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+6. Limitation of Liability
+IN NO EVENT SHALL MAINSEQUENCE GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OR INABILITY TO USE THE SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
+
+7. Governing Law
+This License Agreement shall be governed by and construed in accordance with the laws of the jurisdiction where MainSequence GmbH is located, without regard to its conflict of law provisions.

mainsequence-2.0.0/PKG-INFO

@@ -0,0 +1,105 @@
+Metadata-Version: 2.4
+Name: mainsequence
+Version: 2.0.0
+Summary: Main Sequence SDK
+Author-email: Main Sequence GmbH <dev@main-sequence.io>
+License: MainSequence GmbH SDK License Agreement
+        This License Agreement (the "License") governs the use, modification, and distribution of the software provided by MainSequence GmbH (the "Licensor"). The software (the "Software") is provided to you under the terms of this License. By using the Software, you agree to the terms of this License.
+        
+        TERMS AND CONDITIONS
+        0. Definitions
+        "Personal Use": Use by an individual for personal purposes that are not connected to any business, organization, or commercial activity.
+        "Internal Use": Use within a business, organization, or other entity, provided it is not made accessible to third parties or used for commercial purposes.
+        "Commercial Use": Use of the Software in exchange for monetary or other compensation, including hosting, offering Software as a service, selling the Software, or using it in a product or service for sale.
+        "License Agreement": The legally binding agreement between the Licensor and the licensee (you), subject to the terms outlined in this License.
+        1. Grant of License
+        1.1 Personal and Internal Use
+        The Licensor grants you a limited, non-exclusive, non-transferable, revocable license to use and modify the Software for personal or internal use only, provided that such use is strictly subject to this License Agreement and continues only while the License Agreement remains in effect.
+        Upon termination of this License Agreement, all rights to use the Software for personal or internal purposes shall be immediately revoked, and you must cease all use of the Software.
+        1.2 Modification
+        You are permitted to modify the Software solely for your own personal or internal use, subject to the restrictions outlined in this License Agreement.
+        You are not permitted to distribute, sublicense, or otherwise transfer modified or unmodified versions of the Software to any third party.
+        1.3 Prohibited Redistribution
+        You may not redistribute, sublicense, sell, lease, rent, or otherwise transfer the Software, whether in its original form or as modified by you, to any third party.
+        Any attempt to distribute or transfer the Software in any way, without explicit permission from MainSequence GmbH, is strictly prohibited.
+        2. Prohibition of Commercial Use
+        The Software may not be used for any commercial purposes without obtaining a separate commercial license from MainSequence GmbH.
+        Examples of prohibited commercial use include, but are not limited to:
+        Hosting or offering the Software as a service to others, either modified or unmodified.
+        Using the Software as part of a commercial product or service provided to customers for a fee.
+        Using the Software in any production environment that generates income, directly or indirectly, from its use.
+        3. Termination
+        This License Agreement will automatically terminate if you fail to comply with any of its terms.
+        Upon termination of this License, you must immediately cease all use of the Software, destroy all copies (modified or unmodified), and remove the Software from any devices or systems on which it is installed.
+        MainSequence GmbH reserves the right to terminate this License at its discretion for any violation of its terms or for any other reason.
+        4. Commercial License
+        If you wish to use the Software for commercial purposes, you must contact MainSequence GmbH to negotiate and obtain a separate commercial license. The terms of the commercial license, including any fees, will be negotiated separately from this License Agreement.
+        Without a valid commercial license, you are strictly prohibited from using the Software for any commercial activity.
+        5. Disclaimer of Warranty
+        THE SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, AND NONINFRINGEMENT. IN NO EVENT SHALL MAINSEQUENCE GMBH OR THE COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES, OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT, OR OTHERWISE, ARISING FROM, OUT OF, OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+        
+        6. Limitation of Liability
+        IN NO EVENT SHALL MAINSEQUENCE GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES ARISING OUT OF THE USE OR INABILITY TO USE THE SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
+        
+        7. Governing Law
+        This License Agreement shall be governed by and construed in accordance with the laws of the jurisdiction where MainSequence GmbH is located, without regard to its conflict of law provisions.
+Project-URL: Homepage, https://github.com/mainsequence-sdk/mainsequence-sdk
+Project-URL: Issues, https://github.com/mainsequence-sdk/mainsequence-sdk/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: cloudpickle
+Requires-Dist: cryptography
+Requires-Dist: fire
+Requires-Dist: google-auth
+Requires-Dist: joblib
+Requires-Dist: matplotlib
+Requires-Dist: nbconvert
+Requires-Dist: numexpr
+Requires-Dist: opentelemetry-api
+Requires-Dist: opentelemetry-exporter-otlp
+Requires-Dist: opentelemetry-sdk
+Requires-Dist: pandas
+Requires-Dist: pandas-market-calendars
+Requires-Dist: psutil
+Requires-Dist: pyarrow
+Requires-Dist: pydantic
+Requires-Dist: python-binance
+Requires-Dist: python-dotenv
+Requires-Dist: python-logstash
+Requires-Dist: pyyaml
+Requires-Dist: s3fs
+Requires-Dist: scikit-learn
+Requires-Dist: scipy
+Requires-Dist: structlog
+Requires-Dist: tqdm
+Requires-Dist: colorama
+Requires-Dist: kaleido
+Requires-Dist: newspaper3k
+Requires-Dist: lxml-html-clean
+Requires-Dist: docstring-parser
+Requires-Dist: concurrent-log-handler
+Requires-Dist: duckdb
+Requires-Dist: plotly
+Requires-Dist: typer
+Requires-Dist: QuantLib
+Provides-Extra: dev
+Requires-Dist: mkdocs; extra == "dev"
+Requires-Dist: mkdocs-material; extra == "dev"
+Requires-Dist: mkdocs-gen-files; extra == "dev"
+Requires-Dist: mkdocs-literate-nav; extra == "dev"
+Requires-Dist: mkdocs-autorefs; extra == "dev"
+Requires-Dist: mkdocstrings[python]; extra == "dev"
+Requires-Dist: mkdocs-mermaid2-plugin; extra == "dev"
+Requires-Dist: pymdown-extensions; extra == "dev"
+Requires-Dist: pipdeptree; extra == "dev"
+Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-freezegun; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Dynamic: license-file
+
+# Main Sequence Python SDK

mainsequence-2.0.0/README.md

@@ -0,0 +1 @@
+# Main Sequence Python SDK

mainsequence-2.0.0/mainsequence/__main__.py

@@ -0,0 +1,9 @@
+# mainsequence/__main__.py
+from .cli import app
+
+def main():
+    # Typer app is callable
+    app(prog_name="mainsequence")
+
+if __name__ == "__main__":
+    main()

mainsequence-2.0.0/mainsequence/cli/__init__.py

@@ -0,0 +1 @@
+from .cli import app

mainsequence-2.0.0/mainsequence/cli/api.py

@@ -0,0 +1,157 @@
+# mainsequence/cli/api.py
+from __future__ import annotations
+import os, re, subprocess, json, platform, pathlib, shlex, sys
+import requests
+from .config import backend_url, get_tokens, save_tokens, set_env_access
+
+AUTH_PATHS = {
+    "obtain": "/auth/jwt-token/token/",
+    "refresh": "/auth/jwt-token/token/refresh/",
+    "ping": "/auth/rest-auth/user/",
+}
+
+S = requests.Session()
+S.headers.update({"Content-Type": "application/json"})
+
+class ApiError(RuntimeError): ...
+class NotLoggedIn(ApiError): ...
+
+def _full(path: str) -> str:
+    p = "/" + path.lstrip("/")
+    return backend_url() + p
+
+def _normalize_api_path(p: str) -> str:
+    p = "/" + (p or "").lstrip("/")
+    if not re.match(r"^/(api|auth|pods|orm|user)(/|$)", p):
+        raise ApiError("Only /api/*, /auth/*, /pods/*, /orm/*, /user/* allowed")
+    return p
+
+def _access_token() -> str | None:
+    t = os.environ.get("MAIN_SEQUENCE_USER_TOKEN")
+    if t:
+        return t
+    tok = get_tokens()
+    return tok.get("access")
+
+def _refresh_token() -> str | None:
+    tok = get_tokens()
+    return tok.get("refresh")
+
+def login(email: str, password: str) -> dict:
+    url = _full(AUTH_PATHS["obtain"])
+    payload = {"email": email, "password": password}  # server expects 'email'
+    r = S.post(url, data=json.dumps(payload))
+    try:
+        data = r.json()
+    except Exception:
+        data = {}
+    if not r.ok:
+        msg = data.get("detail") or data.get("message") or r.text
+        raise ApiError(f"{msg}")
+    access = data.get("access") or data.get("token") or data.get("jwt") or data.get("access_token")
+    refresh = data.get("refresh") or data.get("refresh_token")
+    if not access or not refresh:
+        raise ApiError("Server did not return expected tokens.")
+    save_tokens(email, access, refresh)
+    set_env_access(access)
+    return {"username": email, "backend": backend_url()}
+
+def refresh_access() -> str:
+    refresh = _refresh_token()
+    if not refresh:
+        raise NotLoggedIn("Not logged in. Run `mainsequence login <email>`.")
+    r = S.post(_full(AUTH_PATHS["refresh"]), data=json.dumps({"refresh": refresh}))
+    data = r.json() if r.headers.get("content-type","").startswith("application/json") else {}
+    if not r.ok:
+        raise NotLoggedIn(data.get("detail") or "Token refresh failed.")
+    access = data.get("access")
+    if not access:
+        raise NotLoggedIn("Refresh succeeded but no access token returned.")
+    tokens = get_tokens()
+    save_tokens(tokens.get("username") or "", access, refresh)
+    set_env_access(access)
+    return access
+
+def authed(method: str, api_path: str, body: dict | None = None) -> requests.Response:
+    api_path = _normalize_api_path(api_path)
+    access = _access_token()
+    if not access:
+        # try to refresh once
+        access = refresh_access()
+    headers = {"Authorization": f"Bearer {access}"}
+    r = S.request(method.upper(), _full(api_path), headers=headers,
+                  data=None if method.upper() in {"GET","HEAD"} else json.dumps(body or {}))
+    if r.status_code == 401:
+        # retry after refresh
+        access = refresh_access()
+        headers = {"Authorization": f"Bearer {access}"}
+        r = S.request(method.upper(), _full(api_path), headers=headers,
+                      data=None if method.upper() in {"GET","HEAD"} else json.dumps(body or {}))
+    if r.status_code == 401:
+        raise NotLoggedIn("Not logged in.")
+    return r
+
+# ---------- Helper APIs ----------
+
+def safe_slug(s: str) -> str:
+    x = re.sub(r"[^a-z0-9-_]+", "-", (s or "project").lower()).strip("-")
+    return x[:64] or "project"
+
+def repo_name_from_git_url(url: str | None) -> str | None:
+    if not url: return None
+    s = re.sub(r"[?#].*$", "", url.strip())
+    last = s.split("/")[-1] if "/" in s else s
+    if last.lower().endswith(".git"): last = last[:-4]
+    return re.sub(r"[^A-Za-z0-9._-]+", "-", last)
+
+def deep_find_repo_url(extra) -> str | None:
+    if not isinstance(extra, dict): return None
+    cand = ["ssh_url","git_ssh_url","repo_ssh_url","git_url","repo_url","repository","url"]
+    for k in cand:
+        v = extra.get(k)
+        if isinstance(v, str) and (v.startswith("git@") or re.search(r"\.git($|\?)", v)):
+            return v
+        if isinstance(v, dict):
+            for vv in v.values():
+                if isinstance(vv, str) and (vv.startswith("git@") or re.search(r"\.git($|\?)", vv)):
+                    return vv
+    for v in extra.values():
+        if isinstance(v, dict):
+            found = deep_find_repo_url(v)
+            if found: return found
+    return None
+
+def get_current_user_profile() -> dict:
+    who = authed("GET", AUTH_PATHS["ping"])
+    d = who.json() if who.ok else {}
+    uid = d.get("id") or d.get("pk") or (d.get("user") or {}).get("id") or d.get("user_id")
+    if not uid:
+        return {}
+    full = authed("GET", f"/user/api/user/{uid}/")
+    u = full.json() if full.ok else {}
+    org_name = (u.get("organization") or {}).get("name") or u.get("organization_name") or ""
+    return {"username": u.get("username") or "", "organization": org_name}
+
+def get_projects() -> list[dict]:
+    r = authed("GET", "/orm/api/pods/projects/")
+    # If the API shape ever changes, still try to pull a list.
+    if not r.ok:
+        raise ApiError(f"Projects fetch failed ({r.status_code}).")
+    data = r.json() if r.headers.get("content-type","",).startswith("application/json") else {}
+    if isinstance(data, list):
+        return data
+    return data.get("results") or []
+
+def fetch_project_env_text(project_id: int | str) -> str:
+    r = authed("GET", f"/orm/api/pods/projects/{project_id}/get_environment/")
+    raw = r.json() if r.headers.get("content-type","").startswith("application/json") else r.text
+    if isinstance(raw, dict):
+        raw = raw.get("environment") or raw.get("env") or raw.get("content") or raw.get("text") or ""
+    return (raw or "")
+
+def add_deploy_key(project_id: int | str, key_title: str, public_key: str) -> None:
+    try:
+        authed("POST", f"/orm/api/pods/projects/{project_id}/add_deploy_key/",
+               {"key_title": key_title, "public_key": public_key})
+    except Exception:
+        pass