Added delete/disable/enable/check

This commit is contained in:
2010-03-05 14:29:10 +00:00
committed by root
parent 9a21f1c8a2
commit 1285453ba0

View File

@@ -16,7 +16,9 @@ class MediawikiService(BaseService):
SQL_ADD_USER = r"INSERT INTO user (user_name, user_password, user_newpassword, user_options, user_email) VALUES (%s, %s, '', '', '')" SQL_ADD_USER = r"INSERT INTO user (user_name, user_password, user_newpassword, user_options, user_email) VALUES (%s, %s, '', '', '')"
SQL_DEL_USER = r"DELETE FROM user WHERE username = %s" SQL_DIS_USER = r"UPDATE user SET user_password = '', user_email = '' WHERE username = %s"
SQL_ENABLE_USER = r"UPDATE user SET user_password = %s WHERE user_name = %s"
SQL_CHECK_USER = r"SELECT user_name from user WHERE user_name = %s"
def __init__(self): def __init__(self):
@@ -50,19 +52,27 @@ class MediawikiService(BaseService):
def delete_user(self, username): def delete_user(self, username):
""" Delete a user """ """ Delete a user """
self._dbcursor.execute(self.SQL_DEL_USER, [username]) self.disable_user(username)
self._db.connection.commit()
def disable_user(self, username): def disable_user(self, username):
""" Disable a user """ """ Disable a user """
pass self._dbcursor.execute(self.SQL_DIS_USER, [username])
self._db.connection.commit()
def enable_user(self, username, password): def enable_user(self, username, password):
""" Enable a user """ """ Enable a user """
pwhash = self._gen_mw_hash(password)
self._dbcursor.execute(self.SQL_ENABLE_USER, [pwhash, username.strip().capitalize()])
pass pass
def check_user(self, username): def check_user(self, username):
""" Check if the username exists """ """ Check if the username exists """
pass self._dbcursor.execute(self.SQL_CHECK_USER, [username.strip().capitalize()])
row = self._dbcursor.fetchone()
if row:
return True
return False
ServiceClass = 'MediawikiService' ServiceClass = 'MediawikiService'