Source code for pydmt.main

"""
main.py
"""

import sys
import os
import os.path
import pathlib
import logging

import pylogconf.core
from pytconf import register_endpoint, register_main, config_arg_parse_and_launch

from pydmt.configs import ConfigSudo, ConfigFlow, ConfigOutput, ConfigLogging, ConfigVenv, \
    ConfigImport, ConfigReqs
from pydmt.core.pydmt import PyDMT
from pydmt.static import APP_NAME, VERSION_STR, DESCRIPTION, LOGGER_NAME
from pydmt.utils.subprocess import check_call

from pydmt.features.sphinx import FeatureSphinx
from pydmt.features.mako import FeatureMako
from pydmt.features.yaml import FeatureYaml
from pydmt.features.apt import FeatureApt
from pydmt.features.npm import FeatureNpm
from pydmt.features.venv import FeatureVenv
from pydmt.features.reqs import FeatureReqs
from pydmt.features.make import FeatureMake
from pydmt.features.gem import FeatureGem


[docs] def check_config_and_load(): if not os.path.isfile(".pydmt.config"): print("pydmt: not a pydmt folder", file=sys.stderr) sys.exit(1)
[docs] def add_all_features(p): """ This method should be automatic, order matters here which is not good, order should be derived and not set in stone """ f = FeatureApt() f.setup(p) f = FeatureNpm() f.setup(p) f = FeatureGem() f.setup(p) f = FeatureVenv() f.setup(p) f = FeatureReqs() f.setup(p) f = FeatureMako() f.setup(p) f = FeatureSphinx() f.setup(p) f = FeatureMake() f.setup(p) f = FeatureYaml() f.setup(p)
[docs] def add_to_path(): """ This adds to PYTHONPATH various paths we need If you disable this then templates would not be able to find things like 'config/python.py' """ if ConfigImport.import_system: folder = "/etc/pydmt" if folder not in sys.path: sys.path.insert(0, folder) if ConfigImport.import_home: folder = os.path.join(str(pathlib.Path.home()), ".config/pydmt") if folder not in sys.path: sys.path.insert(0, folder) if ConfigImport.import_cwd: if "" not in sys.path: sys.path.insert(0, "")
[docs] @register_endpoint( description="Build the project", configs=[ ConfigSudo, ConfigFlow, ConfigOutput, ConfigLogging, ConfigVenv, ConfigReqs, ], ) def build(): check_config_and_load() add_to_path() pylogconf.core.setup() p = PyDMT() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) add_all_features(p) stats = p.build_all() sys.exit(stats.get_os_error_code())
[docs] @register_endpoint( description="Build python virtual environment", configs=[ ConfigSudo, ConfigFlow, ConfigOutput, ConfigLogging, ConfigVenv, ConfigReqs, ], ) def build_venv(): check_config_and_load() add_to_path() pylogconf.core.setup() p = PyDMT() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) f = FeatureVenv() f.setup(p) stats = p.build_all() sys.exit(stats.get_os_error_code())
[docs] @register_endpoint( description="Install python prerequisites", configs=[ ConfigSudo, ConfigFlow, ConfigOutput, ConfigLogging, ConfigVenv, ConfigReqs, ], ) def build_reqs(): check_config_and_load() add_to_path() pylogconf.core.setup() p = PyDMT() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) f = FeatureReqs() f.setup(p) stats = p.build_all() sys.exit(stats.get_os_error_code())
[docs] @register_endpoint( description="Build tools", configs=[ ConfigSudo, ConfigFlow, ConfigOutput, ConfigLogging, ConfigVenv, ConfigReqs, ], ) def build_tools(): check_config_and_load() add_to_path() pylogconf.core.setup() p = PyDMT() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) f = FeatureApt() f.setup(p) f = FeatureNpm() f.setup(p) f = FeatureGem() f.setup(p) stats = p.build_all() sys.exit(stats.get_os_error_code())
[docs] @register_endpoint( description="Clean all generated files by asking each builder to clean it's generated files", configs=[ ConfigLogging, ], ) def clean() -> None: check_config_and_load() add_to_path() pylogconf.core.setup() p = PyDMT() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) add_all_features(p) # shutil.rmtree(".pydmt") p.clean_all()
[docs] @register_endpoint( description="Clean all non git tracked files (including pydmt cache)", configs=[ ConfigLogging, ], ) def clean_hard() -> None: check_config_and_load() logger = logging.getLogger(LOGGER_NAME) logger.setLevel(ConfigLogging.loglevel) # shutil.rmtree(".pydmt") check_call(["git", "clean", "-qffxd"])
[docs] @register_main( app_name=APP_NAME, version=VERSION_STR, main_description=DESCRIPTION, ) def main(): pylogconf.core.setup() config_arg_parse_and_launch()
if __name__ == '__main__': main()