1
0
mirror of https://github.com/enpaul/kodak.git synced 2024-11-14 10:36:55 +00:00

Add trivial CLI for maintenance tasks

This commit is contained in:
Ethan Paul 2021-11-12 23:34:47 -05:00
parent 462dc572be
commit 25b61a612c
No known key found for this signature in database
GPG Key ID: D0E2CBF1245E92BF
2 changed files with 78 additions and 0 deletions

75
kodak/cli.py Normal file
View File

@ -0,0 +1,75 @@
import argparse
import logging
import sys
from kodak import __about__
from kodak import configuration
from kodak import database
from kodak import tools
def get_args() -> argparse.Namespace:
"""Setup the flags for the basic CLI"""
parser = argparse.ArgumentParser(
prog=__about__.__name__, description=__about__.__summary__
)
parser.add_argument(
"--server", action="store_true", help="Run the Flask development server"
)
parser.add_argument(
"--config", action="store_true", help="Check that the configuration is valid"
)
parser.add_argument(
"--version", action="store_true", help="Show program version and exit"
)
parser.add_argument(
"--index", action="store_true", help="Rebuild the source image index"
)
return parser.parse_args()
def main() -> int:
"""Main entrypoint for the CLI tooling"""
args = get_args()
logging.basicConfig(
stream=sys.stderr,
format="(%(asctime)s) %(levelname)s: %(message)s",
datefmt="%I:%M:%S",
level=logging.DEBUG,
)
if args.version:
print(f"{__about__.__title__} {__about__.__version__}", file=sys.stderr)
return 0
if args.config:
try:
configuration.load()
except Exception as err: # pylint: disable=broad-except
print(f"Configuration check failed: {err}", file=sys.stderr)
return 1
else:
print("Configuration check successful!", file=sys.stderr)
return 0
if args.index:
config = configuration.load()
database.initialize(config)
tools.index.build(config)
return 0
if args.server:
from kodak import application # pylint: disable=import-outside-toplevel
application.APPLICATION.run(host="127.0.0.1")
return 0
return 1
if __name__ == "__main__":
sys.exit(main())

View File

@ -36,6 +36,9 @@ classifiers = [
"Topic :: Multimedia :: Graphics :: Graphics Conversion"
]
[tool.poetry.scripts]
kodak = "kodak.cli:main"
[tool.poetry.dependencies]
python = "^3.7"
flask = "^1.1.2"