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.
111 lines
3.7 KiB
111 lines
3.7 KiB
#!/usr/bin/env python |
|
# -*- encoding: utf-8 -*- |
|
|
|
import sys |
|
import logging |
|
import json |
|
import hmac |
|
import datetime |
|
|
|
from flask import Flask |
|
|
|
|
|
# ---------------------------------------------------------------------- |
|
# Config |
|
# ---------------------------------------------------------------------- |
|
class Settings(object): |
|
SQLALCHEMY_DATABASE_URI = 'sqlite://./luncho.db3' |
|
DEBUG = True |
|
|
|
log = logging.getLogger('luncho.server') |
|
|
|
# ---------------------------------------------------------------------- |
|
# Load the config |
|
# ---------------------------------------------------------------------- |
|
app = Flask(__name__) |
|
app.config.from_object(Settings) |
|
app.config.from_envvar('LUCNHO_CONFIG', True) |
|
|
|
# ---------------------------------------------------------------------- |
|
# Database |
|
# ---------------------------------------------------------------------- |
|
from flask.ext.sqlalchemy import SQLAlchemy |
|
db = SQLAlchemy(app) |
|
|
|
userGroups = db.Table('user_groups', |
|
db.Column('username', |
|
db.String, |
|
db.ForeignKey('user.username')), |
|
db.Column('group_id', |
|
db.Integer, |
|
db.ForeignKey('group.id'))) |
|
|
|
|
|
class User(db.Model): |
|
username = db.Column(db.String, primary_key=True) |
|
fullname = db.Column(db.String, nullable=False) |
|
passhash = db.Column(db.String, nullable=False) |
|
token = db.Column(db.String) |
|
issued_date = db.Column(db.Date) |
|
validated = db.Column(db.Boolean, default=False) |
|
created_at = db.Column(db.DateTime, nullable=False) |
|
groups = db.relationship('Group', |
|
secondary=userGroups, |
|
backref=db.backref('groups', lazy='dynamic')) |
|
|
|
def __init__(self, username, fullname, passhash, token=None, |
|
issued_date=None, verified=False): |
|
self.username = username |
|
self.fullname = fullname |
|
self.passhash = passhash |
|
self.token = token |
|
self.verified = verified |
|
self.created_at = datetime.datetime.now() |
|
|
|
def get_token(self): |
|
"""Generate a user token or return the current one for the day.""" |
|
# create a token for the day |
|
self.token = self._token() |
|
db.session.commit() |
|
return self._token() |
|
|
|
def valid_token(self, token): |
|
"""Check if the user token is valid.""" |
|
return (self.token == self._token()) |
|
|
|
def _token(self): |
|
"""Generate a token with the user information and the current date.""" |
|
phrase = json.dumps({'username': self.username, |
|
'issued_date': datetime.date.today().isoformat()}) |
|
return hmac.new(self.created_at.isoformat(), phrase).hexdigest() |
|
|
|
|
|
class Group(db.Model): |
|
id = db.Column(db.Integer, primary_key=True) |
|
name = db.Column(db.String, nullable=False) |
|
owner = db.Column(db.String, db.ForeignKey('user.username')) |
|
|
|
def __init__(self, name, owner): |
|
self.name = name |
|
self.owner = owner |
|
|
|
# ---------------------------------------------------------------------- |
|
# Blueprints |
|
# ---------------------------------------------------------------------- |
|
from blueprints.index import index |
|
from blueprints.users import users |
|
from blueprints.token import token |
|
from blueprints.groups import groups |
|
|
|
app.register_blueprint(index, url_prefix='/') |
|
app.register_blueprint(token, url_prefix='/token/') |
|
app.register_blueprint(users, url_prefix='/user/') |
|
app.register_blueprint(groups, url_prefix='/group/') |
|
|
|
|
|
# ---------------------------------------------------------------------- |
|
# Main |
|
# ---------------------------------------------------------------------- |
|
if __name__ == '__main__': |
|
log.warning('Use manage.py to run the server.') |
|
sys.exit(1)
|
|
|