喃喃治愈你博客项目文档

 

 

主要功能代码如下：

from flask import Flask
from flask import render_template, request, redirect, url_for, session
import config, os
from functools import wraps
from datetime import datetime
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import or_, and_
from werkzeug.security import generate_password_hash, check_password_hash

app = Flask(__name__)
app.config.from_object(config)
db = SQLAlchemy(app)

# 收藏表
Collection = db.Table(
    'collection',
    db.Column('id', db.Integer, primary_key=True, autoincrement=True),
    db.Column('book_id', db.Integer, db.ForeignKey('question.id')),  # 评论对应的文章的id
    db.Column('collection', db.Integer, db.ForeignKey('user.id')),  # 收藏用户的id
    db.Column('createdate', db.DATETIME)  # 发布时间
)


class User(db.Model):
    __tablename__ = 'user'
    # 建立一个表user
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    username = db.Column(db.String(20), nullable=False)
    _password = db.Column(db.String(200), nullable=False)
    say = db.Column(db.String(50))
    icon = db.Column(db.String(50))
    collection = db.relationship('Question', secondary=Collection, backref=db.backref('user', lazy='dynamic'),
                                 lazy='dynamic')

    @property
    def password(self):
        return self._password

    @password.setter
    def password(self, row_password):
        self._password = generate_password_hash(row_password)

    def check_password(self, row_password):
        return check_password_hash(self._password, row_password)


class Comment(db.Model):
    __tablename__ = 'comment'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    question_id = db.Column(db.Integer, db.ForeignKey('question.id'))
    creat_time = db.Column(db.DateTime, default=datetime.now())
    detail = db.Column(db.TEXT, nullable=False)
    question = db.relationship('Question', backref=db.backref('comment'))
    author = db.relationship('User', backref=db.backref('comment', order_by=creat_time.desc))


class Question(db.Model):
    __tablename__ = 'question'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    title = db.Column(db.String(100), nullable=False)
    detail = db.Column(db.Text, nullable=False)
    creat_time = db.Column(db.DateTime, default=datetime.now)
    author_id = db.Column(db.Integer, db.ForeignKey('user.id'))
    cf = db.Column(db.Integer, db.ForeignKey('cf.id'))
    look = db.Column(db.Integer)
    click = db.Column(db.Integer)
    author = db.relationship('User', backref=db.backref('question'))
    cfClass = db.relationship('Cf', backref=db.backref('question'))


# 分类
class Cf(db.Model):
    __tablname__ = 'cf'
    id = db.Column(db.Integer, primary_key=True, autoincrement=True)  # 数据库唯识别id
    name = db.Column(db.String(30))  # 文章名称
    context = db.Column(db.TEXT)  # 分类内容


db.create_all()


# 退出
@app.route('/logout/')
def logout():
    session.pop('username')
    return redirect(url_for('index'))


# 上下文
@app.context_processor
def myContext():
    username = session.get('username')
    if username:
        user = User.query.filter(User.username == username).first()
    else:
        user = {}
    if user:
        return {'user_id': user.id, 'user': user}
    else:
        return {}


