33 lines
777 B
Python
33 lines
777 B
Python
|
import uuid
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
from app import app, db
|
||
|
|
from ki.models import User, Token
|
||
|
|
|
||
|
|
|
||
|
|
def auth(username, password):
|
||
|
|
auth_file_path = app.config["KI_DATA_DIR"] + "/auth.yml"
|
||
|
|
|
||
|
|
with open(auth_file_path, "r") as auth_file_stream:
|
||
|
|
users = yaml.safe_load(auth_file_stream)
|
||
|
|
|
||
|
|
if username not in users:
|
||
|
|
return None
|
||
|
|
|
||
|
|
auth_user = users[username]
|
||
|
|
|
||
|
|
if auth_user["password"] != password:
|
||
|
|
return None
|
||
|
|
|
||
|
|
user = User.query.filter(User.auth_id.__eq__(username)).first()
|
||
|
|
|
||
|
|
if user is None:
|
||
|
|
user = User(nickname=username, auth_id=username)
|
||
|
|
db.session.add(user)
|
||
|
|
|
||
|
|
token = Token(token=str(uuid.uuid4()), user=user)
|
||
|
|
db.session.add(token)
|
||
|
|
db.session.commit()
|
||
|
|
|
||
|
|
return token
|