data_manager_cli.py

#!/usr/bin/env python3
import argparse
import os
import sys

from packageviewer.db.data_manager import DataManager
import timer


class DataManagerCli:

    def __init__(self, argv):
        self.argv = argv
        self.SCRIPT_VERSION = "v1.0 beta"

    def get_parser(self, action=""):
        return argparse.ArgumentParser(
            prog=f"data_manager_cli {action}".strip(),
            description="Process data",
            epilog=f"script version {self.SCRIPT_VERSION}",
        )

    def run(self):
        parser = self.get_parser()
        parser.add_argument("action", help="Action wanted")

        args = parser.parse_args(self.argv[1:2])

        match args.action:
            case "add":
                self.parse_action_add()
            case "add-indexes":
                self.parse_action_add_indexes()
            case _:
                parser.error(f"Invalid action: '{args.action}'")

    def parse_action_add(self):
        parser = self.get_parser()
        parser.add_argument("--db", default="out.db", help="Set database file to use")

        parser.add_argument(
            "-d", "--distro", required=True, help="Set the distribution"
        )
        parser.add_argument(
            "-v", "--version", default="DEFVERSION", help="Set the distribution version"
        )

        parser.add_argument(
            "--reset-db",
            required=False,
            default=False,
            action="store_true",
            help="Reset database if already existing",
        )
        parser.add_argument(
            "--add-indexes",
            required=False,
            default=False,
            action="store_true",
            help="Reset database if already existing",
        )

        self.args = parser.parse_args(self.argv[2:])

        self.do_action_add()

    def do_action_add(self):
        print(
            f"Operation requested : process '{self.args.distro}/{self.args.version}' to database '{self.args.db}'"
        )

        self.init_data_manager(self.args.reset_db)

        path = os.path.join("archives", self.args.distro, self.args.version)

        timer.call(
            self.data_manager.process_data_point,
            distro_name=self.args.distro,
            distro_version=self.args.version,
            dir_path=path,
        )

    def parse_action_add_indexes(self):
        parser = self.get_parser()

        parser.add_argument("--db", default="out.db", help="Set database file to use")

        self.args = parser.parse_args(self.argv[2:])

        self.do_action_add_indexes()

    def do_action_add_indexes(self):
        print(f"Operation requested : add indexes to database '{self.args.db}'")

        self.init_data_manager()

        self.data_manager.add_indexes()

    def init_data_manager(self, force_reset=False):
        need_init = False
        if force_reset:
            need_init = True
            if os.path.exists(self.args.db):
                print(
                    f"Warning: Removing database at path {os.path.abspath(self.args.db)}"
                )
                os.remove(self.args.db)
        else:
            need_init = not os.path.exists(self.args.db)
            if not os.path.exists(self.args.db):
                print(
                    f"Warning: Database at path {os.path.abspath(self.args.db)} doesn't exist. It will be created"
                )

        self.data_manager = DataManager(self.args.db)
        if need_init:
            self.data_manager.create_tables()


cli = DataManagerCli(sys.argv)
cli.run()