db_import_lists.py

from application import db, dataLibs
from application.model import *
import os
nowdir = os.path.abspath(os.path.dirname(__file__))

f = open(os.path.join(nowdir, 'mar_total.txt'), 'r')
while True:
	line = f.readline()
	if line:
		new_matter = matterDB(line[0:line.find('|')].strip(), line[line.find('|') + 1:].strip())
		#new_matter.save()
		db.session.add(new_matter)
	else:
		break
f.close()

fr = open(os.path.join(nowdir, 'all_names.txt'), 'r')
while True:
	line = fr.readline().strip()
	if line:
		line2 = fr.readline()
		new_flora = floraDB(line[2 : len(line) - 1], line2.strip())
		#new_flora.save()
		db.session.add(new_flora)
	else:
		break
fr.close()

medium_list = ['Glucose&Ammonium phosphate, monobasic.txt', 'sucrose&Ammonium phosphate, monobasic.txt']
medium_path = os.path.join(nowdir, 'medium')
for filename in medium_list:
	file = open(os.path.join(medium_path, filename), 'r')
	new_medium = mediumDB(filename[0 : filename.rfind('.')])
	while True:
		line = file.readline()
		# print line
		if line:
			# print line[line.rfind('\t') + 1 : ].strip()
			new_matter = matterDB.query.filter_by(matter_code = line[line.rfind('\t') + 1 : ].strip()).first()
			if new_matter:
				# print new_matter.id
				new_medium.matters = libs_list_insert(new_medium.matters, new_matter.id)
				new_medium.concentration = libs_dict_insert(new_medium.concentration, new_matter.id, line[0 : line.find('\t')])
		else:
			break
	#new_medium.save()
	db.session.add(new_medium)
	file.close()

plasmid_list = ['pSB1C3.txt', 'pSB1K3.txt', 'pSB1T3.txt', 'pSB1AK3.txt', 'pSB1AT3.txt']
plasmid_path = os.path.join(nowdir, 'plasmid')
for filename in plasmid_list:
	file = open(os.path.join(plasmid_path, filename), 'r')
	sequence = ''
	while True:
		line = file.readline().strip()
		if line:
			sequence += line.upper()
		else:
			break
	new_plasmid = plasmidDB(filename[0 : filename.rfind('.')], sequence)
	#new_plasmid.save()
	db.session.add(new_plasmid)
	file.close()
	
db.session.commit()