--- 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