Added unique naming for output files

.gitignore

@@ -128,10 +128,15 @@ dmypy.json
 # Pyre type checker
 .pyre/
 
+
 # output by script, except the example files
 *.docx
 !example.docx
 !example.csv
 
+
 # vim files
 *.swp
+
+# vscode files
+.vscode

csv2docx/cli.py

@@ -1,5 +1,5 @@
 import click
-import csv2docx.csv2docx as c2d
+from . import csv2docx
 
 
 @click.command()
@@ -14,6 +14,14 @@
 @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(
+    '--name', '-n',
+    required=True,
+    help='Naming scheme for output files.')
+@click.option(
+    '--path', '-p',
+    default="output",
+    help='The location to store the files.')
+def main(data, template, name, path, delimiter):
+    csv2docx.convert(data, template, name, path, delimiter)

csv2docx/csv2docx.py

@@ -1,12 +1,47 @@
 import csv
 from mailmerge import MailMerge
-def convert(data, template, delimiter=";"):
-    print ("Getting .docx template and .csv data files ...")
+from pathlib import Path
+
+
+def create_output_folder(path: str) -> Path:
+    """Creates a path to store output data if it does not exists.
+    Args:
+        path: the path from user in any format (relative, absolute, etc.)
+    Returns:
+        A path to store output data.
+    """
+    path = Path(path)
+    if not path.exists():
+        path.mkdir(parents=True, exist_ok=True)
+    return path
+
+
+def create_unique_name(filename: str, path: Path) -> Path:
+    """Creates an unique filename for specified path.
+    Args:
+        filename: the name of file to create
+        path: the path where the file is stored
+    Returns:
+        An absolute path with directory.
+    """
+    filename = filename.strip() + ".docx"
+    filepath = path / filename
+    if filepath.exists():
+        # Count available files with same name
+        counter = len(list(path.glob(f"{filepath.stem}*docx"))) + 1
+        filepath = f"{path}/{filepath.stem}_{counter}.docx"
+    return filepath
+
+
+def convert(data, template, name, path="output", delimiter=";"):
+    print("Getting .docx template and .csv data files ...")
 
     with open(data, 'rt') as csvfile:
         csvdict = csv.DictReader(csvfile, delimiter=delimiter)
         csv_headers = csvdict.fieldnames
-
+        if (name not in csv_headers):
+            print("Column name not found. Please enter valid column name")
+            exit()
         docx = MailMerge(template)
         docx_mergefields = docx.get_merge_fields()
 
@@ -16,15 +51,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 = create_output_folder(path)
 
-        for counter, row in enumerate(csvdict):
+        for row in 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")
+            filename = create_unique_name(row[name], path)
+            docx.write(filename)