Destination Output and Unit Testing

csv2docx/cli.py

@@ -1,5 +1,6 @@
-import click
 import csv2docx.csv2docx as c2d
+import click
+
 
 
 @click.command()
@@ -14,6 +15,10 @@
 @click.option(
     '--delimiter', '-d',
     default=";",
-    help='delimiter used in your csv. Default is \';\'')
-def main(data, template, delimiter):
-	c2d.convert(data, template, delimiter)
+    help='Delimiter used in your csv. Default is \';\'')
+@click.option(
+    '--destination', '-dest',
+    default="output",
+    help='The destination to store the files.')
+def main(data, template, delimiter, destination):
+    c2d.convert(data, template, delimiter, destination)

csv2docx/csv2docx.py

@@ -1,7 +1,26 @@
-import csv
 from mailmerge import MailMerge
-def convert(data, template, delimiter=";"):
-    print ("Getting .docx template and .csv data files ...")
+import csv
+import pathlib
+
+
+def absolute_path(path: str) -> pathlib.Path:
+    """Creates a path to store output data if it does not exists.
+
+    This method will create the directory if non-existent.
+
+    Args:
+        path: the path from user in any format (relative, absolute, etc.)
+    Returns:
+        A path to store output data.
+    """
+    path = pathlib.Path(path)
+    if not path.exists():
+        path.mkdir(parents=True, exist_ok=True)
+    return path
+
+
+def convert(data, template, delimiter=";", destination="output"):
+    print("Getting .docx template and .csv data files ...")
 
     with open(data, 'rt') as csvfile:
         csvdict = csv.DictReader(csvfile, delimiter=delimiter)
@@ -16,15 +35,16 @@ def convert(data, template, delimiter=";"):
         # see if all fields are accounted for in the .csv header
         column_in_data = set(docx_mergefields) - set(csv_headers)
         if len(column_in_data) > 0:
-            print (f"{column_in_data} is in the word document, but not csv.")
+            print(f"{column_in_data} is in the word document, but not csv.")
             return
 
         print("All fields are present in your csv. Generating Word docs ...")
 
+        path = absolute_path(destination)
+
         for counter, row in enumerate(csvdict):
             # Must create a new MailMerge for each file
             docx = MailMerge(template)
-            single_document = {key : row[key] for key in docx_mergefields}
+            single_document = {key: row[key] for key in docx_mergefields}
             docx.merge_templates([single_document], separator='page_break')
-            # TODO: write to user-defined subfolder
-            docx.write(f"{counter}.docx")
+            docx.write(f"{path}/{counter}.docx")