Rewrite [UUID tool]

Author: rexzhang

afw.py

@@ -1,9 +1,11 @@
 #!/usr/bin/env python
 
+import dataclasses
 import importlib
 import json
 import subprocess
 import sys
+from logging import Logger
 from pathlib import Path
 from uuid import uuid4
 from zipfile import ZipFile
@@ -13,6 +15,8 @@
 
 from afw_runtime import afw_responses_to_feedback
 
+logger = Logger(__file__)
+
 AFW_RUNTIME_FILES = ["afw.py", "afw_runtime.py"]
 AFW_REQUIREMENTS = ["click", "ualfred"]
 
@@ -107,7 +111,7 @@ def afw_entry(workflow):
     # This is also necessary for "magic" arguments to work.
     # args = wf.args
 
-    args = [workflow.args[1]]
+    args = workflow.args[1:]
 
     # Do stuff here ...
     from main import main as afw_workflow_main
@@ -147,15 +151,7 @@ def test(workflow_path: str, query: str):
         print(e)
         exit(1)
 
-    from logging import Logger
-    import dataclasses
-    import json
-
-    logger = Logger(__file__)
-
     wf = Workflow3()
-    print(wf.args)
-    print(wf.args[2:])
 
     responses = module.main(args=wf.args[2:], logger=logger)
     for response in responses:

afw_runtime.py

@@ -65,7 +65,7 @@ class AFWResponse:
     title: str
     arg: str
 
-    subtitle: Optional[str] = None
+    subtitle: str | None = None
     icon: str = ICON_NOTE
     valid: bool = False
 
@@ -81,3 +81,81 @@ def afw_responses_to_feedback(responses: list[AFWResponse]) -> list[dict]:
         result.append(data)
 
     return result
+
+
+class AFWFuncAbc:
+    def __init__(self, args: list[str], logger) -> None:
+        self.args = args
+        self.logger = logger
+
+    @property
+    def _data_defaulte(self) -> list:
+        raise NotImplementedError
+
+    @property
+    def _data_tips(self) -> list:
+        raise NotImplementedError
+
+    @property
+    def _data_success(self) -> list:
+        raise NotImplementedError
+
+    @property
+    def _data_error(self) -> list:
+        raise NotImplementedError
+
+    def get_response_success(self) -> list[AFWResponse]:
+        return [
+            AFWResponse(
+                title=title, arg=title, subtitle=subtitle, icon=ICON_INFO, valid=True
+            )
+            for title, subtitle in self._data_success
+        ] + [
+            AFWResponse(
+                title=title, arg=title, subtitle=subtitle, icon=ICON_NOTE, valid=False
+            )
+            for title, subtitle in self._data_tips
+        ]
+
+    def get_response_fail(self) -> list[AFWResponse]:
+        return (
+            [
+                AFWResponse(
+                    title=title,
+                    arg=title,
+                    subtitle=subtitle,
+                    icon=ICON_ERROR,
+                    valid=False,
+                )
+                for title, subtitle in self._data_error
+            ]
+            + [
+                AFWResponse(
+                    title=title,
+                    arg=title,
+                    subtitle=subtitle,
+                    icon=ICON_INFO,
+                    valid=True,
+                )
+                for title, subtitle in self._data_defaulte
+            ]
+            + [
+                AFWResponse(
+                    title=title,
+                    arg=title,
+                    subtitle=subtitle,
+                    icon=ICON_NOTE,
+                    valid=False,
+                )
+                for title, subtitle in self._data_tips
+            ]
+        )
+
+    def _process(self) -> bool:
+        raise NotImplementedError
+
+    def __call__(self) -> list[AFWResponse]:
+        if self._process():
+            return self.get_response_success()
+
+        return self.get_response_fail()

afw_uuid_tool/__init__.py

@@ -0,0 +1 @@
+from .main import main

afw_uuid_tool/main.py

@@ -1,31 +1,59 @@
 import uuid
 
-from afw_runtime import *
+from afw_runtime import AFWFuncAbc, AFWResponse
 
-DEFAULT_RESPONSE = [
-    AFWResponse(title="Please enter a UUID like string", arg="", icon=ICON_HELP),
-    AFWResponse(title="Examples: 1111", arg="", icon=ICON_HELP),
-]
 
+class AFWFunc(AFWFuncAbc):
+    _u: uuid.UUID
+    _message_error: str
 
-def main(args: list[str], logger) -> list[AFWResponse]:
-    logger.debug("test..")
-    query = args[0]
-    logger.debug(query)
-
-    responses = list()
-    try:
-        u = uuid.UUID(query)
-    except ValueError as e:
-        u = None
-        v = str(uuid.uuid4()).upper()
-        responses += [
-            AFWResponse(title=v, arg=v, valid=True),
-            AFWResponse(title=str(e), arg="", icon=ICON_ERROR),
+    @staticmethod
+    def _make_data_from_uuid(u: uuid.UUID) -> list[AFWResponse]:
+        return (
+            (u.hex, "32-character lowercase hexadecimal string."),
+            (str(u), "32-character lowercase hexadecimal string with dash."),
+            # (str(u).upper(), ""),
+            (str(u.int), "The UUID as a 128-bit integer."),
+            (str(u.urn), "The UUID as a URN as specified in RFC 4122."),
+            (
+                str(u.version),
+                "The UUID version number (1 through 5, specified is RFC_4122).",
+            ),
+        )
+
+    @property
+    def _data_defaulte(self) -> list:
+        u = uuid.uuid4()
+        return self._make_data_from_uuid(u)
+
+    @property
+    def _data_tips(self) -> list:
+        return list()
+
+    @property
+    def _data_success(self) -> list:
+        return self._make_data_from_uuid(self._u)
+
+    @property
+    def _data_error(self) -> list:
+        return [
+            (self._message_error, ""),
+            # ("Please enter a UUID like string", ""),
+            # ("Examples: 1111", ""),
         ]
-        responses += DEFAULT_RESPONSE
 
-    if u is None:
-        return responses
+    def _process(self) -> list[AFWResponse]:
+        try:
+            if len(self.args) == 0:
+                raise ValueError(" Missing argument 'QUERY'")
 
-    return [AFWResponse(title=str(u), arg=str(u), icon=ICON_NOTE, valid=True)]
+            self._u = uuid.UUID(self.args[0])
+            return True
+
+        except ValueError as e:
+            self._message_error = str(e)
+            return False
+
+
+def main(args: list[str], logger) -> list[AFWResponse]:
+    return AFWFunc(args, logger)()