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