def get_file_type(file):
df = normalize.parse_file(file)
column_list = [x for x in df.columns.values if not x.startswith('Unnamed:')]
file_type = ''
for t,l in file_types.items():
if column_list == l:
file_type = t
if file_type == '':
normalize.change_column_names(df)
normalize.merge_zhaiyao_and_operation(df)
if 'operation' in df.columns:
for o in df['operation']:
if o in [u'融资买入', u'融资利息扣款', u'融资利息']:
file_type = 'rzrq'
if file_type == '':
file_type = 'normal'
else:
file_type = 'unknown!!!'
return file_type
