diff --git a/frontmatter-to-csv/__init__.py b/frontmatter_to_csv/__init__.py similarity index 100% rename from frontmatter-to-csv/__init__.py rename to frontmatter_to_csv/__init__.py diff --git a/frontmatter-to-csv/__main__.py b/frontmatter_to_csv/convert.py similarity index 93% rename from frontmatter-to-csv/__main__.py rename to frontmatter_to_csv/convert.py index 40931dc..5760bb0 100644 --- a/frontmatter-to-csv/__main__.py +++ b/frontmatter_to_csv/convert.py @@ -25,7 +25,9 @@ fields = { } -def convert(directory="testfiles", outputfile="test.csv"): +def convert(directory="f", outputfile="test.csv"): + if not os.path.isdir(directory): + exit(f"The directory {directory} does not exist!") with open(outputfile, "w", newline="") as csvfile: writer = csv.DictWriter(csvfile, fieldnames=fields.values()) writer.writeheader() diff --git a/setup.cfg b/setup.cfg index 4ea94d2..0010e84 100644 --- a/setup.cfg +++ b/setup.cfg @@ -15,9 +15,9 @@ classifiers = Operating System :: OS Independent [options] -packages = frontmatter-to-csv +packages = frontmatter_to_csv python_requires = >=3.6 [options.entry_points] console_scripts = - frontmatter-to-csv = frontmatter-to-csv \ No newline at end of file + frontmatter-to-csv = frontmatter_to_csv.convert:convert \ No newline at end of file