# 登陆
@app.route('/login/', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        username = request.form.get('username')
        password = request.form.get('password')
        user = User.query.filter(User.username == username).first()
        if user:
            if user.check_password(password):
                session['username'] = user.username
                session['user_id'] = user.id
                session.permanent = True
                # 重新定位到首页
                return redirect(url_for('index'))
            else:
                # 重新定位到注册
                return redirect(url_for('login'))
        else:
            return redirect(url_for('login'))


# 注册
@app.route('/regist/', methods=['GET', 'POST'])
def regist():
    if request.method == 'GET':
        # 打开注册页的模板
        return render_template('regist.html')
    else:  # 收到用户上传的信息
        username = request.form.get('username')
        password = request.form.get('password')
        user = User.query.filter(User.username == username).first()
        if user:
            return 'error:user exitst'
        else:
            user = User(username=username, password=password)
            db.session.add(user)  # 加入数据库
            db.session.commit()
            return redirect(url_for('login'))


# 定义一个装饰器出验证用户有是否是登陆
# 定义一个参数函数
def loginFirst(func):
    # 定义一个函数将其返回
    @wraps(func)
    def wrapper(*args, **kwargs):
        if session.get('username'):
            return func(*args, **kwargs)
        else:
            return redirect(url_for('login'))
            # 返回一个函数

    return wrapper


@app.route('/setPassword/<id>', methods=['GET', 'POST'])
@loginFirst
def setPassword(id):
    if request.method == 'GET':
        return render_template('setPassword.html')
    else:
        user = User.query.filter(User.id == id).first()
        if user:
            if user.check_password(request.form.get('old')):
                user.password = request.form.get('new1')
                db.session.commit()
                info = '修改成功'
            else:
                info = '原密码错误'
        else:
            info = '未知错误'
        return redirect(url_for('index', info=info))


# 发布问答
@app.route('/question', methods=['GET', 'POST'])
@loginFirst
def question():
    if request.method == 'GET':
        cf = Cf.query.all()
        return render_template('question.html', cf=cf)
    else:
        title = request.form.get('title')
        detail = request.form.get('detail')
        author_id = request.form.get('author_id')
        cf = request.form.get('cf')
        question = Question(title=title, detail=detail, author_id=author_id, cf=cf)
        db.session.add(question)  # 加入数据库
        db.session.commit()
        return redirect(url_for('index'))


# 上传头像
@app.route('/uploadLogo/<user_id>', methods=['GET', 'POST'])
def uploadLogo(user_id):
    user = User.query.filter(User.id == user_id).first()
    f = request.files['logo']
    basepath = os.path.dirname(__file__)  # 当前文件所在路径
    upload_path = os.path.join(basepath, 'static/uploads', f.filename)  # 注意：没有的文件夹一定要先创建，不然会提示没有该路径
    f.save(upload_path)
    user.icon = 'uploads/' + f.filename
    db.session.commit()
    return redirect(url_for('setPassword', id=user_id));


# 发布评论
@app.route('/answer/', methods=['GET', 'POST'])
def answer():
    if request.method == 'POST':
        question_id = request.form.get('question_id')
        author_id = request.form.get('author_id')
        detail = request.form.get('detail')
        comment = Comment(question_id=question_id, author_id=author_id, detail=detail)
        db.session.add(comment)
        db.session.commit()
        return redirect(url_for('detail', question_id=question_id))


# 某用户发布过的所有评论
@app.route('/commentaries/<user_id>')
def commentaries(user_id):
    userCommentaries = Comment.query.filter(Comment.answer_id == user_id).all()
    num = len(userCommentaries)
    id = session.get('user_id')
    if id:
        user = User.query.filter(User.id == id).first()
    else:
        user = {}
    return render_template('commentaries.html', userCommentaries=userCommentaries, user=user, num=num)


# 首页
@app.route('/')
def index():
    if request.args.get('info'):
        info = request.args.get('info')
    else:
        info = None;
    context = {
        'questions': Question.query.order_by('-creat_time').all(),
        'cf': Cf.query.all(),
        'info': info,
        'hot': Question.query.order_by('-click').all()[0:5]
    }
    return render_template('index.html', **context)


# 某用户发布过的所有评论
@app.route('/comment/<user_id>/<num>')
def comment(user_id, num):
    user = User.query.filter(User.id == user_id).first()
    content = {
        'comment': user.comment,
        'questions': user.question,
        'user2': user,
    }
    if (num == '1'):
        return render_template('subComment1.html', **content, title='全部问题')
    elif (num == '2'):
        return render_template('subComment2.html', **content)
    elif (num == '3'):
        return render_template('subComment3.html', **content)
    elif (num == '4'):
        content = {
            'comment': user.comment,
            'questions': user.collection.all(),
            'user2': user,
        }
        return render_template('subComment1.html', **content, title='我的收藏')
    else:
        return render_template('subComment1.html', **content)


@app.route('/c/<cf>')
def c(cf):
    content = {
        'questions': Question.query.filter(Question.cf == cf).order_by('-creat_time').all(),
        'cf': Cf.query.all(),
        'hot': Question.query.order_by('-click').all()[0:5]
    }
    return render_template('index.html', **content)


# 详情页
@app.route('/detail/<question_id>', methods=['GET', 'POST'])
@loginFirst
def detail(question_id):
    quest = Question.query.filter(Question.id == question_id).first()
    u = User.query.filter(User.id == session.get('user_id')).first()
    if request.method == 'POST':
        if request.form.get('click') == '1':
            quest.click = quest.click + 1
        if request.form.get('collection') == '1':
            user = u
            user.collection.append(quest)
            db.session.add(user)
    col = u.collection.filter_by(id=question_id).first()
    if col is None:
        col = {}
    comment = Comment.query.filter(Comment.question_id == question_id).order_by('-creat_time').all()
    quest.look = quest.look + 1
    content = {
        'ques': quest,
        'comment': comment,
        'col': col,
        'questions': Question.query.filter(Question.cf == quest.cf).all(),
    }
    return render_template('detail.html', **content)


# 模糊查找
@app.route('/search')
def search():
    qu = request.args.get('q')
    c = '' if request.args.get('c') == '' else request.args.get('c')
    y = '' if request.args.get('y') == '' else request.args.get('y')
    query = Question.query.filter(
        or_(
            Question.title.contains(qu),
            Question.detail.contains(qu),
        ),
        Question.cf.like('%' + c + '%'),
        Question.creat_time.like('%' + y + '%'),
    ).order_by('-creat_time').all()
    context = {
        'questions': query,
        'cf': Cf.query.all(),
        'hot': Question.query.order_by('-click').all()[0:5]
    }
    return render_template('index.html', **context)


if __name__ == '__main__':
    app.run()