diff options
Diffstat (limited to 'tools/mmd/MmdDb.py')
-rw-r--r-- | tools/mmd/MmdDb.py | 27 |
1 files changed, 22 insertions, 5 deletions
diff --git a/tools/mmd/MmdDb.py b/tools/mmd/MmdDb.py index 1065bb6..cba395d 100644 --- a/tools/mmd/MmdDb.py +++ b/tools/mmd/MmdDb.py @@ -10,6 +10,7 @@ class Db: def __init__ (self): self.conn = sqlite3.connect ('/usr/local/mmd/db/mmd.db') self.conn.text_factory = str + self.conn.row_factory = sqlite3.Row self.cursor = self.conn.cursor () def sessionFind (self, session_id): @@ -44,9 +45,17 @@ class Db: # # user related methods # - def userCreate (self, user_id, firstname, lastname, email, callsign, password, confreg): - self.cursor.execute ('INSERT INTO user (id, firstname, lastname, email, callsign, password, confreg) VALUES (?,?,?,?,?,?,?)', - (user_id, firstname, lastname, email, callsign, password, confreg)) + def userCreate (self, user_id, firstname, lastname, email, callsign, password, code, regtimeout): + self.cursor.execute ('INSERT INTO user (id, firstname, lastname, email, callsign, password, code, regtimeout) VALUES (?,?,?,?,?,?,?,?)', + (user_id, firstname, lastname, email, callsign, password, code, regtimeout)) + self.conn.commit () + + def userConfirm (self, code): + self.cursor.execute ('UPDATE user SET status="valid" WHERE code=?', (code, )) + self.conn.commit () + + def userCancel (self, code): + self.cursor.execute ('DELETE FROM user WHERE code=?', (code, )) self.conn.commit () def userDelete (self, user_id): @@ -57,6 +66,14 @@ class Db: self.cursor.execute ('SELECT * FROM user WHERE id=?', (user_id,)) return self.cursor.fetchone () + def userFindEmail (self, email): + self.cursor.execute ('SELECT * FROM user WHERE email=?', (email,)) + return self.cursor.fetchone () + + def userFindPending (self, code): + self.cursor.execute ('SELECT * FROM user WHERE code=? and status="pending"', (code,)) + return self.cursor.fetchone () + def close (self): self.cursor.close () self.conn.close () @@ -66,11 +83,11 @@ if __name__ == "__main__": print db.sessionFind ('1234567890') db.sessionInit ('1234567890', 'jogi@mur.at', '1') data = db.sessionFind ('1234567890') - print data[0], data[1] + print data['id'], data['email'] db.sessionDelete ('1234567890') print db.sessionFind ('1234567890') print db.userFindId ('1') - db.userCreate ('1', 'Jogi', 'Hofmüller', 'jogi@mur.at', '', '', '') + db.userCreate ('1', 'Jogi', 'Hofmüller', 'jogi@mur.at', '', '', '', '') print db.userFindId ('1') db.userDelete ('1') |