realms-wiki/realms/models.py

106 lines
2.6 KiB
Python
Raw Normal View History

2013-10-04 05:57:19 +03:00
import bcrypt
2013-12-03 01:50:19 +02:00
from sqlalchemy import Column, Integer, String, Time
from sqlalchemy.ext.declarative import declarative_base
2013-10-05 00:42:45 +03:00
from flask import session, flash
from flask.ext.login import login_user, logout_user
2013-12-03 01:50:19 +02:00
from realms.lib.util import gravatar_url, to_dict
2013-11-08 20:20:40 +02:00
from realms.lib.services import db
2013-12-03 01:50:19 +02:00
Base = declarative_base()
2013-10-02 04:50:48 +03:00
2013-10-03 17:58:07 +03:00
class CurrentUser():
id = None
def __init__(self, id):
self.id = id
if id:
user = User()
session['user'] = user.get_by_id(id)
2013-10-03 17:58:07 +03:00
def get_id(self):
return self.id
def is_active(self):
2013-10-05 00:42:45 +03:00
return True if self.id else False
2013-10-03 17:58:07 +03:00
def is_anonymous(self):
return False if self.id else True
def is_authenticated(self):
return True if self.id else False
2013-10-05 08:04:38 +03:00
@staticmethod
def get(key):
try:
return session['user'][key]
except KeyError:
return None
2013-10-03 17:58:07 +03:00
2013-12-03 01:50:19 +02:00
class Site(Base):
__tablename__ = 'sites'
id = Column(Integer, primary_key=True)
name = Column(String(100))
pages = Column(Integer)
views = Column(Integer)
created = Column(Time)
class User(Base):
__tablename__ = 'users'
id = Column(Integer, primary_key=True)
username = Column(String(100))
email = Column(String(255))
password = Column(String(255))
joined = Column(Time)
2013-10-02 04:50:48 +03:00
2013-10-03 17:58:07 +03:00
def get_by_email(self, email):
return to_dict(self.get_one(email, 'email'), True)
def get_by_username(self, username):
return to_dict(self.get_one(username, 'username'), True)
2013-10-02 04:50:48 +03:00
def login(self, login, password):
2013-10-03 17:58:07 +03:00
pass
@classmethod
def auth(cls, username, password):
u = User()
data = u.get_by_email(username)
if not data:
return False
2013-10-04 05:57:19 +03:00
if bcrypt.checkpw(password, data['password']):
cls.login(data['id'])
2013-10-03 17:58:07 +03:00
return True
else:
2013-10-05 00:42:45 +03:00
return False
@classmethod
def register(cls, username, email, password):
user = User()
email = email.lower()
if user.get_by_email(email):
flash('Email is already taken')
return False
if user.get_by_username(username):
flash('Username is already taken')
return False
# Create user and login
u = User.create(email=email,
username=username,
password=bcrypt.hashpw(password, bcrypt.gensalt(10)),
avatar=gravatar_url(email))
User.login(u.id)
2013-10-05 00:42:45 +03:00
@classmethod
2013-11-08 20:20:40 +02:00
def login(cls, id):
2013-10-05 00:42:45 +03:00
login_user(CurrentUser(id), True)
@classmethod
def logout(cls):
logout_user()
session.pop('user', None)