added cli

Author: dmy.berezovskyi

README.md

@@ -59,7 +59,7 @@ A simple UI automation framework built with:
    ```
 
 ### Latest Drivers
-
+- #### CLI util [util](utils/driver_management.md)
 - #### [Chrome Drivers](https://googlechromelabs.github.io/chrome-for-testing/#stable)
 - #### [Firefox Drivers](https://github.com/mozilla/geckodriver)
 - You can download the latest version of the driver for macOS ARM64 using the `chromedriver.sh` script located in the `resources` folder.

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "simple-framework"
-version = "0.1.3"
+version = "0.3.3"
 description = "SIMPLE Python selenium framework"
 authors = ["Dmytro Berezovskyi"]
 readme = "README.md"
@@ -28,14 +28,15 @@ ruff="0.6.8"
 secure-test-automation="^1.3.1"
 colorama="==0.4.6"
 click="==8.1.8"
-
+inquirer = "==2.10.1"
 
 [tool.pytest.ini_options]
 addopts = "-rA -v --env=dev --type=local --capture=no -p no:cacheprovider"
 markers = [
     { name = "smoke", description = "run smoke tests" },
     { name = "regression", description = "run regression tests" },
-    { name = "sanity", description = "run sanity tests" }
+    { name = "sanity", description = "run sanity tests" },
+    { include = "cli.py", from = "utils" },
 ]
 testpaths = ["tests"]
 python_files = ["*.py"]

utils/cli.py

@@ -1,61 +1,112 @@
 import argparse
+import os
+import sys
 from pathlib import Path
+from typing import Optional
+
+from colorama import init
+from pyfiglet import Figlet
+import inquirer
 
 from utils.scraper.chrome_scraper import ChromePageScraper
 
+project_root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 
-def main():
-    parser = argparse.ArgumentParser(prog="psaf", description="PSAF CLI Tool")
+sys.path.insert(0, project_root)
 
-    # Subcommand for 'get' operations
-    subparsers = parser.add_subparsers(dest="command")
 
-    # Subcommand for 'get chromedriver'
-    get_parser = subparsers.add_parser("get", help="Get a specific resource")
-    get_subparsers = get_parser.add_subparsers(dest="resource")
+__version__ = "1.1.0"
 
-    # Subcommand for 'chromedriver' under 'get'
-    chromedriver_parser = get_subparsers.add_parser(
-        "chromedriver", help="Download the ChromeDriver"
-    )
+TASKS = {
+    "chromedriver": ["get_driver_by_milestone", "exit"],
+    "firefox": ["download_geckodriver", "exit"],
+}
+
+
+def parse_args() -> argparse.Namespace:
+    """Parse command line arguments."""
+    parser = argparse.ArgumentParser(description="Driver Management Tool")
+
+    # Adding subparsers for different drivers like chromedriver and geckodriver
+    subparsers = parser.add_subparsers(dest='driver', help='Choose a driver to manage')
+
+    # ChromeDriver specific subparser
+    chromedriver_parser = subparsers.add_parser('chromedriver', help='Manage ChromeDriver')
     chromedriver_parser.add_argument(
-        "--platform",
-        type=str,
-        help="Specify the platform (e.g., 'win32', 'linux64').",
+        "-v", "--version", action="store_true", help="Show version and exit."
     )
     chromedriver_parser.add_argument(
-        "--version",
-        type=str,
-        help="Specify the version of ChromeDriver.",
+        "-d", "--docs", action="store_true", help="Show documentation and exit."
     )
     chromedriver_parser.add_argument(
+        "-milestone",
         "--milestone",
         type=str,
-        help="Specify the milestone of ChromeDriver.",
-    )
-    chromedriver_parser.add_argument(
-        "--output-dir",
-        type=str,
-        help="Directory to save the downloaded ChromeDriver.",
-    )
-    chromedriver_parser.add_argument(
-        "--extract",
-        action="store_true",
-        help="Extract the downloaded ChromeDriver.",
+        required=True,  # Make milestone required
+        help="Specify the milestone version (e.g., '131').",
     )
 
-    args = parser.parse_args()
+    return parser.parse_args()
+
+
+def execute_task(task: str, milestone: str, version: Optional[str]) -> None:
+    """Execute the selected task."""
+    try:
+        match task:
+            case "get_driver_by_milestone":
+                print(f"Downloading ChromeDriver for version {milestone}...")
+                ChromePageScraper.get_chromedriver(milestone=milestone, version=version)
+            case "TODO_download_geckodriver_TODO":
+                print(f" Downloading GeckoDriver for version {milestone}...")
+            case "exit":
+                exit(0)
+            case _:
+                print(f"Task '{task}' is not supported.")
+    except Exception as e:
+        print(f"Error executing task '{task}': {e}")
+
+
+def main() -> None:
+    """Entry point for the Driver Management CLI."""
+    init()
+    f = Figlet(font="Slant")
+    print(f.renderText("Driver Tool"))
+
+    args = parse_args()
+
+    # Show version or documentation if requested
+    if args.version:
+        print(f"Driver Tool version: {__version__}")
+        return
+    if args.docs:
+        print("Documentation: This is a CLI tool to manage drivers like ChromeDriver and GeckoDriver.")
+        return
+
+    # Get available tasks based on the selected driver and milestone
+    if args.driver == 'chromedriver' and args.milestone:
+        available_tasks = TASKS.get("chromedriver", [])
+        if not available_tasks:
+            print(f"No tasks available for the selected milestone: {args.milestone}")
+            return
 
-    if args.command == "get" and args.resource == "chromedriver":
-        output_dir = Path(args.output_dir) if args.output_dir else None
-        ChromePageScraper.get_chromedriver(
-            platform=args.platform,
-            version=args.version,
-            milestone=args.milestone,
-            d_dir=output_dir,
-            is_extracted=args.extract,
-        )
+        try:
+            while True:
+                answers = inquirer.prompt(
+                    [
+                        inquirer.List(
+                            "task",
+                            message=f"What task do you want to perform for ChromeDriver {args.milestone}?",
+                            choices=available_tasks,
+                        )
+                    ]
+                )
+                task = answers["task"]
+                execute_task(task, milestone=args.milestone, version=None)
+        except KeyboardInterrupt:
+            print("\nExiting...")
+        except Exception as e:
+            print(f"Unexpected error: {e}")
 
 
 if __name__ == "__main__":
-    main()
+    main()

utils/driver_management.md

@@ -0,0 +1,22 @@
+# Driver Management Tool
+
+## Version: 1.1.0
+
+A command-line interface (CLI) tool to manage driver versions for ChromeDriver. This tool allows you to download the correct driver based on milestones and other specifications.
+
+### Features
+
+- **ChromeDriver Management**: Download ChromeDriver for a specific milestone version.
+
+### Usage
+
+```bash
+   psaf chromedriver --milestone 131
+```
+### you will see a message in console
+```bash
+   [?] What task do you want to perform for ChromeDriver 131?:
+  1. get_driver_by_milestone
+  2. exit
+```
+### select option and pass enter