shixun_linshi/woniu2/module/comment.py

from datetime import datetime

from flask import session

from main import db
from module.thumb import Thumb


class Comment(db.Model):
    __tablename__ = 'comment'
    commentid = db.Column(db.Integer, primary_key=True)
    userid = db.Column(db.String(64), db.ForeignKey('user.userid'))
    articleid = db.Column(db.String(64), db.ForeignKey('article.articleid'))
    content = db.Column(db.String(64), nullable = False) # 正文内容
    ipaddr = db.Column(db.String(64), nullable = False) # 地址
    replyid = db.Column(db.Integer, nullable=False) # 如果是普通评论，为0，如果是回复的评论，显示commentid
    agreecount = db.Column(db.Integer, nullable=False) # 赞成该评论的数量
    opposecount = db.Column(db.Integer, nullable=False) # 反对该评论的数量
    hidden = db.Column(db.Integer, nullable=False) # 默认为0，不隐藏
    createtime = db.Column(db.DateTime, nullable=False)  # 使用 DateTime 类型，并设置默认值为当前 UTC 时间
    updatetime = db.Column(db.DateTime, nullable=False)

    def is_upComment(self,articleid,commentid):
        return Thumb().is_upComment(articleid,commentid)


    # 关联 Users 表
    user = db.relationship('Users', backref=db.backref('comment_user', lazy=True))



    # 新增评论
    @classmethod
    def add_comment(cls,articleid,content,ipaddr):
        userid = session.get('userid')
        comment = Comment(userid=userid,articleid=articleid,content=content,ipaddr=ipaddr,replyid=0,
                          agreecount=0,opposecount=0,hidden=0,createtime=datetime.utcnow(),updatetime=datetime.utcnow())
        try:
            db.session.add(comment)
            db.session.commit()
            return True
        except Exception as e:
            print("新增失败，文章id:", articleid, "错误:", e)
            return False

    # 根据文章编号查询所有的评论(未分页)
    @classmethod
    def find_comment_by_articleid(cls,articleid):
        cls.query.filter_by(hidden=0).order_by(Comment.commentid.desc()).all()


    # 查询用户和评论信息
    @classmethod
    def find_comment_with_user(cls,articleid):
        options = db.joinedload(cls.user)
        return cls.query.options(options).filter_by(articleid=articleid,hidden=0).order_by(Comment.createtime.desc()).all()

    # 评论的赞同数量加一
    @classmethod
    def update_add_agreecount(cls, commentid):
        cls.query.filter_by(commentid=commentid).update({cls.agreecount: cls.agreecount+1})
        db.session.commit()
        return True

    # 评论的赞同数量减一
    @classmethod
    def update_sub_agreecount(cls, commentid):
        cls.query.filter_by(commentid=commentid).update({cls.agreecount: cls.agreecount - 1})
        db.session.commit()

    # 评论的反对数量加一
    @classmethod
    def update_add_opposecount(cls, commentid):
        cls.query.filter_by(commentid=commentid).update({cls.opposecount: cls.opposecount + 1})
        db.session.commit()

    # 评论的反对数量减一
    @classmethod
    def update_sub_opposecount(cls, commentid):
        cls.query.filter_by(commentid=commentid).update({cls.opposecount: cls.opposecount - 1})
        db.session.commit()


    # 评论隐藏
    @classmethod
    def update_hide_comment(cls, commentid):
        cls.query.filter_by(commentid=commentid).update({cls.hide:1})
        db.session.commit()


    @classmethod
    def find_coment_by_id(cls, commentid,userid):
        '''
        根据userid查找用户
        :param userid: userid
        :return: 查找到的对象
        '''
        return cls.query.filter(cls.userid == userid,cls.commentid == commentid).first()

    # 删除评论信息
    @classmethod
    def do_delete_comment(cls, commentid):
        result = cls.query.filter_by(commentid=commentid).first()
        if result:
            db.session.delete(result)
            db.session.commit()