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')