randtrix-cli.py

import argparse, cmd2
from colorama import init, Fore, Back, Style
from randtrix_vault.assembler import *
from randtrix_vault.db import RandtrixDBManager
from randtrix_vault.tools import *

init(autoreset=True)

banner = """
################################################################
#                     THE RANDTRIX VAULT                        #
# RANDOM MECHANISM WITH AES-256 ENCRYPTION FOR PASSWORD PROTECT #
#                                                               #
# DEVELOPER NAME: BALAVIGNESH M                                 #
# LICENSE: MIT                                                  #
# VERSION: 0.1.0                                                #
#################################################################
"""


class RandtrixCLI(cmd2.Cmd):
    def __init__(self):
        super().__init__(multiline_commands=['orate'])
        self.hidden_commands += [
            'alias', 'edit', 'history', 'macro', 'py', 'run_pyscript', 'run_script', 'set', 'shell', 'shortcuts'
        ]
        self.intro = Style.BRIGHT + Fore.GREEN + banner + '\x00'
        self.prompt = Style.BRIGHT + Fore.RED + 'randtrix-cli>'

    def randtrix(self, args):
        if args.command == 'cr_profile_trix':
            vals = {
                'six_d_seed': args.six_d_seed,
                'first_secret': args.first_secret,
                'second_secret': args.second_secret,
                'third_secret': args.third_secret,
                'profile_pass': args.profile_pass,
                'profile_id': args.profile_id,
                'tags': args.tags
            }
            record_id = RandtrixAssembler.create_new_profile_entry(vals)
            RandtrixTools.parse_string(args.profile_id, record_id)

        if args.command == 'rd_profile_trix':
            vals = {
                'profile_id': args.profile_id,
                'six_d_seed': args.six_d_seed,
                'first_secret': args.first_secret,
                'second_secret': args.second_secret,
                'third_secret': args.third_secret,
            }
            data = RandtrixAssembler.get_profile_pass(vals)
            RandtrixTools.parse_string(args.profile_id, data, t='get')

        if args.command == 'show_all_profile_ids':
            vals = {
                'tags': args.tags or ''
            }
            data = RandtrixAssembler.get_profile_ids(vals)
            RandtrixTools.parse_string(data=data, t='show')

    randtrix_parser = argparse.ArgumentParser(prog="randtrix",
                                              description="",
                                              conflict_handler='resolve',
                                              formatter_class=argparse.RawTextHelpFormatter,
                                              usage="randtrix [-h]"
                                              )

    subparsers = randtrix_parser.add_subparsers(dest='command', title="Options")
    subparser = subparsers.add_parser('cr_profile_trix', help='Create Password Entry')
    subparser.add_argument('profile_id', help='Profile ID', default=None)
    subparser.add_argument('profile_pass', help='Profile Password', default=None)
    subparser.add_argument('six_d_seed', help='6 Digit Integer Seed Value', default=None)
    subparser.add_argument('first_secret', help='First Secret', default=None)
    subparser.add_argument('second_secret', help='Second Secret', default=None)
    subparser.add_argument('third_secret', help='Third Secret', default=None)
    subparser.add_argument('--tags', help='Tags', default=None)

    subparser = subparsers.add_parser('rd_profile_trix', help='Get Password Entry')
    subparser.add_argument('profile_id', help='Profile ID', default=None)
    subparser.add_argument('six_d_seed', help='6 Digit Integer Seed Value', default=None)
    subparser.add_argument('first_secret', help='First Secret', default=None)
    subparser.add_argument('second_secret', help='Second Secret', default=None)
    subparser.add_argument('third_secret', help='Third Secret', default=None)

    subparser = subparsers.add_parser('show_all_profile_ids', help='Show All Profile ID')
    subparser.add_argument('--tags', help='Tags', default=None)

    randtrix_parser.set_defaults(func=randtrix)

    @cmd2.with_argparser(randtrix_parser)
    def do_randtrix(self, args):
        func = getattr(args, "func", None)
        if func is not None:
            func(self, args)


cli = RandtrixCLI()
cli.debug = True
RandtrixDBManager.generate_database_file()
cli.cmdloop()