85 lines
2.3 KiB
Python
85 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
|
|
)
|