GenerateCSV.py

import csv
import sys
import os
import re
import random
import string
        
        
InputFields = ["CSV File Name", "Numbr of Issues"]

if __name__ == "__main__":
    main()

def main():
    csvFields = ["Project Name", "Project Key", "Issue Type", "Status", "Priority", "Issue ID", "Summary", "Comment", "Description", "Project Type"]
    issueTypes = ["Story", "Task", "Bug", "Epic"]
    statuses = ["Open", "Reopened", "Ready For Launch", "In Progress", "Resolved"]
    priority = ["high", "medium", "low"]

    csvName, projectNames, projectKeys, numIssues = processCommandLineArguments()
    csvName = csvName + ".csv"
    numIssues = [int(num) for num in numIssues]

    if os.path.exists(csvName) == True:
        while True:
            answer = input("A file named " + csvName + "already exists.  Are you sure you want to overwrite it? [y/n]").lower()
            if answer == 'y':
                break
            elif answer == 'n':
                exit()

    with open(csvName, 'w', newline = '') as file:
        writer = csv.writer(file)

        writer.writerow(csvFields)

        letters = string.ascii_lowercase + " "
        for projectName, projectKey, issues in zip(projectNames, projectKeys, numIssues):
            for issueNumber in range(1, issues + 1):
                description = ''.join(random.choice(letters) for i in range(random.randint(5, 2000)))
                comment = ''.join(random.choice(letters) for i in range(random.randint(5, 100)))
                writer.writerow([projectName, projectKey, random.choice(issueTypes), random.choice(statuses), random.choice(priority), str(issueNumber), "This is a well written summary of issue " + str(issueNumber), comment, description, "software"])

def validProjectKey(projectKey):
    keyPattern = re.compile("[^A-Z0-9]")
    if re.search(keyPattern, projectKey):
       return False
    return True

def processCommandLineArguments():
    projectNames = []
    projectKeys = []
    numIssues = []
    
    if len(sys.argv)%3 -2 != 0 or len(sys.argv) < 5:
        fields = ""
        for field in InputFields:
            fields = fields + "[" + field + "] "
        raise ValueError("run python3 GenerateCSV.py {[Project Name] [Project Key] [Number of Tickets]}. See README.md")

    csvName = sys.argv[1]
    numberOfProjects = int((len(sys.argv) -2) /3)
    for project in range(numberOfProjects):
        if not validProjectKey(sys.argv[3*project + 3]):
            raise ValueError("Project Keys must consist of Capital Letters and Numbers Only")
        if not sys.argv[3*project + 4].isnumeric():
            raise ValueError("Number of Issues must be a number")
        projectNames.append(sys.argv[3*project + 2])
        projectKeys.append(sys.argv[3*project + 3])
        numIssues.append(sys.argv[3*project + 4])

    return csvName, projectNames, projectKeys, numIssues