You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.2 KiB
Python
52 lines
1.2 KiB
Python
import datetime as dt
|
|
from apistar import Component, Settings, http, Route, Include
|
|
from apistar.backends.sqlalchemy_backend import Session
|
|
from apistar_jwt.authentication import get_jwt
|
|
from apistar_jwt.token import JWT
|
|
|
|
from .models.schema import User
|
|
|
|
# /auth/register
|
|
# /auth/login
|
|
# /auth/logout
|
|
# /auth/user
|
|
|
|
auth_components = [
|
|
Component(JWT, init=get_jwt)
|
|
]
|
|
|
|
|
|
def login(settings: Settings, json_data: http.RequestData, session: Session):
|
|
user_id = json_data.get('email')
|
|
password = json_data.get('password')
|
|
|
|
user = session.query(User).filter_by(email=user_id).one()
|
|
if not user.check_password(password):
|
|
error = {'error': 'Password auth failed'},
|
|
return http.Response(error, status=401, headers={'WWW-Authenticate': 'Bearer'})
|
|
|
|
SECRET = settings['JWT'].get('SECRET')
|
|
|
|
payload = {
|
|
'exp': dt.datetime.utcnow() + dt.timedelta(days=0, minutes=60),
|
|
'iat': dt.datetime.utcnow(),
|
|
'sub': user.id
|
|
}
|
|
|
|
token = JWT.encode(payload, secret=SECRET)
|
|
|
|
data = {
|
|
'status': 'success',
|
|
'message': 'Successfully logged in.',
|
|
'auth_token': token
|
|
}
|
|
|
|
return data
|
|
|
|
|
|
routes = [
|
|
Route('/login', 'POST', login)
|
|
]
|
|
|
|
auth_routes = [Include('/auth', routes)]
|