import sys, os
import pandas as pd
from gchat_eml import Gchat, locate_eml
work_dir = sys.argv[1]
corpus_folder = sys.argv[2]
files = locate_eml(work_dir)
if corpus_folder[-1] != '/':
corpus_folder += '/'
print corpus_folder
l = len(files) / 5
for (ii, f) in enumerate(files):
try:
gc = Gchat(f)
gc.corpus_writer_yearly(corpus_folder + gc.msg_from_address)
except:
print 'uwotmate'
pass
if ii % l == 0 and ii != 0:
print ('%d percent finished..' % ((ii // l) * 20) )