29 lines
943 B
Python
Executable file
29 lines
943 B
Python
Executable file
#!/usr/bin/env python3
|
|
### Script for adding a new category. Handles adding data to config.csv.
|
|
|
|
from utils import csv as util_csv
|
|
|
|
divider = "----------"
|
|
|
|
print(f"\nWe'll ask for the name and rules of the new category.\n\n{divider}\n")
|
|
|
|
# All input handling
|
|
tk_category_name = input("Name: ").replace('"', "")
|
|
tk_category_rules = input("Rules: ").replace('"', "")
|
|
|
|
# Generate dashname (specifically for directory names)
|
|
tk_category_dashname = tk_category_name.replace(" ", "_").replace("%", "")
|
|
|
|
# Define category dictionary
|
|
categoryDict = {
|
|
"tk_category_dashname": tk_category_dashname,
|
|
"tk_category_name": tk_category_name,
|
|
"tk_category_rules": tk_category_rules,
|
|
}
|
|
|
|
# Write to csv
|
|
util_csv.dictWriter("../csv/categories.csv", categoryDict, "a")
|
|
|
|
print(
|
|
f"\n{divider}\n\nAdded category! If you made a mistake, you can manually edit csv/categories.csv (and csv/runs.csv if you've added any runs to the borked category)."
|
|
)
|