class Comment(CommonInfoMixin):
"""
There is a possibility to comment Stickers or Boards. This model store
all comments.
"""
__tablename__ = 'comments'
text = db.Column(db.Text())
object_id = Column(db.Integer)
object_type = db.Column(ChoiceType(OBJECT_TYPE))
@property
def object(self):
"""
Return commented object.
"""
MODEL = {
BOARD_TYPE: Board,
SPRINT_TYPE: Sprint,
STICKER_TYPE: Sticker
}
return MODEL[self.object_type].filter_by(
object_id=self.object_id
).scalar()
def __repr__(self):
return '<Comment {} | {}>'.format(self.author, self.creation_date)
class Label(PKMixin):
"""
Represents status of task for particular `Sticker` object.
"""
__tablename__ = 'labels'
color = Column(db.String(7))
css_class = db.Column(ChoiceType(CSS_CLASS))
status = db.Column(ChoiceType(TASK_STATUS))
sticker_set = db.relationship('Sticker', backref='label')
def __repr__(self):
return '<Label {} | {}>'.format(self.status, self.color)
class Profile(UserMixin, PKMixin):
"""
Users' model.
"""
__tablename__ = 'profiles'
username = db.Column(db.String(64), unique=True, index=True)
email = db.Column(db.String(64), unique=True, index=True)
passwd = db.Column(db.String(128))
avatar = Column(db.String(256), nullable=True)
display_name = Column(db.String(128))
# @desc: self and a ManyToMany relation.
friends = db.relationship(
'Profile',
secondary=friends,
primaryjoin=('Profile.id==friends.c.left_profile_id'),
secondaryjoin=('Profile.id==friends.c.right_profile_id'),
backref='left_friends')
motivation_quote = Column(db.String(256), nullable=True)
# user = models.OneToOneField(User, verbose_name=_('user'))
comments = db.relationship('Comment', backref='profile')
desk_author = db.relationship('Desk',
uselist=False,
backref='desk_author',
foreign_keys='Desk.author')
desk_owner = db.relationship('Desk',
uselist=False,
backref='desk_owner',
foreign_keys='Desk.owner_id')
board_set = db.relationship('Board', backref='board_author')
sprint_set = db.relationship('Sprint', backref='sprint_author')
sticker_set = db.relationship('Sticker', backref='sticker_author')
activity_set = db.relationship('Activities', backref='activity_who')
def __repr__(self):
return '<Profile {}>'.format(self.display_name)
def __str__(self):
return '{}'.format(self.display_name)
@property
def password(self):
"""
Don't allow to see user's password.
"""
raise AttributeError('You cannot see a password!')
@password.setter
def password(self, password):
"""
Set user's password.
"""
self.passwd = generate_password_hash(password)
def verify_password(self, password):
"""
Check if given password complies to user's one.
"""
return check_password_hash(self.passwd, password)
def generate_reset_token(self):
"""
Generate unique token.
"""
token = TimedJSONWebSignatureSerializer(
current_app.config['SECRET_KEY'],
current_app.config['TOKEN_EXPIRATION'],
)
return token.dumps({'reset': self.id})
def reset_password(self, in_token, new_password):
"""
Reset password.
"""
token = TimedJSONWebSignatureSerializer(
current_app.config['SECRET_KEY'])
try:
data = token.loads(in_token)
except:
return False
if data.get('reset') != self.id:
return False
self.password = new_password
db.session.add(self)
return True
class PKMixin(db.Model):
__abstract__ = True
id = db.Column(db.Integer, primary_key=True)
from datetime import datetime
from flask import current_app
from flask.ext.login import UserMixin, AnonymousUserMixin
from itsdangerous import TimedJSONWebSignatureSerializer
from werkzeug.security import generate_password_hash, check_password_hash
from beattime.config import db, auth_manager
from beattime.fields import Column
friends = db.Table(
'friends',
db.Column('left_profile_id',
db.Integer,
db.ForeignKey('profiles.id'),
primary_key=True),
db.Column('right_profile_id',
db.Integer,
db.ForeignKey('profiles.id'),
primary_key=True))
class PKMixin(db.Model):
__abstract__ = True
id = db.Column(db.Integer, primary_key=True)
class AnonymousUser(AnonymousUserMixin):
"""
Anonymous User's model.