68 lines
2.5 KiB
Python
68 lines
2.5 KiB
Python
|
import re
|
||
|
|
from datetime import datetime
|
||
|
|
from szurubooru import config, db, errors
|
||
|
|
from szurubooru.util import auth, misc
|
||
|
|
|
||
|
|
def create_user(name, password, email):
|
||
|
|
''' Create an user with given parameters and returns it. '''
|
||
|
|
user = db.User()
|
||
|
|
update_name(user, name)
|
||
|
|
update_password(user, password)
|
||
|
|
update_email(user, email)
|
||
|
|
user.access_rank = config.config['service']['default_user_rank']
|
||
|
|
user.creation_time = datetime.now()
|
||
|
|
user.avatar_style = db.User.AVATAR_GRAVATAR
|
||
|
|
return user
|
||
|
|
|
||
|
|
def update_name(user, name):
|
||
|
|
''' Validate and update user's name. '''
|
||
|
|
name = name.strip()
|
||
|
|
name_regex = config.config['service']['user_name_regex']
|
||
|
|
if not re.match(name_regex, name):
|
||
|
|
raise errors.ValidationError(
|
||
|
|
'Name must satisfy regex %r.' % name_regex)
|
||
|
|
user.name = name
|
||
|
|
|
||
|
|
def update_password(user, password):
|
||
|
|
''' Validate and update user's password. '''
|
||
|
|
password_regex = config.config['service']['password_regex']
|
||
|
|
if not re.match(password_regex, password):
|
||
|
|
raise errors.ValidationError(
|
||
|
|
'Password must satisfy regex %r.' % password_regex)
|
||
|
|
user.password_salt = auth.create_password()
|
||
|
|
user.password_hash = auth.get_password_hash(user.password_salt, password)
|
||
|
|
|
||
|
|
def update_email(user, email):
|
||
|
|
''' Validate and update user's email. '''
|
||
|
|
email = email.strip() or None
|
||
|
|
if not misc.is_valid_email(email):
|
||
|
|
raise errors.ValidationError(
|
||
|
|
'%r is not a vaild email address.' % email)
|
||
|
|
user.email = email
|
||
|
|
|
||
|
|
def update_rank(user, rank, authenticated_user):
|
||
|
|
rank = rank.strip()
|
||
|
|
available_access_ranks = config.config['service']['user_ranks']
|
||
|
|
if not rank in available_access_ranks:
|
||
|
|
raise errors.ValidationError(
|
||
|
|
'Bad access rank. Valid access ranks: %r' % available_access_ranks)
|
||
|
|
if available_access_ranks.index(authenticated_user.access_rank) \
|
||
|
|
< available_access_ranks.index(rank):
|
||
|
|
raise errors.AuthError('Trying to set higher access rank than one has')
|
||
|
|
user.access_rank = rank
|
||
|
|
|
||
|
|
def bump_login_time(user):
|
||
|
|
''' Update user's login time to current date. '''
|
||
|
|
user.last_login_time = datetime.now()
|
||
|
|
|
||
|
|
def reset_password(user):
|
||
|
|
''' Reset password for an user. '''
|
||
|
|
password = auth.create_password()
|
||
|
|
user.password_salt = auth.create_password()
|
||
|
|
user.password_hash = auth.get_password_hash(user.password_salt, password)
|
||
|
|
return password
|
||
|
|
|
||
|
|
def get_by_name(session, name):
|
||
|
|
''' Retrieve an user by its name. '''
|
||
|
|
return session.query(db.User).filter_by(name=name).first()
|