lib/my_db.py
changeset 66 a0aa025ee6a2
parent 65 b975a3d7606a
child 70 1798859f7f6c
--- a/lib/my_db.py	Tue Dec 30 22:32:10 2008 -0500
+++ b/lib/my_db.py	Tue Dec 30 23:16:05 2008 -0500
@@ -9,15 +9,16 @@
 
 class MyDbConnexion:
 	def __init__(self, config):
+		self.userid = None 
 		self.cnx = pgdb.connect(database = config.db_name,
 			host = config.db_host+':%d'%config.db_port,
 			user = config.db_user, password = config.db_passwd)
 		self.crs = self.cnx.cursor();
 	def connect_user(self, name):
 		if (self.userid == None):
-			self.userid = self._get_user_id(username)
+			self.userid = self._get_user_id(name)
 			if self.userid == None:
-				self.userid = self._create_user(username)
+				self.userid = self._create_user(name)
 			else:
 				self.userid = self.userid[0]
 		return self.userid
@@ -67,7 +68,7 @@
 			DELETE FROM preferences WHERE userid = %d;
 			DELETE FROM keywords WHERE userid = %d;
 			DELETE FROM db_sequence WHERE userid = %d;
-			""" % ((userid,)*5))
+			""" % ((userid,)*6))
 		self.cnx.commit()
 	def get_preferences(self):
 		self.crs.execute("""
@@ -398,4 +399,7 @@
 
 def connect(config, username):
 	db = MyDbConnexion(config)
-	db.connect_user(username)
+	userid = db.connect_user(username)
+	if userid == None:
+		raise "Can't connect user!"
+	return db