archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Ensure unique username on user creation

Browse Source
This commit is contained in:
Francesco Siddi 2015-11-04 22:12:42 +01:00
parent 5f818fdc39
commit e9cc719f36
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
pillar/application/__init__.py
View File

@ -86,7 +86,22 @@ def validate_token():
users = app.data.driver.db['users'] users = app.data.driver.db['users']
email = validation['data']['user']['email'] email = validation['data']['user']['email']
db_user = users.find_one({'email': email}) db_user = users.find_one({'email': email})
# Ensure unique username
username = email.split('@')[0] username = email.split('@')[0]
def make_unique_username(username, index=1):
"""Ensure uniqueness of a username by appending an incremental
digit at the end of it.
"""
user_from_username = users.find_one({'username': username})
if user_from_username:
if index > 1:
index += 1
username = username[:-1]
username = "{0}{1}".format(username, index)
return make_unique_username(username, index=index)
return username
username = make_unique_username(username)
full_name = username full_name = username
if not db_user: if not db_user:
user_data = { user_data = {