qmenta-core 4.1.dev708__tar.gz → 4.1.dev709__tar.gz

{qmenta_core-4.1.dev708 → qmenta_core-4.1.dev709}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: qmenta-core
-Version: 4.1.dev708
+Version: 4.1.dev709
 Summary: QMENTA core library to communicate with the QMENTA platform.
 License: Proprietary
 Author: QMENTA

{qmenta_core-4.1.dev708 → qmenta_core-4.1.dev709}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "qmenta-core"
-version = "4.1.dev708"
+version = "4.1.dev709"
 description = "QMENTA core library to communicate with the QMENTA platform."
 license = "Proprietary"
 authors = ["QMENTA <dev@qmenta.com>"]

{qmenta_core-4.1.dev708 → qmenta_core-4.1.dev709}/src/qmenta/core/auth.py

@@ -1,4 +1,5 @@
 import argparse
+from enum import Enum, unique
 import os
 import requests
 from getpass import getpass
@@ -32,6 +33,15 @@ class Needs2FAError(ActionFailedError):
     pass
 
 
+@unique
+class PlatformURL(Enum):
+    platform = 'platform.qmenta.com'
+    staging = 'staging.qmenta.com'
+    test = 'test.qmenta.com'
+    local_ip = "127.0.0.1:8080"
+    localhost = "localhost:8080"
+
+
 class Auth:
     """
     Class for authenticating to the platform.
@@ -41,11 +51,13 @@ class Auth:
     Attributes
     ----------
     base_url : str
-        The base URL of the platform. Example: 'https://platform.qmenta.com'
+        The URL of the platform to connect to.
+        Default value: 'https://platform.qmenta.com'
     token : str
         The authentication token, returned by the platform when logging in.
     """
     def __init__(self, base_url: str, token: str) -> None:
+        validate_url(base_url)
         self.base_url = base_url
         self.token = token
         self._session: Optional[requests.Session] = None
@@ -101,7 +113,7 @@ class Auth:
     def login(cls, username: str, password: str,
               code_2fa: Optional[str] = None,
               ask_for_2fa_input: bool = False,
-              base_url: str = 'https://platform.qmenta.com') -> 'Auth':
+              base_url: str = PlatformURL.platform.value) -> 'Auth':
         """
         Authenticate to the platform using username and password.
 
@@ -232,6 +244,8 @@ def write_dot_env_file(token: str, url: str,
     OSError
         When the output file could not be written
     """
+    validate_url(url)
+
     filepath = filename or Auth.qmenta_auth_env_file()
     os.makedirs(os.path.dirname(os.path.abspath(filepath)), exist_ok=True)
 
@@ -251,6 +265,7 @@ def validate_url(url: str) -> None:
     ValueError
         When the URL is not valid
     """
+
     parsed_url = urlparse(url)
 
     if parsed_url.scheme == 'http':
@@ -271,6 +286,13 @@ def validate_url(url: str) -> None:
             'Example: https://platform.qmenta.com'
         )
 
+    url_values = [p_url.value for p_url in PlatformURL]
+    if parsed_url.netloc not in url_values:
+        raise ValueError(
+            "base_url must be one of '{}', ".format("', '".join(url_values))
+            + f"not '{parsed_url.netloc}'."
+        )
+
 
 def main() -> None:
     parser = argparse.ArgumentParser(

{qmenta_core-4.1.dev708 → qmenta_core-4.1.dev709}/src/qmenta/core/platform.py

@@ -29,8 +29,8 @@ class ChooseDataError(ActionFailedError):
     """
 
     def __init__(
-        self, warning: str, data_to_choose: Dict[str, List[str]]
-        , analysis_id: int
+        self, warning: str, data_to_choose: Dict[str, List[str]],
+        analysis_id: int
     ) -> None:
         self.warning: str = warning
         self.data_to_choose: dict = data_to_choose