import os
import sys
from pathlib import Path
from typing import NoReturn
from . import find_requirements
from .exceptions import RequirementsNotFound
from .formatters import FORMATTERS
def _die(message) -> NoReturn:
sys.stderr.write("%s\n" % message)
sys.exit(1)
def run() -> NoReturn:
if len(sys.argv) > 1:
path = Path(sys.argv[1])
else:
path = Path.cwd()
if not path.exists():
_die("%s does not exist" % path)
if not path.is_dir():
_die("%s is not a directory" % path)
try:
requirements = find_requirements(path)
except RequirementsNotFound:
_die("Unable to find requirements at %s" % path)
format_name = "requirements_file" # TODO: other output formats such as JSON
FORMATTERS[format_name](requirements)
sys.exit(0)
if __name__ == "__main__":
run()
|