diff -r c1f1491f098c -r d79722064d8d lib/my_db.py --- a/lib/my_db.py Wed Dec 31 00:00:09 2008 -0500 +++ b/lib/my_db.py Wed Dec 31 00:07:25 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