39 lines
1 KiB
Python
39 lines
1 KiB
Python
|
#!/usr/bin/python3
|
||
|
|
|
||
|
|
import os
|
||
|
|
import csv
|
||
|
|
import itertools
|
||
|
|
import shutil
|
||
|
|
import errno
|
||
|
|
|
||
|
|
# TODO: filter out quotes which are formatted incorrectly before reorganizing
|
||
|
|
|
||
|
|
quotes = []
|
||
|
|
for root, dirs, files in os.walk('quotes/'):
|
||
|
|
for name in files:
|
||
|
|
(base, ext) = os.path.splitext(name)
|
||
|
|
if ext == '.csv':
|
||
|
|
full_name = os.path.join(root, name)
|
||
|
|
with open(full_name) as csvfile:
|
||
|
|
csvreader = csv.reader(csvfile, delimiter=',', quotechar='`', skipinitialspace=True)
|
||
|
|
quotes += list(csvreader)
|
||
|
|
|
||
|
|
# TODO: back up the quotes directory and implement recovery if the quote writing fails
|
||
|
|
shutil.rmtree('quotes')
|
||
|
|
|
||
|
|
for quote in quotes:
|
||
|
|
# path: ./quotes/author/work/
|
||
|
|
path = 'quotes/' + quote[1] + '/' + quote[3] + '/'
|
||
|
|
|
||
|
|
# https://stackoverflow.com/a/600612/11736197
|
||
|
|
try:
|
||
|
|
os.makedirs(path)
|
||
|
|
except OSError as exc:
|
||
|
|
if exc.errno == errno.EEXIST and os.path.isdir(path):
|
||
|
|
pass
|
||
|
|
else:
|
||
|
|
raise
|
||
|
|
|
||
|
|
with open(path + 'quotes.csv', 'a') as csvfile:
|
||
|
|
writer = csv.writer(csvfile, delimiter=',', quotechar='`')
|
||
|
|
writer.writerow(quote)
|