import _mysql

def gen_dict(result):
	d = {}
	for i in result:
		k = int(i[0])
		d[k] = i[1:]

	return d


class PartDB:
	def __init__(self):
		self.db = _mysql.connect("localhost", "part-db", "PARTdb", "part-db")
		self.name = "part-db"

	def add_footprints(self, dict_footprints):
		sql_insert = "INSERT INTO `footprints`"
		q = sql_insert + "( `id` , `name`) VALUES "
		l = dict_footprints.keys()
		for i in l[:-2]:
			q += "( NULL, '%s' ), " % i

		q += "( NULL, '%s' ) " % l[-1]

		return self.db.query(q)
			
	def add_parts(self, dict_parts):
		format = "( NULL, '65535', '%s', '0', '0', '%s', '0' )"
		sql_insert = "INSERT INTO `parts`"

		q = sql_insert + """(
`id` ,
`id_category` ,
`name` ,
`instock` ,
`mininstock` ,
`comment` ,
`id_footprint`
) VALUES """


		l = dict_parts.keys()
		for i in l[:-2]:
			q += format % (i, i)
			q += ","
		q += format % (i, i)

		return self.db.query(q)