def create_mapped_file(filepath, data):
_print(('Writing new file: %s' % filepath),
style='BRIGHT', foreground_color='GREEN')
headers = get_new_columns().keys()
with open(filepath, 'wb') as f:
csv_writer = csv.DictWriter(f, fieldnames=headers)
csv_writer.writeheader()
for row in data:
csv_writer.writerow(row)
def map_file(filepath):
_print(('Mapping file found here: %s' % filepath),
style='BRIGHT', foreground_color='GREEN')
new_columns = get_new_columns()
new_data = []
with open(filepath, 'rU') as f:
csv_reader = csv.DictReader(f)
for row in csv_reader:
result = {}
for col in new_columns:
if col == 'Campaign ID':
result[col] = ''
elif col == 'Campaign Status':
result[col] = 'Active'
else:
result[col] = row.get(col, '')
new_data.append(result)
return new_data
import os
import glob
import csv
import shutil
import sys
from xlsxwriter.workbook import Workbook
from image_machine.util import _print
reload(sys)
sys.setdefaultencoding('ISO-8859-1')
csv_files = glob.glob(os.path.join('.', 'tmp', '*.csv'))
for csv_file in csv_files:
_print(('Found file:\t\t{0}'.format(csv_file)), foreground_color='GREEN')
new_file_name = '{0}.xlsx'.format(csv_file.split('.csv')[0])
workbook = Workbook(new_file_name)
worksheet = workbook.add_worksheet()
with open(csv_file, 'rb') as f:
reader = csv.reader(f)
for r, row in enumerate(reader):
for c, col in enumerate(row):
worksheet.write(r, c, col)
_print(('Created file:\t\t{0}'.format(new_file_name)), foreground_color='GREEN')
_print(('*' * 80), foreground_color='GREEN')
workbook.close()
shutil.move(csv_file, 'archived')
