You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.7 KiB
60 lines
1.7 KiB
import logging
|
|
import re
|
|
import sys
|
|
|
|
logging.basicConfig()
|
|
LOGGER = logging.getLogger(__name__)
|
|
MODULE_REGEX = r"^[_a-zA-Z][_a-zA-Z0-9]+$"
|
|
|
|
|
|
class bcolors:
|
|
WARNING = "\033[93m"
|
|
ENDC = "\033[0m"
|
|
BOLD = "\033[1m"
|
|
|
|
|
|
def colorize(escape_code, text):
|
|
code = getattr(bcolors, escape_code)
|
|
return "{code}{text}{end_code}".format(code=code, text=text, end_code=bcolors.ENDC)
|
|
|
|
|
|
def log_module_name_warning(module_name, logger):
|
|
warning = (
|
|
"\n{warning} {fmt_module_name}"
|
|
" is not a valid Python module name!\n"
|
|
"See https://www.python.org/dev/peps/pep-0008/#package-and-module-names"
|
|
" for naming standards.\n"
|
|
).format(
|
|
warning=colorize("WARNING", "WARNING:"),
|
|
fmt_module_name=colorize("BOLD", module_name),
|
|
)
|
|
logger.warning(warning)
|
|
|
|
|
|
def check_python_version():
|
|
python_major_version = sys.version_info[0]
|
|
python_minor_version = sys.version_info[1]
|
|
# Must remain compatible with Python 2 to provide useful error message.
|
|
warning = (
|
|
"\nWARNING: You are running cookiecutter using "
|
|
"Python {}.{}, but a version >= Python 3.6+ is required.\n"
|
|
"Either install a more recent version of Python, or use the Docker instructions.\n"
|
|
).format(python_major_version, python_minor_version)
|
|
if (python_major_version == 2) or (
|
|
python_major_version == 3 and python_minor_version < 6
|
|
):
|
|
LOGGER.warning(warning)
|
|
sys.exit(1)
|
|
|
|
|
|
def validate_python_module_name():
|
|
module_name = "{{ cookiecutter.app_name }}"
|
|
if not re.match(MODULE_REGEX, module_name):
|
|
log_module_name_warning(module_name, LOGGER)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
check_python_version()
|
|
|
|
validate_python_module_name()
|
|
|