from pandas import *

import os

import math

from AESCipher import AESCipher

def list_files(path):

return files

# Configuration

csv_dir = "/path/to/split/dir"

csv_files = list_files(os.path.join(csv_dir))

cypher_obj = AESCipher('2805')

header_row = ['col1', 'col2', 'col3', 'col4', 'col5', 'col6', 'col7',

'col8', 'col9', 'col10', 'col11', 'col12', 'col13', 'col14',

'col15']

columns_to_encrypt = ['col6', 'col7']

encrpyt_mapping = []

for col in columns_to_encrypt:

encrpyt_mapping.append(col)

encrpyt_mapping.append(str(col) + " encrypted_value")

for csv_file in csv_files:

print "Parsing csv file %s" % csv_file

mapping_df = DataFrame(columns=encrpyt_mapping)

data = read_csv(os.path.join(csv_dir, csv_file), error_bad_lines=False, names=header_row, dtype='unicode')

print "Encrypting of data in csv file %s is starting" % csv_file

for index, row in data.iterrows():

# i = 0

# print(index)

for col in columns_to_encrypt:

if isinstance(row[col], float):

if math.isnan(row[col]):

row[col] = str(row[col])

else:

row[col] = int(row[col])

encoded = cypher_obj.encrypt(row[col])

data.loc[index, col] = encoded

# print(encoded + " " + cypher_obj.decrypt(encoded))

# mapping_df.loc[index, encrpyt_mapping[i]] = row[col]

# mapping_df.loc[index, encrpyt_mapping[i+1]] = encoded

# i += 2

print "Encrypting of data in csv file %s is completed" % csv_file

print "Writing CSV file for encrypted data"

data.to_csv(os.path.join(csv_dir, "enc_" + csv_file), encoding='utf-8')

# mapping_df.to_csv(os.path.join(csv_dir, "mapping_" + csv_file), encoding='utf-8')