Updated SQLextract.py

other_things/scripts/SQLextract.py

@@ -6,6 +6,80 @@ import sys, os, sqlite3
 home = os.environ.get("HOME")
 db = "%s/rollback/rollback.sqlite" % (home)
 
+#############
+## Node IDs
+#
+
+def get_node_ids():
+	try:
+		conn = sqlite3.connect(db)
+		cur = conn.cursor()
+	except sqlite3.OperationalError as err:
+		print(err)
+		sys.exit(1)
+	try:
+		cur.execute("SELECT * FROM node")
+	except sqlite3.OperationalError as err:
+		print(err)
+		sys.exit(1)
+	else:
+		result = cur.fetchall()
+	finally:
+		cur.close()
+		conn.close()
+	return(result)
+
+node_ids = get_node_ids()
+
+def get_node(nodeid):
+	if not nodeid:
+		return None
+	else:
+		return node_ids[nodeid-1][1]
+
+###############
+## Player IDs
+#
+
+def get_player_ids():
+	try:
+		conn = sqlite3.connect(db)
+		cur = conn.cursor()
+	except sqlite3.OperationalError as err:
+		print(err)
+		sys.exit(1)
+	try:
+		cur.execute("SELECT * FROM actor")
+	except sqlite3.OperationalError as err:
+		print(err)
+		sys.exit(1)
+	else:
+		result = cur.fetchall()
+	finally:
+		cur.close()
+		conn.close()
+	return result
+
+player_ids = get_player_ids()
+
+def get_player(id):
+	if id:
+		return player_ids[id-1][1][len("player:"):]
+	else:
+		return None
+
+########################################################################
+## Utilities
+#
+
+def ston(a):
+	"""
+		Returns a string equal to a or None
+	"""
+	if a:
+		return str(a)
+	else:
+		return a
 
 def select_all_nodes(stamp):
 	try:
@@ -27,83 +101,6 @@ def select_all_nodes(stamp):
 		conn.close()
 	return(result)
 
-
-
-
-def select_nodes_id():
-	try:
-		conn = sqlite3.connect(db)
-		cur = conn.cursor()
-	except sqlite3.OperationalError as err:
-		print(err)
-		sys.exit(1)
-	try:
-		cur.execute("SELECT * FROM node")
-	except sqlite3.OperationalError as err:
-		print(err)
-		sys.exit(1)
-	else:
-		result = cur.fetchall()
-	finally:
-		cur.close()
-		conn.close()
-	return(result)
-
-
-def get_node(nodeid):
-	if not nodeid:
-		return None
-	try:
-		return select_nodes_id()[nodeid-1][1]
-	except IndexError:
-		print(nodeid)
-
-
-def select_player_id(table, player_name):
-	try:
-		conn = sqlite3.connect(db)
-		cur = conn.cursor()
-	except:
-		print("Problème avec la base de données")
-		return("")
-	try:
-		cur.execute("SELECT id FROM " + str(table) + " WHERE name=:name", {"name": "player:%s" % player_name})
-	except sqlite3.OperationalError as err:
-		print(err)
-		return ("")
-	else:
-		result = cur.fetchall()
-	finally:
-		cur.close()
-		conn.close()
-	return(result)
-
-def get_player(id):
-	try:
-		conn = sqlite3.connect(db)
-		cur = conn.cursor()
-	except:
-		print("problème avec la base de données")
-		return ("")
-	try:
-		cur.execute("SELECT name from actor WHERE id=%d" % id)
-	except sqlite3.OperationalError as err:
-		print(err)
-		return ("")
-	else:
-		result = cur.fetchall()[0][0]
-	finally:
-		cur.close()
-		conn.close()
-	return result[len("player:"):]
-
-
-def ston(a):
-	if a:
-		return str(a)
-	else:
-		return a
-
 def write_list(nodes):
 	try:
 		name = "database-output.txt"
@@ -113,31 +110,40 @@ def write_list(nodes):
 		sys.exit(1)
 
 	for node in nodes:
+		# Init bracket and basic datas
 		f.write("[")
 		f.write("id=%d" % node[0])
 		f.write(";actor=%s" % get_player(node[1]))
 		f.write(";type=%d" % node[3])
+		# Inventory Manipulation
 		f.write(";list=%s" % ston(node[4]))
 		f.write(",index=%s" % ston(node[5]))
 		f.write(",add=%s" % ston(node[6]))
 		f.write(",stacknode=%s" % get_node(node[7]))
 		f.write(",stackquantity=%s" % ston(node[8]))
 		f.write(",nodemeta=%s" % ston(node[9]))
+		# Position
 		f.write(";x=%s,y=%s,z=%s" % (node[10], node[11], node[12]))
+		# Old node
 		f.write(";oldnode=%s" % get_node(node[13]))
 		f.write(",oldparam1=%s" % node[14])
 		f.write(",oldparam2=%s" % ston(node[15]))
 		f.write(",oldmeta=%s" % ston(node[16]))
+		# New node
 		f.write(":newnode=%s" % get_node(node[17]))
 		f.write(",newparam1=%s" % ston(node[18]))
 		f.write(",newparam2=%s" % ston(node[19]))
 		f.write(",newmeta=%s" % ston(node[20]))
+		# Ending
 		f.write("]\n")
 
 	f.close()
 
 
 
+########################################################################
+## Main
+#
 
 if __name__ == '__main__':
 	#select all nodes player as set i where time >= time