tft-cli 0.0.25__py3-none-any.whl → 0.0.28__py3-none-any.whl

tft/cli/utils.py

@@ -4,18 +4,24 @@
 import glob
 import itertools
 import os
+import re
 import shlex
+import shutil
 import subprocess
 import sys
+import tempfile
 import uuid
+from dataclasses import dataclass
+from enum import Enum
 from typing import Any, Dict, List, NoReturn, Optional, Union
 
+import pendulum
 import requests
 import requests.adapters
 import typer
-from click.core import ParameterSource  # pyre-ignore[21]
+from click.core import ParameterSource
 from rich.console import Console
-from ruamel.yaml import YAML
+from ruamel.yaml import YAML  # type: ignore
 from urllib3 import Retry
 
 from tft.cli.config import settings
@@ -24,6 +30,56 @@ console = Console(soft_wrap=True)
 console_stderr = Console(soft_wrap=True, file=sys.stderr)
 
 
+@dataclass
+class Age:
+    value: int
+    unit: str
+
+    _unit_multiplier = {"s": 1, "m": 60, "h": 3600, "d": 86400}
+    _unit_human = {"s": "second", "m": "minute", "h": "hour", "d": "day"}
+
+    @classmethod
+    def from_string(cls, age_string: str) -> 'Age':
+        value, unit = age_string[:-1], age_string[-1]
+        if unit not in cls._unit_multiplier:
+            raise typer.BadParameter(f"Age must end with {', '.join(cls._unit_multiplier.keys())}")
+
+        if not value.isdigit():
+            raise typer.BadParameter(f"Invalid age value {value}")
+
+        return cls(value=int(age_string[:-1]), unit=age_string[-1])
+
+    @property
+    def birth_date(self) -> pendulum.DateTime:
+        now = pendulum.now(tz="UTC")
+        return now - pendulum.duration(seconds=self.value * self._unit_multiplier[self.unit])
+
+    @property
+    def human(self) -> str:
+        return f"{self.value} {self._unit_human[self.unit]}{'s' if self.value > 1 else ''}"
+
+    @staticmethod
+    def available_units() -> str:
+        return "s (seconds), m (minutes), h (hours) or d (days)"
+
+    def to_string(self, format="%Y-%m-%dT%H:%M:%S") -> str:
+        return self.birth_date.strftime(format)
+
+    def __str__(self):
+        return f"{self.value}{self.unit}"
+
+
+class OutputFormat(str, Enum):
+    text = "text"
+    json = "json"
+    yaml = "yaml"
+    table = "table"
+
+    @staticmethod
+    def available_formats():
+        return "text, json or table"
+
+
 def exit_error(error: str) -> NoReturn:
     """Exit with given error message"""
     console.print(f"⛔ {error}", style="red")
@@ -125,6 +181,19 @@ def hw_constraints(hardware: List[str]) -> Dict[Any, Any]:
             value_mixed = False
 
         container[path_splitted.pop()] = value_mixed
+
+        # Only process additional path elements if they exist
+        if path_splitted:
+            next_path = path_splitted.pop()
+
+            # handle compatible.distro
+            if next_path == 'distro':
+                container[next_path] = (
+                    container[next_path].append(value_mixed) if next_path in container else [value_mixed]
+                )
+            else:
+                container[next_path] = value_mixed
+
     return constraints
 
 
@@ -187,6 +256,27 @@ def uuid_valid(value: str, version: int = 4) -> bool:
         return False
 
 
