Added unique naming for output files

.gitignore

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

csv2docx/cli.py

@@ -1,5 +1,5 @@
 import click
-import csv2docx.csv2docx as c2d
+from . import csv2docx
 
 
 @click.command()
@@ -15,5 +15,13 @@
     '--delimiter', '-d',
     default=";",
     help='delimiter used in your csv. Default is \';\'')
-def main(data, template, delimiter):
-	c2d.convert(data, template, delimiter)
+@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, delimiter, name, path):
+	csv2docx.convert(data, template, delimiter, name, path)

csv2docx/csv2docx.py

@@ -1,12 +1,44 @@
 import csv
 from mailmerge import MailMerge
-def convert(data, template, delimiter=";"):
+from pathlib import Path
+
+def create_path_if_not_exists(path: str):
+    """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 generate_name(filename, path):
+    """Checks whether file with same name exists or not
+    Args:
+        filename: the file we want to check
+        path: the directory we want to check in
+    Returns:
+        An unique full path with directory.
+    """
+    filename += ".docx"
+    checkpath = path / filename
+    if checkpath.exists():
+        # Count available files with same name
+        counter = len(list(path.glob(checkpath.stem + "*docx" ))) + 1
+        return f"{path}/{checkpath.stem}_{counter}.docx"
+    return checkpath
+
+def convert(data, template, delimiter=";", custom_name=None, path="output"):
     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 (custom_name != None and custom_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()
 
@@ -20,11 +52,13 @@ def convert(data, template, delimiter=";"):
             return
 
         print("All fields are present in your csv. Generating Word docs ...")
+        path = create_path_if_not_exists(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}
             docx.merge_templates([single_document], separator='page_break')
-            # TODO: write to user-defined subfolder
-            docx.write(f"{counter}.docx")
+            # Striping every name to remove extra spaces
+            filename = generate_name(row[custom_name].strip(), path)
+            docx.write(filename)