flask中写一个基础的sqlHelper类

news2024/10/6 16:30:13

在这里插入图片描述

写一个SQLHelper类：

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class SQLHelper:

    @staticmethod
    def add(record):
        db.session.add(record)
        return SQLHelper.session_commit()

    @staticmethod
    def add_all(records):
        db.session.add_all(records)
        return SQLHelper.session_commit()

    @staticmethod
    def delete(record):
        db.session.delete(record)
        return SQLHelper.session_commit()

    @staticmethod
    def update():
        return SQLHelper.session_commit()

    @staticmethod
    def query_all(model):
        return model.query.all()

    @staticmethod
    def query_by_id(model, record_id):
        return model.query.get(record_id)

    @staticmethod
    def query_by_field(model, **kwargs):
        return model.query.filter_by(**kwargs).all()

    @staticmethod
    def session_commit():
        try:
            db.session.commit()
        except Exception as e:
            db.session.rollback()
            reason = str(e)
            return reason

在这个扩展的SQLHelper类中，我们添加了以下几个方法：

add_all()：一次性添加多条记录到数据库。
query_all()：获取表中所有的记录。
query_by_id()：根据ID查询记录。
query_by_field()：根据字段查询记录。

这个类提供了更全面的数据库操作，但是请注意，根据你的实际需求，你可能需要添加更多的方法。

下面是如何使用这个SQLHelper类的例子：

from flask import Flask, request
from models import Post
from sqlhelper import SQLHelper

@app.route('/add_post', methods=['POST'])
def add_post():
    title = request.form.get('title')
    content = request.form.get('content')
    post = Post(title=title, content=content)
    SQLHelper.add(post)
    return 'Post added successfully!', 200

@app.route('/get_post', methods=['GET'])
def get_post():
    post_id = request.args.get('id')
    post = SQLHelper.query_by_id(Post, post_id)
    if post:
        return f'Title: {post.title}, Content: {post.content}', 200
    else:
        return 'Post not found!', 404