initialize_graph.py 783 Bytes
import csv, re, sqlite3

con = sqlite3.connect("database_modfire.db")
cur = con.cursor()
f = open("adj_init.txt", "w+")
f2 = open("border_init.txt", "w+")

toWrite = ''


for i in range(0, 1406):
	cur.execute("Select ug2 from adj_int where ug1 like "+str(i))
	result = cur.fetchall()
	if(result == []):
		toWrite = "-1\n"
	else:
		for j in range(0, len(result)):	
			toWrite += str(result[j]).replace(')','').replace('(','')
		toWrite = toWrite[:-1]+"\n"
	f.write(toWrite)
	toWrite = ''
	
	cur.execute("Select front from adj_int where ug1 like "+str(i))
	result = cur.fetchall()
	if(result == []):
		toWrite = "-1\n"
	else:
		for j in range(0, len(result)):	
			toWrite += str(result[j]).replace(')','').replace('(','')
		toWrite = toWrite[:-1]+"\n"
	f2.write(toWrite)
	toWrite = ''