72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
# pyChaChaDummyProject (c) by chacha
|
|
#
|
|
# pyChaChaDummyProject is licensed under a
|
|
# Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International Unported License.
|
|
#
|
|
# You should have received a copy of the license along with this
|
|
# work. If not, see <https://creativecommons.org/licenses/by-nc-sa/4.0/>.
|
|
|
|
from __future__ import annotations
|
|
from typing import TYPE_CHECKING
|
|
|
|
from abc import ABC,abstractmethod
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
|
|
if TYPE_CHECKING: # Only imports the below statements during type checking
|
|
from typing import Union
|
|
|
|
class helper_base(ABC):
|
|
project_rootdir_path: Union[Path,None] = None
|
|
pyproject: Union[dict,None] = None
|
|
current_dir: Union[Path,None] = None
|
|
|
|
@classmethod
|
|
def set_context(cls,project_rootdir_path:Path, pyproject:dict):
|
|
cls.project_rootdir_path = project_rootdir_path
|
|
cls.pyproject = pyproject
|
|
cls.current_dir = Path(__file__).parent.absolute()
|
|
|
|
@classmethod
|
|
def get_result_dir(cls):
|
|
return None
|
|
|
|
@staticmethod
|
|
def _reset_dir(dirpath:Path):
|
|
dirpath = Path(dirpath)
|
|
if not os.path.exists(dirpath):
|
|
os.makedirs(dirpath)
|
|
[f.unlink() for f in Path(dirpath).glob("*") if f.is_file()]
|
|
|
|
@classmethod
|
|
def reset_result_dir(cls):
|
|
result_dir = cls.get_result_dir()
|
|
if result_dir != None:
|
|
cls._reset_dir(result_dir)
|
|
|
|
@classmethod
|
|
@abstractmethod
|
|
def do_job(cls):
|
|
raise NotImplementedError()
|
|
|
|
@classmethod
|
|
def run_cmd_(cls, cmdarray):
|
|
process = subprocess.run(cmdarray, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, check=True)
|
|
return process.stdout
|
|
|
|
@classmethod
|
|
def run_cmd(cls, cmdarray):
|
|
p = subprocess.run(cmdarray, capture_output=True)
|
|
print(p.stdout)
|
|
print(p.stderr)
|
|
return p.stdout
|
|
|
|
class helper_withresults_base(helper_base):
|
|
helper_results_dir: Union[Path,None] = None
|
|
|
|
@classmethod
|
|
def get_result_dir(cls):
|
|
if cls.helper_results_dir==None:
|
|
cls.helper_results_dir=cls.__name__
|
|
return Path(__file__).parent.parent.absolute() / "helpers-results" / cls.helper_results_dir |