Add spreadsheet

This commit is contained in:
Samuel Lazareanu
2023-06-05 10:45:33 +03:00
parent c03d7765b3
commit 31ad3921e6
2 changed files with 20 additions and 0 deletions

View File

@@ -169,3 +169,10 @@ def fix_text_syntax(font: str, text_file):
# close the file # close the file
write_file.close() write_file.close()
def add_sheets(file_names: str, output_path: str, customer_name: str, authors: str, quotes: str):
with open(f'{output_path}/{customer_name}.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow(["File Name", "Reference", "Verse"])
for i in range(len(file_names)):
writer.writerow([file_names[i], authors[i], quotes[i]])

View File

@@ -47,6 +47,11 @@ def create_posts(images_folder, text_file, quote_font, author_font, output_folde
image_num.append((random_for_image + i) % len(image_files)) image_num.append((random_for_image + i) % len(image_files))
random.shuffle(image_num) random.shuffle(image_num)
# Data for spreadsheet
spreadsheet_col1 = list()
spreadsheet_col2 = list()
spreadsheet_col3 = list()
# Creating folder for customer # Creating folder for customer
output_path = create_dirs(output_folder, customer_name) output_path = create_dirs(output_folder, customer_name)
@@ -79,11 +84,19 @@ def create_posts(images_folder, text_file, quote_font, author_font, output_folde
quote_font=quote_font, author_font=author_font, output_path=output_path, file_name=file_name, quote_font=quote_font, author_font=author_font, output_path=output_path, file_name=file_name,
logo_file=logo_file, customer_name=customer_name, author_text=author_text) logo_file=logo_file, customer_name=customer_name, author_text=author_text)
# Add to spreadsheet
spreadsheet_col1.append(file_name.strip("/"))
spreadsheet_col2.append(author_text)
spreadsheet_col3.append(quote_text)
end_time = time.time() end_time = time.time()
run_time = end_time - start_time run_time = end_time - start_time
run_time_average += run_time run_time_average += run_time
print(f"\033[0;34m DONE #{i}, Run time:", round(run_time, 2), "seconds! \033[0m", output_path) print(f"\033[0;34m DONE #{i}, Run time:", round(run_time, 2), "seconds! \033[0m", output_path)
helper.add_sheets(file_names=spreadsheet_col1, output_path=output_path, customer_name=customer_name,
authors=spreadsheet_col2, quotes=spreadsheet_col3)
if number_of_posts > 1: if number_of_posts > 1:
run_time_average /= number_of_posts run_time_average /= number_of_posts
end_time_total = time.time() end_time_total = time.time()