# Compare 2 arrays of ip
import csv
import sys
res_csv_file = open('result.csv', 'wb')
fieldnames = [
    'start', 'end', 'time', 'IMSI', 'int_ip', 'int_port', 'trans_ip', 'trans_port', 'dst_ip', 'dst_port',
    'traffic']
reswriter = csv.DictWriter(res_csv_file, delimiter=';', fieldnames=fieldnames)
reswriter.writerow(dict((fn, fn) for fn in fieldnames))
src_csv_file = open(sys.argv[1], 'rb')
ip_csv_file = open(sys.argv[2], 'rb')
src_csv_dict = csv.DictReader(src_csv_file, delimiter=';', quotechar='"')
ip_csv_dict = csv.DictReader(ip_csv_file, delimiter=';', quotechar='"')
rows = list(src_csv_dict)
totalrows = len(rows)
print "Total values to check: %s " % (totalrows)
i = 0
src_csv_file.seek(0)
for req_ip in src_csv_dict:
    i += 1
    sys.stdout.write("Progress %s%%\r" % ((i * 100) / totalrows))
    sys.stdout.flush()
    ip_csv_file.seek(0)
    for host_ip in ip_csv_dict:
        if req_ip['dst_ip'] in host_ip.values():
            reswriter.writerow(req_ip)
print "Job completed"