import json
import os

root_path = '/Users/vegardjervell/Documents/7_semester/irrev_prosjekt/'


data = [None for f in os.listdir(root_path+'fluids')]
for i, filename in enumerate(os.listdir(root_path+'fluids')):
    with open(root_path+'fluids/'+filename) as infile:
        data[i] = json.load(infile)

latex_strs = [None for d in data]
with open(root_path+'comp_db.txt', 'w') as outfile:
    for i, file in enumerate(data):
        ID = file['ident']
        formula = file['formula']
        name = file['name'][0] + ''.join(a.lower() for a in file['name'][1:])
        aliases = ', '.join(file['aliases'])
        mole_wheight = str(file['mol_weight'])
        cas = file['cas_number']

        txt_str = '     '.join((ID, formula, name, aliases, cas, mole_wheight))
        outfile.write(txt_str + '\n')

        latex_strs[i] = ' & '.join((ID, formula, name, cas, mole_wheight))

with open(root_path+'latex_db_smal.txt', 'w') as outfile:
    for i in range(len(latex_strs)):
        #write_str = ' & '.join(latex_strs[i-1:i+1]) + '\\\ '
        outfile.write(latex_strs[i]+'\\\ '+'\n')