+def extract_uuid(value: str) -> str:
+    """
+    Extracts a UUID from a string. If the string is already a valid UUID, returns it.
+    If the string contains a UUID, extracts and returns it.
+    Raises typer.Exit with error message if no valid UUID is found.
+    """
+    # Check if the value is already a valid UUID
+    if uuid_valid(value):
+        return value
+
+    # UUID pattern for extracting UUIDs from strings
+    uuid_pattern = re.compile('[0-9a-f]{8}-[0-9a-f]{4}-[1-5][0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}')
+
+    # Try to extract UUID from string
+    uuid_match = uuid_pattern.search(value)
+    if uuid_match:
+        return uuid_match.group()
+
+    exit_error(f"Could not find a valid Testing Farm request id in '{value}'.")
+
+
 class TimeoutHTTPAdapter(requests.adapters.HTTPAdapter):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         self.timeout = kwargs.pop('timeout', settings.DEFAULT_API_TIMEOUT)
@@ -216,7 +306,9 @@ def install_http_retries(
 
     status_forcelist_extend = status_forcelist_extend or []
 
-    params = {
+    from typing import Any, Dict
+
+    params: Dict[str, Any] = {
         "total": retries,
         "status_forcelist": [
             429,  # Too Many Requests
@@ -226,9 +318,9 @@ def install_http_retries(
             504,  # Gateway Timeout
         ]
         + status_forcelist_extend,
-        allowed_retry_parameter: ['HEAD', 'GET', 'POST', 'DELETE', 'PUT'],
         "backoff_factor": retry_backoff_factor,
     }
+    params[allowed_retry_parameter] = ['HEAD', 'GET', 'POST', 'DELETE', 'PUT']
     retry_strategy = Retry(**params)
 
     timeout_adapter = TimeoutHTTPAdapter(timeout=timeout, max_retries=retry_strategy)
@@ -263,8 +355,60 @@ def read_glob_paths(glob_paths: List[str]) -> str:
 
 def check_unexpected_arguments(context: typer.Context, *args: str) -> Union[None, NoReturn]:
     for argument in args:
-        if context.get_parameter_source(argument) == ParameterSource.COMMANDLINE:  # pyre-ignore[16]
+        if context.get_parameter_source(argument) == ParameterSource.COMMANDLINE:
             exit_error(
                 f"Unexpected argument '{context.params.get(argument)}'. "
                 "Please make sure you are passing the parameters correctly."
             )
+
+
+def validate_age(value: str) -> Age:
+    if value.endswith("m"):
+        return Age(int(value[:-1]), "m")
+    elif value.endswith("d"):
+        return Age(int(value[:-1]), "d")
+    else:
+        raise ValueError("Age must end with 'm' for months or 'd' for days.")
+
+
+def authorization_headers(api_key: str) -> Dict[str, str]:
+    """
+    Return a dict with headers for a request to Testing Farm API.
+    Used for authentication.
+    """
+    return {'Authorization': f'Bearer {api_key}'}
+
+
+def edit_with_editor(data: Any, description: Optional[str]) -> Any:
+    """
+    Open data in an editor for user modification and return it back.
+    If description specified, print it as a user message together with the used editor.
+    """
+    # Get the editor from environment variable, fallback to sensible defaults
+    editor = os.environ.get('EDITOR')
+    if not editor:
+        # Try common editors in order of preference
+        for candidate in ['vim', 'vi', 'nano', 'emacs']:
+            if shutil.which(candidate):
+                editor = candidate
+                break
+
+    if not editor:
+        exit_error("No editor found. Please set the 'EDITOR' environment variable.")
+
+    # Create a temporary file with the JSON content
+    with tempfile.NamedTemporaryFile(mode='w') as temp_file:
+        temp_file.write(data)
+        temp_file.flush()
+
+        # Open the editor
+        if description:
+            console.print(f"✏️  {description}, editor '{editor}'")
+        result = subprocess.run([editor, temp_file.name])
+
+        if result.returncode != 0:
+            exit_error(f"Editor '{editor}' exited with non-zero status: {result.returncode}")
+
+        # Read the modified content
+        with open(temp_file.name, 'r') as modified_file:
+            return modified_file.read()

{tft_cli-0.0.25.dist-info → tft_cli-0.0.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tft-cli
-Version: 0.0.25
+Version: 0.0.28
 Summary: Testing Farm CLI tool
 License: Apache-2.0
 Author: Miroslav Vadkerti
@@ -14,6 +14,7 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Dist: click (>=8.0.4,<9.0.0)
 Requires-Dist: colorama (>=0.4.4,<0.5.0)
 Requires-Dist: dynaconf (>=3.1.2,<4.0.0)
+Requires-Dist: pendulum (>=3.0.0,<4.0.0)
 Requires-Dist: requests (>=2.27.1,<3.0.0)
 Requires-Dist: rich (>=12)
 Requires-Dist: ruamel-yaml (>=0.18.5,<0.19.0)

tft_cli-0.0.28.dist-info/RECORD

@@ -0,0 +1,12 @@
+tft/cli/__init__.py,sha256=uEJkNJbqC583PBtNI30kxWdeOr3Wj6zJzIYKf0AD72I,92
+tft/cli/command/__init__.py,sha256=uEJkNJbqC583PBtNI30kxWdeOr3Wj6zJzIYKf0AD72I,92
+tft/cli/command/listing.py,sha256=gEkoeKHOcEiBt2TyLzSAongOKHKi7BgEO7GQ0AAa0Ss,31276
+tft/cli/commands.py,sha256=aZ_P7B9z4IorT6ksQH0ciCY6KxvL5q5NJdWuHNv9fHE,85890
+tft/cli/config.py,sha256=9JWjEwBiS4_Eq1B53lxxCIceERJ7AeryUyqxj9fm9c4,1714
+tft/cli/tool.py,sha256=AaJ7RZwWEoP7WA_f2NbmcZSolNETkTWvVAt9pgt9j-o,975
+tft/cli/utils.py,sha256=j8B30qpieBUieaiVo_3h5bRqoKnGeGKxR9eJ7GZcNiM,13702
+tft_cli-0.0.28.dist-info/LICENSE,sha256=YpVAQfXkIyzQAdm5LZkI6L5UWqLppa6O8_tgDSdoabQ,574
+tft_cli-0.0.28.dist-info/METADATA,sha256=MybsQ9HApc32-yILZ14mMu8YcWCSlhtH0gioJoMI8ZM,830
+tft_cli-0.0.28.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+tft_cli-0.0.28.dist-info/entry_points.txt,sha256=xzdebHkH5Bx-YRf-XPMsIoVpvgfUqqcRQGuo8DFkiao,49
+tft_cli-0.0.28.dist-info/RECORD,,

tft_cli-0.0.25.dist-info/RECORD

@@ -1,10 +0,0 @@
-tft/cli/__init__.py,sha256=uEJkNJbqC583PBtNI30kxWdeOr3Wj6zJzIYKf0AD72I,92
-tft/cli/commands.py,sha256=5fcoNlJ_K1r6j99qoWKh9hJM2nJ3xY1Tp6vL4wcdQxk,82322
-tft/cli/config.py,sha256=JiVLrgM4REWdljA9DjAuH4fNR1ekE7Crv2oM6vBPt9Q,1254
-tft/cli/tool.py,sha256=nuz57u3yE4fKgdMgNtAFM7PCTcF6PSNFBQbCYDzxBOw,897
-tft/cli/utils.py,sha256=t3ZSnviGxVbBQ4u4ltwQwRgN647BvuyL1QrSlQAAT1Q,9136
-tft_cli-0.0.25.dist-info/LICENSE,sha256=YpVAQfXkIyzQAdm5LZkI6L5UWqLppa6O8_tgDSdoabQ,574
-tft_cli-0.0.25.dist-info/METADATA,sha256=tIo9HRQmpNjq58ru63CRTun0owhYGucYQDFuWhB_fFE,789
-tft_cli-0.0.25.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-tft_cli-0.0.25.dist-info/entry_points.txt,sha256=xzdebHkH5Bx-YRf-XPMsIoVpvgfUqqcRQGuo8DFkiao,49
-tft_cli-0.0.25.dist-info/RECORD,,