qmenta-core 4.1.dev708__py3-none-any.whl → 4.1.dev710__py3-none-any.whl

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/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

{qmenta_core-4.1.dev708.dist-info → qmenta_core-4.1.dev710.dist-info}/METADATA

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

{qmenta_core-4.1.dev708.dist-info → qmenta_core-4.1.dev710.dist-info}/RECORD

@@ -1,15 +1,15 @@
 qmenta/__init__.py,sha256=jv2YF__bseklT3OWEzlqJ5qE24c4aWd5F4r0TTjOrWQ,65
 qmenta/core/.gitignore,sha256=T_6cEPGYAfQ6-gdGwpLCLe1SpRnTl4vwi4qfVM1QsYI,8
 qmenta/core/__init__.py,sha256=J_Se74WI6plv-7tRufJDsXpgxG_WRX4pwLvK-Hhq4Ls,304
-qmenta/core/auth.py,sha256=xiamkcIaQQPD1OppLScv0dxF0PhAjqQOA3DqzUv79l4,10026
+qmenta/core/auth.py,sha256=4siYjRuoNgjYvpkBbiV-llkCdKMX-IS8S2h3kHdLPvs,10590
 qmenta/core/errors.py,sha256=eu72sENP9YUVN1w9Urji1nH0V41pdg4UiSvxjun6ggE,1145
-qmenta/core/platform.py,sha256=Gd-_8SmTNNcPqIM_wNG_EQE-3hIg49xKeeG0TxxslT0,5504
+qmenta/core/platform.py,sha256=5nKxE2xwvwMRgNAHcCPVscgX8LHdb5lAP-VTBzfKZcM,5503
 qmenta/core/upload/__init__.py,sha256=-VHwhSQkdUmw697v4o59iDiqdU6kh-8_3BklGp4W864,182
 qmenta/core/upload/multi.py,sha256=3XiGxguP7V2zrGyKsaQx6xarbsVr-wUINkgLZJFQbno,4963
 qmenta/core/upload/prepare.py,sha256=JcFSl8RRCsNnzcNvn7-d7ARUfZvvEJhuCuW5VbMjmpI,8139
 qmenta/core/upload/single.py,sha256=FE4ZucI3qIEfzqhZKQPOKremzez7CjMRS28t-tsep4E,19356
 qmenta/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-qmenta_core-4.1.dev708.dist-info/METADATA,sha256=HFtZoFaFQfIA6GFEdEl7hhGcnMpLFgTOip60SgRgLYI,1155
-qmenta_core-4.1.dev708.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
-qmenta_core-4.1.dev708.dist-info/entry_points.txt,sha256=8c0Nh22PxTp8Auui-k8qIinhdaotgc-wulDC3-PC92c,53
-qmenta_core-4.1.dev708.dist-info/RECORD,,
+qmenta_core-4.1.dev710.dist-info/METADATA,sha256=Xr5TlwfWApkknDvofPwmI_yztZEgs36EmPv_MZ8EZNE,1155
+qmenta_core-4.1.dev710.dist-info/WHEEL,sha256=fGIA9gx4Qxk2KDKeNJCbOEwSrmLtjWCwzBz351GyrPQ,88
+qmenta_core-4.1.dev710.dist-info/entry_points.txt,sha256=8c0Nh22PxTp8Auui-k8qIinhdaotgc-wulDC3-PC92c,53
+qmenta_core-4.1.dev710.dist-info/RECORD,,