def do_register(self):
if self.__has_logged():
# We directly let that user log in
return True, "Logged"
try:
db = tingDB_service.get_config_db()
users = db["users"]
# We first lookup if the user has been registered
user_cursor = users.find({"name": self.__username})
if user_cursor.count() > 0:
return False, "This user name has been registered."
# Prepare for user record
raw_password = self.__password
encrypted_password = utils.encrypt_string(raw_password)
new_user = {"name": self.__username, "password": encrypted_password, "privilege": "normal"}
users.insert(new_user)
register_user_cache(self)
return True, "Register successfully."
except Exception, e:
print e.message
def __save_servers():
"""
Save the plan instances into database.
"""
global __servers_queue
config_db = tingDB_service.get_config_db()
for plan in __servers_queue:
plan_doc = plan["server_doc"]
_type = plan["type"]
plan_collection = config_db[_type]
plan_collection.insert(plan_doc)
def do_login(self):
if self.__has_logged():
return True, "Logged"
try:
db = tingDB_service.get_config_db()
users = db["users"]
users = users.find({"name": self.__username})
# results is in the type of Cursor
if users.count() == 1:
user = users.next()
raw_password = self.__password
encrypted_password = utils.encrypt_string(raw_password)
if user["password"] == encrypted_password:
register_user_cache(self)
return True, "Corrected"
self.__privilege = user["privilege"]
except Exception, e:
# Connect to database failed
print e.message
