class Article(Text):
title = db.Column(db.String(50), nullable=False)
like = db.Column(db.Integer, default=0, nullable=False)
reviews = db.relationship('ArticleReview', backref='article')
def __repr__(self):
return '<Article %r>' % self.id
class Text(db.Model):
# 作为基类，将不创建表
__abstract__ = True
id = db.Column(db.Integer, autoincrement=True, primary_key=True)
body = db.Column(db.String(), nullable=True)
_datetime = db.Column(db.DateTime, default=datetime.now, nullable=False)
@declared_attr
def user_id(self):
return db.Column('author', db.ForeignKey('user.id'), nullable=False)
@property
def author(self):
author = User.query.get(self.user_id)
return author
@property
def datetime(self):
return self._datetime.strftime('%Y-%m-%d %H:%M')
@property
def username(self):
return self.author.username
@property
def icon_path(self):
return self.author.icon_path
class Role(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
users = db.relationship('User', back_populates='role')
permissions = db.relationship('Permission',
secondary=roles_permissions,
back_populates='roles')
@staticmethod
def init_role():
roles_permissions_map = {
'Locked': ['COLLECT'],
'User': ['POST', 'COLLECT', 'COMMENT'],
'Moderator': ['POST', 'COLLECT', 'COMMENT', 'MODERATE'],
'Administrator':
['POST', 'COLLECT', 'COMMENT', 'MODERATE', 'ADMINISTER']
}
for role_name in roles_permissions_map:
role = Role.query.filter_by(name=role_name).first()
if role is None:
role = Role(name=role_name)
db.session.add(role)
role.permissions = []
for permission_name in roles_permissions_map[role_name]:
permission = Permission.query.filter_by(
name=permission_name).first()
if permission is None:
permission = Permission(name=permission_name)
db.session.add(permission)
role.permissions.append(permission)
db.session.commit()
class Work_(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(50))
abstract = db.Column(db.Text)
time = db.Column(db.String(20))
company = db.Column(db.String(30))
class Tag(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), index=True, unique=True)
articles = db.relationship('Article',
secondary=tagging,
back_populates='tags')
class Menu(db.Model):
__tablename__ = 'menus'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
types = db.relationship('ArticleType', backref='menu', lazy='dynamic')
order = db.Column(db.Integer, default=0, nullable=False)
def sort_delete(self):
for menu in Menu.query.order_by(Menu.order).offset(self.order).all():
menu.order -= 1
db.session.add(menu)
@staticmethod
def insert_menus():
menus = [u'Web开发', u'数据库', u'网络技术', u'爱生活，爱自己', u'Linux世界', u'开发语言']
for name in menus:
menu = Menu(name=name)
db.session.add(menu)
db.session.commit()
menu.order = menu.id
db.session.add(menu)
db.session.commit()
@staticmethod
def return_menus():
menus = [(m.id, m.name) for m in Menu.query.all()]
menus.append((-1, u'不选择导航（该分类将单独成一导航）'))
return menus
def __repr__(self):
return '<Menu %r>' % self.name
class ArticleTypeSetting(db.Model):
__tablename__ = 'articleTypeSettings'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
protected = db.Column(db.Boolean, default=False)
hide = db.Column(db.Boolean, default=False)
types = db.relationship('ArticleType', backref='setting', lazy='dynamic')
@staticmethod
def insert_system_setting():
system = ArticleTypeSetting(name='system', protected=True, hide=True)
db.session.add(system)
db.session.commit()
@staticmethod
def insert_default_settings():
system_setting = ArticleTypeSetting(name='system',
protected=True,
hide=True)
common_setting = ArticleTypeSetting(name='common',
protected=False,
hide=False)
db.session.add(system_setting)
db.session.add(common_setting)
db.session.commit()
@staticmethod
def return_setting_hide():
return [(2, u'公开'), (1, u'隐藏')]
def __repr__(self):
return '<ArticleTypeSetting %r>' % self.name
class Follow(db.Model):
__tablename__ = 'follows'
follower_id = db.Column(db.Integer,
db.ForeignKey('comments.id'),
primary_key=True)
followed_id = db.Column(db.Integer,
db.ForeignKey('comments.id'),
primary_key=True)
class Question(db.Model):
__tablename__ = 'question'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(100), nullable=False)
content = db.Column(db.Text, nullable=False)
ques_time = db.Column(db.DateTime, default=datetime.now)
blogger_id = db.Column(db.Integer, db.ForeignKey('user.id'))
blogger = db.relationship('User', backref=db.backref('questions'))
class Blog(db.Model):
__tablename__ = 'blog'
id = db.Column(db.Integer, primary_key=True)
blog_title = db.Column(db.String(128), nullable=False)
blog_content = db.Column(db.Text)
create_time = db.Column(db.DateTime, default=datetime.now)
blogger_id = db.Column(db.Integer, db.ForeignKey('user.id'))
blogger = db.relationship('User', backref=db.backref('blogs'))
class Link(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
url = db.Column(db.String(200), unique=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
category_id = db.Column(db.Integer, db.ForeignKey('link_category.id'))
category = db.relationship('LinkCategory', back_populates='links')
class Book(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(100))
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
comment = db.Column(db.TEXT)
category_id = db.Column(db.Integer, db.ForeignKey('book_category.id'))
category = db.relationship('BookCategory', back_populates='books')
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
body = db.Column(db.TEXT)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
category = db.relationship('Category', back_populates='posts')
class Notification(db.Model):
id = db.Column(db.Integer, primary_key=True)
message = db.Column(db.Text, nullable=False)
is_read = db.Column(db.Boolean, default=False)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
receiver_id = db.Column(db.Integer, db.ForeignKey('user.id'))
receiver = db.relationship('User', back_populates='notifications')
class Post(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(60))
body = db.Column(db.Text)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
category_id = db.Column(db.Integer, db.ForeignKey('category.id'))
category = db.relationship('Category', back_populates='posts')
comments = db.relationship('Comment', back_populates='post',
cascade='all') #设置级联删除
class Answer(db.Model):
__tablename__ = 'answer'
id = db.Column(db.Integer, primary_key=True)
content = db.Column(db.Text, nullable=False)
answer_time = db.Column(db.DateTime, default=datetime.now)
question_id = db.Column(db.Integer, db.ForeignKey('question.id'))
answer_id = db.Column(db.Integer, db.ForeignKey('user.id'))
answer_name = db.relationship('User', backref=db.backref('answers'))
question = db.relationship('Question', backref=db.backref('answers',
order_by=id.desc()))
class Author(db.Model):
__tablename__ = 'authors'
id = db.Column(db.Integer, primary_key=True)
user_id = db.Column(ForeignKey('users.id'), nullable=False, unique=True)
user = relationship('User', back_populates='author')
articles = relationship('Article', back_populates='author')
def __str__(self):
return self.user.username
class Tag(db.Model):
__tablename__ = 'tags'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), nullable=False)
articles = relationship('Article',
secondary=article_tag_associations_table,
back_populates='tags')
def __str__(self):
return self.name
class Category(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
posts = db.relationship('Post', back_populates='category')
def delete(self):
default_category = Category.query.get(1)
posts = self.posts[:]
for post in posts:
post.category = default_category
db.session.delete(self)
db.session.commit()
class Category(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(30), unique=True)
articles = db.relationship('Article', back_populates='category')
def delete(self):
default = Category.query.get(1)
articles = self.articles[:]
for a in articles:
a.category = default
db.session.delete(self)
db.session.commit()
class ArticleType(db.Model):
__tablename__ = 'articleTypes'
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(64), unique=True)
introduction = db.Column(db.Text, default=None)
articles = db.relationship('Article',
backref='articleType',
lazy='dynamic')
menu_id = db.Column(db.Integer, db.ForeignKey('menus.id'), default=None)
setting_id = db.Column(db.Integer, db.ForeignKey('articleTypeSettings.id'))
@staticmethod
def insert_system_articleType():
articleType = ArticleType(
name=u'未分类',
introduction=u'系统默认分类，不可删除。',
setting=ArticleTypeSetting.query.filter_by(protected=True).first())
db.session.add(articleType)
db.session.commit()
@staticmethod
def insert_articleTypes():
articleTypes = [
'Python', 'Java', 'JavaScript', 'Django', 'CentOS', 'Ubuntu',
'MySQL', 'Redis', u'Linux成长之路', u'Linux运维实战', u'其它', u'思科网络技术',
u'生活那些事', u'工作那些事', u'感情那些事', 'Flask'
]
for name in articleTypes:
articleType = ArticleType(name=name,
setting=ArticleTypeSetting(name=name))
db.session.add(articleType)
db.session.commit()
@property
def is_protected(self):
if self.setting:
return self.setting.protected
else:
return False
@property
def is_hide(self):
if self.setting:
return self.setting.hide
else:
return False
# if the articleType does not have setting,
# its is_hie and is_protected property will be False.
def __repr__(self):
return '<Type %r>' % self.name
class BookCategory(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(60), unique=True)
books = db.relationship('Book', back_populates='category')
def delete(self):
default_category = BookCategory.query.get(1)
books = self.books[:]
for book in books:
book.category = default_category
db.session.delete(self)
db.session.commit()
class LinkCategory(db.Model):
id = db.Column(db.Integer, primary_key=True)
name = db.Column(db.String(60), unique=True)
links = db.relationship('Link', back_populates='category')
def delete(self):
default_category = LinkCategory.query.get(1)
links = self.links[:]
for link in links:
link.category = default_category
db.session.delete(self)
db.session.commit()
class BlogInfo(db.Model):
__tablename__ = 'blog_info'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64))
signature = db.Column(db.Text)
navbar = db.Column(db.String(64))
@staticmethod
def insert_blog_info():
blog_mini_info = BlogInfo(title=u"Jazpenn's Blog",
signature=u'一个程序猿,除了敲代码,还得找对象!',
navbar='inverse')
db.session.add(blog_mini_info)
db.session.commit()
class Article(db.Model):
__tablename__ = 'articles'
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(64), unique=True)
thumbnail = db.Column(db.String(255), default='')
content = db.Column(db.Text)
summary = db.Column(db.Text)
create_time = db.Column(db.DateTime, index=True, default=datetime.utcnow)
update_time = db.Column(db.DateTime, index=True, default=datetime.utcnow)
num_of_view = db.Column(db.Integer, default=0)
articleType_id = db.Column(db.Integer, db.ForeignKey('articleTypes.id'))
source_id = db.Column(db.Integer, db.ForeignKey('sources.id'))
comments = db.relationship('Comment', backref='article', lazy='dynamic')
@staticmethod
def generate_fake(count=100):
from sqlalchemy.exc import IntegrityError
from random import seed, randint
import forgery_py
seed()
articleType_count = ArticleType.query.count()
source_count = Source.query.count()
for i in range(count):
aT = ArticleType.query.offset(randint(0, articleType_count -
1)).first()
s = Source.query.offset(randint(0, source_count - 1)).first()
a = Article(title=forgery_py.lorem_ipsum.title(randint(3, 5)),
content=forgery_py.lorem_ipsum.sentences(
randint(15, 35)),
summary=forgery_py.lorem_ipsum.sentences(randint(2,
5)),
num_of_view=randint(100, 15000),
articleType=aT,
source=s)
db.session.add(a)
try:
db.session.commit()
except IntegrityError:
db.session.rollback()
@staticmethod
def add_view(article, db):
article.num_of_view += 1
db.session.add(article)
db.session.commit()
def __repr__(self):
return '<Article %r>' % self.title
class Collect(db.Model):
collector_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
collected_id = db.Column(db.Integer,
db.ForeignKey('article.id'),
primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
collector = db.relationship('User',
back_populates='collections',
lazy='joined')
collected = db.relationship('Article',
back_populates='collectors',
lazy='joined')
class Comment(db.Model):
id = db.Column(db.Integer, primary_key=True)
body = db.Column(db.Text)
author = db.Column(db.String(30))
email = db.Column(db.String(254))
reviewed = db.Column(db.Boolean, default=False)
from_admin = db.Column(db.Boolean, default=False)
timestamp = db.Column(db.DateTime, default=datetime.utcnow, index=True)
post_id = db.Column(db.Integer, db.ForeignKey('post.id'))
post = db.relationship('Post', back_populates='comments')
replied_id = db.Column(db.Integer, db.ForeignKey('comment.id'))
replied = db.relationship('Comment', back_populates='replies', remote_side=[id])
replies = db.relationship('Comment', back_populates='replied', cascade='all')
class User(UserMixin, db.Model):
__tablename__ = 'user'
id = db.Column(db.Integer, primary_key=True)
telephone = db.Column(db.String(11), nullable=False)
username = db.Column(db.String(20), nullable=False)
password = db.Column(db.String(100), nullable=False)
def __init__(self, *args, **kwargs):
telephone = kwargs.get('telephone')
username = kwargs.get('username')
password = kwargs.get('password')
self.telephone = telephone
self.username = username
self.password = generate_password_hash(password)
def check_password(self, password):
return check_password_hash(self.password, password)
class Follow(db.Model):
follower_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
followed_id = db.Column(db.Integer,
db.ForeignKey('user.id'),
primary_key=True)
timestamp = db.Column(db.DateTime, default=datetime.utcnow)
follower = db.relationship('User',
foreign_keys=[follower_id],
back_populates='following',
lazy='joined')
followed = db.relationship('User',
foreign_keys=[followed_id],
back_populates='followers',
lazy='joined')
class BlogView(db.Model):
__tablename__ = 'blog_view'
id = db.Column(db.Integer, primary_key=True)
num_of_view = db.Column(db.BigInteger, default=0)
@staticmethod
def insert_view():
view = BlogView(num_of_view=0)
db.session.add(view)
db.session.commit()
@staticmethod
def add_view(db):
view = BlogView.query.first()
view.num_of_view += 1
db.session.add(view)
db.session.commit()
