maple-bbs/maple/user/models.py

#!/usr/bin/env python
# -*- coding=UTF-8 -*-
# **************************************************************************
# Copyright © 2016 jianglin
# File Name: models.py
# Author: jianglin
# Email: xiyang0807@gmail.com
# Created: 2016-05-20 13:24:19 (CST)
# Last Update:星期六 2016-6-25 0:13:28 (CST)
#          By:
# Description:
# **************************************************************************
from maple import db
from datetime import datetime
from flask_login import UserMixin
from werkzeug.security import generate_password_hash, \
    check_password_hash


class UserRole(db.Model):
    __tablename__ = 'user_role'
    id = db.Column(db.Integer, primary_key=True)
    user_id = db.Column(db.Integer, db.ForeignKey('users.id'))
    roles_id = db.Column(db.Integer, db.ForeignKey('roles.id'))


class Follow(db.Model):
    __tablename__ = 'follows'
    id = db.Column(db.Integer, primary_key=True)
    follower_id = db.Column(db.Integer,
                            db.ForeignKey('users.id'))
    following_user_id = db.Column(db.Integer,
                                  db.ForeignKey('users.id'))
    following_tag_id = db.Column(db.Integer,
                                 db.ForeignKey('tags.id',
                                               ondelete="CASCADE"))
    following_collect_id = db.Column(db.Integer,
                                     db.ForeignKey('collects.id',
                                                   ondelete="CASCADE"))
    followinf_topic_id = db.Column(db.Integer,
                                   db.ForeignKey('topics.id',
                                                 ondelete="CASCADE"))


class User(db.Model, UserMixin):
    __tablename__ = 'users'
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(49), unique=True)
    email = db.Column(db.String(81), unique=True)
    password = db.Column(db.String, nullable=False)
    is_superuser = db.Column(db.Boolean, default=False)
    is_confirmed = db.Column(db.Boolean, default=False)
    register_time = db.Column(db.DateTime, default=datetime.now())

    likes = db.relationship('Reply',
                            secondary='likes',
                            lazy='dynamic',
                            backref="likers"
                            )
    following_tags = db.relationship('Tags',
                                     secondary='follows',
                                     primaryjoin="User.id==follows.c.follower_id",
                                     lazy='dynamic',
                                     backref="followers",
                                     # cascade='all,delete-orphan',
                                     # single_parent=True,
                                     )
    following_topics = db.relationship('Topic',
                                       secondary='follows',
                                       primaryjoin="User.id==follows.c.follower_id",
                                       lazy='dynamic',
                                       backref="followers",
                                       # cascade='all,delete-orphan',
                                       # single_parent=True,
                                       )
    following_collects = db.relationship('Collect',
                                         secondary='follows',
                                         primaryjoin="User.id==follows.c.follower_id",
                                         lazy='dynamic',
                                         backref="followers",
                                         # cascade='all,delete-orphan',
                                         # single_parent=True,
                                         )
    following_users = db.relationship('User',
                                      secondary='follows',
                                      primaryjoin="User.id==follows.c.follower_id",
                                      secondaryjoin="User.id==follows.c.following_user_id",
                                      backref=db.backref(
                                          'followers', lazy='dynamic'),
                                      lazy='dynamic'
                                      )

    setting_id = db.Column(db.Integer,
                           db.ForeignKey('usersetting.id',
                                         ondelete="CASCADE"))
    setting = db.relationship("UserSetting",
                              backref="users",
                              cascade='all,delete',
                              uselist=False)

    infor_id = db.Column(db.Integer,
                         db.ForeignKey('userinfor.id',
                                       ondelete="CASCADE"))
    infor = db.relationship("UserInfor",
                            backref="users",
                            cascade='all,delete',
                            uselist=False)

    def __repr__(self):
        return '<User %r>' % self.username

    @staticmethod
    def set_password(password):
        pw_hash = generate_password_hash(password)
        return pw_hash

    def check_password(self, password):
        return check_password_hash(self.password, password)


class Role(db.Model):
    __tablename__ = 'roles'
    id = db.Column(db.Integer, primary_key=True)
    rolename = db.Column(db.String(81), nullable=False, default='unconfirmed')
    description = db.Column(db.String(255), nullable=True)
    users = db.relationship('User',
                            secondary='user_role',
                            backref=db.backref('roles', lazy='dynamic'))


class UserInfor(db.Model):
    __tablename__ = 'userinfor'
    id = db.Column(db.Integer, primary_key=True)
    # confirmed_time = db.Column(db.DateTime, nullable=True)
    # registered_time = db.Column(db.DateTime, nullable=False)
    # score = db.Column(db.Integer, nullable=False, default=100)
    avatar = db.Column(db.String)
    word = db.Column(db.Text, nullable=True)
    introduce = db.Column(db.Text, nullable=True)
    school = db.Column(db.String, nullable=True)

    # def __init__(self):
    #     self.registered_time = datetime.now()

    def __repr__(self):
        return "<UserInfor %r>" % self.id


class UserSetting(db.Model):
    '''
    1:all user
    2:logined user
    3:only own
    '''
    __tablename__ = 'usersetting'
    id = db.Column(db.Integer, primary_key=True)
    online_status = db.Column(db.Integer, nullable=False, default=1)
    topic_list = db.Column(db.Integer, nullable=False, default=1)
    rep_list = db.Column(db.Integer, nullable=False, default=1)
    ntb_list = db.Column(db.Integer, nullable=False, default=3)
    collect_list = db.Column(db.Integer, nullable=False, default=2)
    locale = db.Column(db.String(32), default='zh')
    timezone = db.Column(db.String(32), default='UTC')

    def __repr__(self):
        return "<UserSetting %r>" % self.id


class OpenID(db.Model):
    __tablename__ = 'openids'
    id = db.Column(db.Integer, primary_key=True)
    openid = db.Column(db.Integer, nullable=False)
    openid_type = db.Column(db.String, nullable=False)
    nickname = db.Column(db.String, nullable=False)
    email = db.Column(db.String)
    avatar = db.Column(db.String)
    user_id = db.Column(db.Integer,
                        db.ForeignKey('users.id',
                                      ondelete="CASCADE"))
    user = db.relationship('User',
                           backref=db.backref('openids',
                                              cascade='all,delete-orphan',
                                              lazy='dynamic'))

# class Invite(db.Model):
#     __tablename__ = 'invites'
#     id = db.Column(db.Integer, primary_key=True)
#     invite_id = db.Column(db.Integer,
#                           db.ForeignKey('users.id',
#                                         ondelete="CASCADE"))

#     invite = db.relationship("User",
#                              uselist=False,
#                              foreign_keys='Invite.invite_id',
#                              backref="invited"
#                              )

#     invited_id = db.Column(db.Integer,
#                            db.ForeignKey('users.id',
#                                          ondelete="CASCADE"))

#     invited = db.relationship("User",
#                               uselist=False,
#                               foreign_keys='Invite.invited_id',
#                               backref="invite"
#                               )

#     topic_id = db.Column(db.Integer,
#                          db.ForeignKey('topics.id',
#                                        ondelete="CASCADE"))
#     topic = db.relationship("Topic",
#                             uselist=False,
#                             backref="invited"
#                             )