cudamarkopy.py

Go to the documentation of this file.

#!/usr/bin/python3
 
 
 
 
 
import sys
import os
from importlib.util import spec_from_loader, module_from_spec
from importlib.machinery import SourceFileLoader, ExtensionFileLoader
import inspect 
try:
    spec = spec_from_loader("markopy", SourceFileLoader("markopy", os.path.abspath("markopy.py")))
    markopy = module_from_spec(spec)
    spec.loader.exec_module(markopy)
except (ModuleNotFoundError,FileNotFoundError) as e:
    if(os.path.exists("../../../Markopy/src/CLI/markopy.py")):
        spec = spec_from_loader("markopy", SourceFileLoader("markopy", "../../../Markopy/src/CLI/markopy.py"))
        markopy = module_from_spec(spec)
        spec.loader.exec_module(markopy)
 
try:
    from cudammx import CudaModelMatrixCLI
    from mmx import ModelMatrixCLI
    from mp import MarkovPasswordsCLI
 
except ModuleNotFoundError as e:
    #print("Working in development mode. Trying to load markopy.py from ../../../Markopy/")
    if(os.path.exists("../../../Markopy/src/CLI/cudammx.py")):
        sys.path.insert(1, '../../../Markopy/src/CLI/')
        from cudammx import CudaModelMatrixCLI
        from mmx import ModelMatrixCLI
        from mp import MarkovPasswordsCLI
    else:
        raise e 
 
from termcolor import colored
 
 
class CudaMarkopyCLI(markopy.MarkopyCLI, CudaModelMatrixCLI):
    """!
         @brief CUDA extension to MarkopyCLI. Adds CudaModelMatrixCLI to the interface.
         @belongsto Python::CudaMarkopy
         @extends Python::Markopy::MarkopyCLI
         @extends Python::CudaMarkopy::CudaModelMatrixCLI
    """
    def __init__(self) -> None:
        "! @brief initialize CLI selector"
        markopy.MarkopyCLI.__init__(self,add_help=False)
        self.parserparser.epilog+=f"""
        {__file__.split("/")[-1]} -mt CUDA generate trained.mdl -n 500 -w output.txt
            Import trained.mdl, and generate 500 lines to output.txt
        """
 
    def help(self):
        "! @brief overload help string"
        self.parserparser.print_help = self.stubstub
        self.argsargsargsargs = self.parserparser.parse_known_args()[0]
        if(self.argsargsargsargs.model_type!="_MMX"):
            if(self.argsargsargsargs.model_type=="MP"):
                mp = MarkovPasswordsCLI()
                mp.add_arguments()
                mp.parser.print_help()
            elif(self.argsargsargsargs.model_type=="MMX"):
                mp = ModelMatrixCLI()
                mp.add_arguments()
                mp.parser.print_help()
            elif(self.argsargsargsargs.model_type == "CUDA"):
                mp = CudaModelMatrixCLI()
                mp.add_arguments()
                mp.parser.print_help()
        else:
            print(colored("Model Mode selection choices:", "green"))
            self.print_helpprint_help()
            print(colored("Following are applicable for -mt MP mode:", "green"))
            mp = MarkovPasswordsCLI()
            mp.add_arguments()
            mp.parser.print_help()
            print(colored("Following are applicable for -mt MMX mode:", "green"))
            mp = ModelMatrixCLI()
            mp.add_arguments()
            mp.parser.print_help()
            print(colored("Following are applicable for -mt CUDA mode:", "green"))
            mp = CudaModelMatrixCLI()
            mp.add_arguments()
            mp.parser.print_help()
        exit()
 
    def parse(self):
        markopy.MarkopyCLI.parse(self)
 
    def parse_fail(self):
        "! @brief Not a valid model type"
        if(self.argsargsargsargs.model_type == "CUDA"):
            self.cliclicli = CudaModelMatrixCLI()
        else:
            markopy.MarkopyCLI.parse_fail(self)
        
 
if __name__ == "__main__":
    mp = CudaMarkopyCLI()
    #mp = markopy_cli.MarkopyCLI()
    mp.parse()
    mp.process()