You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.3 KiB
85 lines
2.3 KiB
#!/usr/bin/env python |
|
# -*- encoding: utf-8 -*- |
|
|
|
"""User management.""" |
|
|
|
import logging |
|
|
|
from flask import Blueprint |
|
from flask import request |
|
from flask import jsonify |
|
|
|
from sqlalchemy.exc import IntegrityError |
|
|
|
from luncho.helpers import ForceJSON |
|
from luncho.helpers import auth |
|
|
|
from luncho.server import User |
|
from luncho.server import db |
|
|
|
from luncho.exceptions import LunchoException |
|
|
|
LOG = logging.getLogger('luncho.blueprints.users') |
|
|
|
users = Blueprint('users', __name__) |
|
|
|
|
|
class UsernameAlreadyExistsException(LunchoException): |
|
"""The username is already taken.""" |
|
def __init__(self): |
|
super(UsernameAlreadyExistsException, self).__init__() |
|
self.status = 409 |
|
self.message = 'Username already exists' |
|
|
|
|
|
@users.route('', methods=['PUT']) |
|
@ForceJSON(required=['username', 'full_name', 'password']) |
|
def create_user(): |
|
"""Create a new user. Request must be: |
|
{ "username": "username", "full_name": "Full Name", "password": "hash" }""" |
|
json = request.get_json(force=True) |
|
|
|
try: |
|
new_user = User(username=json['username'], |
|
fullname=json['full_name'], |
|
passhash=json['password'], |
|
verified=False) |
|
|
|
db.session.add(new_user) |
|
db.session.commit() |
|
except IntegrityError: |
|
raise UsernameAlreadyExistsException() |
|
|
|
return jsonify(status='OK') |
|
|
|
|
|
@users.route('/', methods=['POST']) |
|
@ForceJSON() |
|
@auth |
|
def update_user(): |
|
"""Update user information. Request can have the following fields: |
|
{ "full_name": "Full name", "password": "hash" } |
|
Any other field will be ignored; only fields that need to be changed |
|
must be send.""" |
|
json = request.get_json(force=True) |
|
user = request.user |
|
|
|
if 'full_name' in json: |
|
LOG.debug('Fullname = {fullname}'.format(fullname=json['full_name'])) |
|
user.fullname = json['full_name'] |
|
|
|
if 'password' in json: |
|
LOG.debug('Passhash = {password}'.format(password=json['password'])) |
|
user.passhash = json['password'] |
|
|
|
db.session.commit() |
|
return jsonify(status='OK') |
|
|
|
|
|
@users.route('/', methods=['DELETE']) |
|
@auth |
|
def delete_user(): |
|
"""Delete a user. No confirmation is send.""" |
|
db.session.delete(request.user) |
|
db.session.commit() |
|
return jsonify(status='OK')
|
|
|