You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
110 lines
2.7 KiB
Python
110 lines
2.7 KiB
Python
import argparse
|
|
import shutil
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
|
|
def _err(msg, die=None):
|
|
"""Print an error message and exits if an exit code is given"""
|
|
sys.stderr.write(f"ERROR: {msg}\n")
|
|
if die:
|
|
sys.exit(die if type(die) is int else 1)
|
|
|
|
|
|
try:
|
|
import pelican
|
|
except ImportError:
|
|
_err(
|
|
"Cannot import pelican.\nYou must "
|
|
"install Pelican in order to run this script.",
|
|
1,
|
|
)
|
|
|
|
# where data files are stored in pelican-data-files's compatible theme
|
|
SRC_DATA_DIR = "data"
|
|
|
|
# where data file are copied in pelican project directory
|
|
DEST_DATA_DIR = "data"
|
|
|
|
PELICAN_ROOT_DIR = Path(pelican.__file__).absolute().parent
|
|
PELICAN_THEMES_DIR = PELICAN_ROOT_DIR.joinpath("themes")
|
|
|
|
|
|
def _parse_args():
|
|
"""Parse and return cli arguments"""
|
|
parser = argparse.ArgumentParser(
|
|
description="""List and fetch data files from compatible themes"""
|
|
)
|
|
|
|
parser.add_argument(
|
|
"-l",
|
|
"--list",
|
|
action="store_true",
|
|
help="list all themes compatible with this plugin",
|
|
)
|
|
|
|
parser.add_argument(
|
|
"-f",
|
|
"--fetch",
|
|
type=str,
|
|
metavar="THEME_NAME",
|
|
help="fetch theme's data files",
|
|
)
|
|
|
|
# parser.add_argument(
|
|
# "--force",
|
|
# action="store_true",
|
|
# help="overwrite files already present in your project",
|
|
# )
|
|
|
|
# if no args parsed, set --help
|
|
return parser.parse_args(None if sys.argv[1:] else ["--help"])
|
|
|
|
|
|
def _get_themes():
|
|
return [theme for theme in PELICAN_THEMES_DIR.iterdir() if theme.is_dir()]
|
|
|
|
|
|
def _get_compatible_themes(themes):
|
|
return [theme for theme in themes if _is_compatible(theme)]
|
|
|
|
|
|
def _is_compatible(theme):
|
|
return theme.joinpath(SRC_DATA_DIR).exists()
|
|
|
|
|
|
def _fetch_files(theme):
|
|
try:
|
|
shutil.copytree(
|
|
theme.joinpath(SRC_DATA_DIR), Path.cwd().joinpath(DEST_DATA_DIR)
|
|
)
|
|
except FileExistsError:
|
|
_err(
|
|
f"Cannot copy '{theme.name}' files, "
|
|
f"directory '{DEST_DATA_DIR}' already exists.",
|
|
die=2,
|
|
)
|
|
else:
|
|
print(f"'{theme.name}' files successfully fetched.")
|
|
|
|
|
|
def main():
|
|
"""Main function, called by entrypoint"""
|
|
args = _parse_args()
|
|
|
|
if args.list:
|
|
for theme in _get_compatible_themes(_get_themes()):
|
|
print(f"{theme.name}")
|
|
sys.exit(0)
|
|
|
|
if args.fetch:
|
|
theme = PELICAN_THEMES_DIR.joinpath(args.fetch)
|
|
if theme.exists():
|
|
if _is_compatible(theme):
|
|
_fetch_files(theme)
|
|
sys.exit(0)
|
|
else:
|
|
_err(f"'{args.fetch}' is not a compatible theme.", die=2)
|
|
else:
|
|
_err(f"'{args.fetch}' is not an existing theme.", die=2)
|