blob: 7396e52c18d97b81fb9495750d7800d665c83a0a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
import logging
from werkzeug.security import generate_password_hash, check_password_hash
from .abstract import AbstractController
from newspipe.models import User
logger = logging.getLogger(__name__)
class UserController(AbstractController):
_db_cls = User
_user_id_key = "id"
def _handle_password(self, attrs):
if attrs.get("password"):
attrs["pwdhash"] = generate_password_hash(attrs.pop("password"))
elif "password" in attrs:
del attrs["password"]
def check_password(self, user, password):
return check_password_hash(user.pwdhash, password)
def create(self, **attrs):
self._handle_password(attrs)
return super().create(**attrs)
def update(self, filters, attrs):
self._handle_password(attrs)
return super().update(filters, attrs)
|