hardpy 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

hardpy/pytest_hardpy/result/couchdb_config.py

@@ -1,22 +1,96 @@
 # Copyright (c) 2024 Everypin
 # GNU General Public License v3.0 (see LICENSE or https://www.gnu.org/licenses/gpl-3.0.txt)
 
+import ast
+import requests
+import socket
+
 from dataclasses import dataclass
+from urllib3 import disable_warnings
 
 
 @dataclass
 class CouchdbConfig:  # noqa: WPS306
-    """CouchDB loader config."""
+    """CouchDB loader config.
+
+    If `connection_str` arg is not set, it will be created from other args.
+    """
 
     db_name: str = "report"
     user: str = "dev"
     password: str = "dev"
     host: str = "localhost"
     port: int = 5984
+    connection_str: str | None = None
+
+    def __post_init__(self):
+        """Disable urllib3 warnings.
+
+        More info: https://urllib3.readthedocs.io/en/latest/advanced-usage.html#tls-warnings
+        """
+        disable_warnings()
 
     @property
     def connection_string(self) -> str:
-        """Get couchdb connection string."""
+        """Get couchdb connection string.
+
+        Raises:
+            RuntimeError: CouchDB server is not available
+
+        Returns:
+            str: Database connection string.
+        """
+        if self.connection_str:
+            return self.connection_str
+
+        # TODO: Modify connection string creating based on protocol.
+        #       Some problems with http and https, different ports, local
+        #       and cloud databases.
+        protocol = self._get_protocol()
+
+        if protocol == "http":
+            host_url = f"http://{self.host}:{self.port}"
+            uri = f"{self.host}:{str(self.port)}"  # noqa: WPS237
+        elif protocol == "https":
+            host_url = f"https://{self.host}"
+            uri = f"{self.host}"
+
+        try:
+            response = requests.get(host_url, timeout=5)
+        except requests.exceptions.RequestException:
+            raise RuntimeError(f"Error CouchDB connecting to {host_url}.")
+
+        # fmt: off
+        try:
+            couchdb_dict = ast.literal_eval(response._content.decode("utf-8"))  # noqa: WPS437,E501
+            couchdb_dict.get("couchdb", False)
+        except Exception:
+            raise RuntimeError(f"Address {host_url} does not provide CouchDB attributes.")
+        # fmt: on
+
         credentials = f"{self.user}:{self.password}"
-        uri = f"{self.host}:{str(self.port)}"
-        return f"http://{credentials}@{uri}/"
+        return f"{protocol}://{credentials}@{uri}/"
+
+    def _get_protocol(self) -> str:  # noqa: WPS231
+        success = 200
+        try:
+            # HTTPS attempt
+            sock = socket.create_connection((self.host, self.port))
+            sock.close()
+            request = f"https://{self.host}"
+            if requests.get(request, timeout=5).status_code == success:
+                return "https"
+            raise OSError
+        except OSError:
+            try:  # noqa: WPS505
+                # HTTP attempt
+                sock = socket.create_connection((self.host, self.port))
+                sock.close()
+                request = f"http://{self.host}:{self.port}"
+                if requests.get(request, timeout=5).status_code == success:
+                    return "http"
+                raise OSError
+            except OSError:
+                raise RuntimeError(
+                    f"Error connecting to couchdb server {self.host}:{self.port}."
+                )

{hardpy-0.5.0.dist-info → hardpy-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hardpy
-Version: 0.5.0
+Version: 0.5.1
 Summary: HardPy library for device testing
 Project-URL: Homepage, https://github.com/everypinio/hardpy/
 Project-URL: Documentation, https://everypinio.github.io/hardpy/

{hardpy-0.5.0.dist-info → hardpy-0.5.1.dist-info}/RECORD

@@ -52,7 +52,7 @@ hardpy/pytest_hardpy/reporter/base.py,sha256=M-lwli64ty9FW8HlGEpUyoFsZv48tyNgzPj
 hardpy/pytest_hardpy/reporter/hook_reporter.py,sha256=kZ3jx--ne9l3a27TmVdZCPBoqy53EuHYg2fHDGuJNes,10068
 hardpy/pytest_hardpy/reporter/runner_reporter.py,sha256=NXkBIoERqmLI-GYtHavmOWC5t6NIpcAE-NECrUKIAJs,827
 hardpy/pytest_hardpy/result/__init__.py,sha256=NMeCGx3yh8ds9VpaUpuNFDxbwgYFq3e-o7W6rYIv8uI,346
-hardpy/pytest_hardpy/result/couchdb_config.py,sha256=QZryfA2QoHIjzbVT3OAD76DCNppCghtRWdZMZ5v7KhY,611
+hardpy/pytest_hardpy/result/couchdb_config.py,sha256=kttsrWzv9D99R3ms-hFbzVbrZ78B2me6rrJgncfs5jk,3129
 hardpy/pytest_hardpy/result/report_loader/__init__.py,sha256=FuHuD6IFZyaKj0yu5urhf6nkxGbiPONJ-WHnJ2jHwyc,251
 hardpy/pytest_hardpy/result/report_loader/couchdb_loader.py,sha256=1BGncK1NnwYJanm5TYETjMkqR7tmrUfGK8Sgnp2JZC8,2203
 hardpy/pytest_hardpy/result/report_reader/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -65,8 +65,8 @@ hardpy/pytest_hardpy/utils/exception.py,sha256=khifykRuFS7GyIyQcJeMg6VOgwhPPvzNq
 hardpy/pytest_hardpy/utils/node_info.py,sha256=VnEbhKBNAL5xpuFtJTCg90TmkjkFCQA59F5W2RcOlx4,3157
 hardpy/pytest_hardpy/utils/progress_calculator.py,sha256=r0qb3p6_yDIyLeCshF3Ceo5pCzd3BoTahL4rCD2oMNw,1041
 hardpy/pytest_hardpy/utils/singleton.py,sha256=C8cgRDydnG2b5dcN1LCLw4aM-AUMAvJc1W39mTkNWlQ,614
-hardpy-0.5.0.dist-info/METADATA,sha256=uzKt39irvwMS6Y73kXPR1JRW-VEEfg1tEszfUTvPtiM,4305
-hardpy-0.5.0.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
-hardpy-0.5.0.dist-info/entry_points.txt,sha256=q73g5GfznSUpjkayi0SV4uaAtrf7D-7rmDoWoEZmZe0,120
-hardpy-0.5.0.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-hardpy-0.5.0.dist-info/RECORD,,
+hardpy-0.5.1.dist-info/METADATA,sha256=rqydaDWQCSC8_UiPYkz6uF0_C475CleUaREiDWbX1DY,4305
+hardpy-0.5.1.dist-info/WHEEL,sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc,87
+hardpy-0.5.1.dist-info/entry_points.txt,sha256=q73g5GfznSUpjkayi0SV4uaAtrf7D-7rmDoWoEZmZe0,120
+hardpy-0.5.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+hardpy-0.5.1.dist-info/RECORD,,