WireChem-PROTO/convert.py

#!/usr/bin/env python
# -*- coding: utf-8 -*- 

import shelve
import csv
import os
import pyglet
from pyglet import image
from os.path import expanduser

global items,worlds,finished

'''********************* fonctions Chargement ANCIENNE ****************************************************************'''

def readpref(file):
	global finished
	with open(file, 'a+') as f:
		try:
			finished=list(csv.reader(f,delimiter=';'))[0]
			f.close()
		except:
			print "no"

def loaditems(n,file):
	global items
	with open(file, 'rb') as f:
		liste=list(csv.reader(f,delimiter=';'))
		if len(liste)!=0:
			for i in range(1,len(liste)):
				items[liste[i][0]]={}
				for j in range(1,len(liste[i])):
					if liste[i][j][:1]=="#":
						items[liste[i][0]][liste[0][j]]=int(liste[i][j][1:])
					elif liste[i][j][:1]=="[":
						atemp=liste[i][j][1:-1].split(",")
						items[liste[i][0]][liste[0][j]]=[int(atemp[k]) for k in range(len(atemp))]
					elif liste[i][j][:1]=="{":
						atemp=items[liste[i][0]][liste[0][j]]=liste[i][j][1:-1].split(",")
						items[liste[i][0]][liste[0][j]]=[atemp[k] for k in range(len(atemp))]
					elif liste[i][j][:2]=="0x":
						items[liste[i][0]][liste[0][j]]=int(liste[i][j][2:],16)
					elif liste[i][j][:1]=="&":
						items[liste[i][0]][liste[0][j]]=float(liste[i][j][1:])
					elif liste[i][j][:1]=="@":
						items[liste[i][0]][liste[0][j]]=items[liste[i][j][1:]]		
					elif liste[i][j][:1]=="%":					
						items[liste[i][0]][liste[0][j]]=image.load(liste[i][j][1:])	
					else:
						items[liste[i][0]][liste[0][j]]=liste[i][j]
				if n!=0:
					items[liste[i][0]]['value']=n+i-1
				items[items[liste[i][0]]['value']]=liste[i][0]
		f.close()
		return len(liste)-1
		
def readgrid(file):
	global unroll,mousel,mousem,mouser,cout,selected,sizex,sizey,world_old,world_new,world_art,items,zoom,play,stat,cycle,nrj,rayon,tech,decx,decy,unroll,stat,victory,current,temp,debug,nom,descriptif,element,maxnrj,maxrayon,maxcycle,maxtemp
	try:
		with open(file, 'rb') as f:
			liste=list(csv.reader(f,delimiter=';'))
			sizey=(len(liste)-1)/2
			sizex=len(liste[1])
			nom=liste[0][0]
			element=liste[0][1]
			descriptif=liste[0][2]
			debug=int(liste[0][3])
			tech=int(liste[0][7])
			cout=int(liste[0][8])		
			victemp=liste[0][9][1:len(liste[0][9])-1].split(",")
			victory=[int(victemp[k]) for k in range(len(victemp))]
			curtemp=liste[0][10][1:len(liste[0][10])-1].split(",")
			current=[int(curtemp[k]) for k in range(len(curtemp))]
			cycle=int(liste[0][11])
			nrj=int(liste[0][12])
			rayon=int(liste[0][13])
			temp=int(liste[0][14])
			maxcycle=int(liste[0][15])
			maxnrj=int(liste[0][16])
			maxrayon=int(liste[0][17])
			maxtemp=int(liste[0][18])
			world_new = [[int(liste[sizey-i][j]) for i in range(sizey)] for j in range(sizex)]
			world_art = [[int(liste[-i-1][j]) for i in range(sizey)] for j in range(sizex)]		
			f.close()
			return True
	except IOError:
		return False
		
'''********************* fonction ecriture NOUVELLE ****************************************************************'''

def sync():
	global Uworlds,finished
	write(gethome()+"/dbdata",["Uworlds","finished"])

def verifyhome():
	global Uworlds,finished
	if not os.path.exists(gethome()):
		os.makedirs(gethome())
	if not os.path.exists(gethome()+"/dbdata"):
		Uworlds=[[{0:0}]]
		finished=[(0,0)]
		sync()
	
def gethome():
	home = expanduser("~")+"/.wirechem"
	return home
			
def write(afile,var):
	d=shelve.open(afile,writeback=True)
	for k in var:
		d[k]=globals()[k]
	d.sync()
	d.close()	
	
items = {}
sizeworld=loaditems(int("0x40000", 16),"data/worlds.dat")	
loaditems(int("0x30000", 16),"data/elements2.dat")	
loaditems(int("0x10000", 16),"data/menus2.dat")	
loaditems(int("0x20000", 16),"data/menus.dat")
loaditems(0,"data/elements.dat")
worlds=[]
Uworlds=[]
verifyhome()
readpref("user/pref.dat")
linked=finished
finished=[]
for k in linked:
	if len(k)>4:
		finished.append((int(k[5]),int(k[7])))
for i in range(sizeworld):
	ele=items[items[int("0x40000",16)+i]]
	readgrid(ele['file'])
	print "situation", ele['world'],ele['grid'],len(worlds)
	while len(worlds)<=ele['world']:
		worlds.append(0)
		worlds[ele['world']]=[]
	while len(worlds[ele['world']])<=ele['grid']:
		worlds[ele['world']].append({})
	link=[]
	for k in ele['validate']:
		if len(k)>4:
			link.append((int(k[5]),int(k[7])))
	worlds[ele['world']][ele['grid']]={'nom':nom, 
'element':element,
'description':descriptif,
'_xx':ele['coordx'],
'_yy':ele['coordy'],
'video':ele['tuto']!="",
'link':link,
'tech':tech,
'cout':cout,
'victory':victory,
'current':current,
'cycle':cycle,
'nrj':nrj,
'rayon':rayon,
'temp':temp,
'maxcycle':maxcycle,
'maxnrj':maxnrj,
'maxrayon':maxrayon,
'maxtemp':maxtemp,
'world_new':world_new,
'world_art':world_art}

	world_art=[]
	if os.path.exists("user/"+ele['file']):
		print ele['file']
		readgrid("user/"+ele['file'])
		while len(Uworlds)<=ele['world']:
			Uworlds.append(0)
			Uworlds[ele['world']]=[]
		while len(Uworlds[ele['world']])<=ele['grid']:
			Uworlds[ele['world']].append({})
		Uworlds[ele['world']][ele['grid']]={'nom':nom, 
'element':element,
'description':descriptif,
'_xx':ele['coordx'],
'_yy':ele['coordy'],
'video':ele['tuto']!="",
'link':link,
'tech':tech,
'cout':cout,
'victory':victory,
'current':current,
'cycle':cycle,
'nrj':nrj,
'rayon':rayon,
'temp':temp,
'maxcycle':maxcycle,
'maxnrj':maxnrj,
'maxrayon':maxrayon,
'maxtemp':maxtemp,
'world_new':world_new,
'world_art':world_art}
write("dbdata",["worlds"])
write(gethome()+"/dbdata",["Uworlds","finished"])
f=open("dbsrc", 'wb+')
afile="""#!/usr/bin/env python
# -*- coding: utf-8 -*- 

import shelve

def write(afile,var):
	d=shelve.open(afile,writeback=True)
	for k in var:
		d[k]=globals()[k]
	d.sync()
	d.close()


global worlds

worlds="""
f.write(afile+str(worlds).replace(", '",",\n '").replace(", [",", \n\t\t\t[")+"""\nwrite("dbdata",["worlds"])""")
f.close()