You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.5 KiB
103 lines
3.5 KiB
3 years ago
|
import csv
|
||
|
|
import re
|
||
|
|
|
||
|
|
# current working script, runs from command line only
|
||
|
|
|
||
|
|
def create_row(raw_input_row_list, extra_item_number, extra_item_index_dict, course_info_list, output_csv_list):
|
||
|
|
# creates new row for extra items
|
||
|
|
|
||
|
|
new_row = course_info_list.copy()
|
||
|
|
# print(course_info_list)
|
||
|
|
# print(extra_item_index_dict[extra_item_number + 1])
|
||
|
|
for index in extra_item_index_dict[extra_item_number + 1]:
|
||
|
|
|
||
|
|
# print(index)
|
||
|
|
# print(new_row)
|
||
|
|
item = raw_input_row_list[index]
|
||
|
|
# print(item)
|
||
|
|
new_row.append(item)
|
||
|
|
# print(new_row)
|
||
|
|
output_csv_list.append(new_row)
|
||
|
|
|
||
|
|
with open ('./report.csv', encoding="utf8") as input_csv:
|
||
|
|
input_reader = csv.reader(input_csv, delimiter=',')
|
||
|
|
data = list(input_reader)
|
||
|
|
|
||
|
|
heading_row = data[0]
|
||
|
|
|
||
|
|
extra_item_index_list = []
|
||
|
|
for index, heading in enumerate(data[0]):
|
||
|
|
if "Another" in heading:
|
||
|
|
extra_item_index_list.append(index)
|
||
|
|
|
||
|
|
empty_column_headers = ["Status", "File Name", "Notes"]
|
||
|
|
|
||
|
|
course_details_list = ["Length of Reserve", "Full Name", "Course Number", "Course Title"]
|
||
|
|
|
||
|
|
item_details = ["Material", "Author", "Title:", "Page", "Chapter"]
|
||
|
|
|
||
|
|
course_details_index_list = []
|
||
|
|
|
||
|
|
|
||
|
|
for course_detail in course_details_list:
|
||
|
|
for index, heading in enumerate(data[0]):
|
||
|
|
if (re.search(course_detail, heading) and (index not in course_details_index_list)):
|
||
|
|
course_details_index_list.append(index)
|
||
|
|
break
|
||
|
|
|
||
|
|
extra_item_index_dict = {}
|
||
|
|
item_details_index_list = []
|
||
|
|
|
||
|
|
# need to split up item headings by item number
|
||
|
|
#
|
||
|
|
for index, heading in enumerate(data[0]):
|
||
|
|
for item_detail in item_details:
|
||
|
|
# needs handling for #10
|
||
|
|
if (re.search("[1-9]*" + item_detail, heading) and (index not in item_details_index_list) and (index not in course_details_index_list) and ("Journal" not in heading)):
|
||
|
|
item_details_index_list.append(index)
|
||
|
|
# break
|
||
|
|
# extra_item_index_dict.update()
|
||
|
|
|
||
|
|
|
||
|
|
# split extra item index ref list into chunks by item and add to dictionary
|
||
|
|
x = 0
|
||
|
|
item_number = 1
|
||
|
|
for i in range(x,len(item_details_index_list),len(item_details)):
|
||
|
|
x = i
|
||
|
|
extra_item_index_dict.update({item_number:item_details_index_list[x:x+5]})
|
||
|
|
item_number += 1
|
||
|
|
|
||
|
|
|
||
|
|
output_csv_list = []
|
||
|
|
output_csv_list.append(empty_column_headers + course_details_list + item_details)
|
||
|
|
|
||
|
|
# need to create function for new row/item creation. func must take in course row data along with indices of the next item info and/or the item data in a list. row parsing could be separate from new row creation?
|
||
|
|
for row in data[1:]:
|
||
|
|
course_info = []
|
||
|
|
new_row_first = []
|
||
|
|
for item in empty_column_headers:
|
||
|
|
new_row_first.append("")
|
||
|
|
course_info.append("")
|
||
|
|
for index in course_details_index_list:
|
||
|
|
course_detail = row[index]
|
||
|
|
course_info.append(course_detail)
|
||
|
|
new_row_first.append(course_detail)
|
||
|
|
# for index in extra_item_index_dict[1]:
|
||
|
|
# new_row_first.append(row[index])
|
||
|
|
# print(new_row_first)
|
||
|
|
for extra_item_number, extra_item in enumerate(extra_item_index_list):
|
||
|
|
if row[extra_item] == "Yes":
|
||
|
|
# print(extra_item_number)
|
||
|
|
create_row(row, extra_item_number, extra_item_index_dict, course_info, output_csv_list)
|
||
|
|
else:
|
||
|
|
break
|
||
|
|
# output_csv_list.append(new_row)
|
||
|
|
# output_csv_list.append(new_row_first)
|
||
|
|
|
||
|
|
# for i in item_details_index_list:
|
||
|
|
# print(data[0][i])
|
||
|
|
|
||
|
|
|
||
|
|
with open("./output.csv", "w", encoding="utf8", newline='') as output_file:
|
||
|
|
write = csv.writer(output_file)
|
||
|
|
write.writerows(output_csv_list)
|