hardpy 0.5.0__tar.gz → 0.5.1__tar.gz

{hardpy-0.5.0 → hardpy-0.5.1}/PKG-INFO

@@ -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.1/hardpy/pytest_hardpy/result/couchdb_config.py

@@ -0,0 +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.
+
+    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.
+
+        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}"
+        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 → hardpy-0.5.1}/pyproject.toml

@@ -3,7 +3,7 @@
 
 [project]
     name = "hardpy"
-    version = "0.5.0"
+    version = "0.5.1"
     description = "HardPy library for device testing"
     authors = [{ name = "Everypin" }]
     readme = "README.md"

hardpy-0.5.0/hardpy/pytest_hardpy/result/couchdb_config.py

@@ -1,22 +0,0 @@
-# Copyright (c) 2024 Everypin
-# GNU General Public License v3.0 (see LICENSE or https://www.gnu.org/licenses/gpl-3.0.txt)
-
-from dataclasses import dataclass
-
-
-@dataclass
-class CouchdbConfig:  # noqa: WPS306
-    """CouchDB loader config."""
-
-    db_name: str = "report"
-    user: str = "dev"
-    password: str = "dev"
-    host: str = "localhost"
-    port: int = 5984
-
-    @property
-    def connection_string(self) -> str:
-        """Get couchdb connection string."""
-        credentials = f"{self.user}:{self.password}"
-        uri = f"{self.host}:{str(self.port)}"
-        return f"http://{credentials}@{uri}/"