当前位置：首页 > news >正文

flask中写一个基础的sqlHelper类

news 2025/7/6 4:24:45

在这里插入图片描述

写一个SQLHelper类：

from flask_sqlalchemy import SQLAlchemydb = SQLAlchemy()class SQLHelper:@staticmethoddef add(record):db.session.add(record)return SQLHelper.session_commit()@staticmethoddef add_all(records):db.session.add_all(records)return SQLHelper.session_commit()@staticmethoddef delete(record):db.session.delete(record)return SQLHelper.session_commit()@staticmethoddef update():return SQLHelper.session_commit()@staticmethoddef query_all(model):return model.query.all()@staticmethoddef query_by_id(model, record_id):return model.query.get(record_id)@staticmethoddef query_by_field(model, **kwargs):return model.query.filter_by(**kwargs).all()@staticmethoddef 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}', 200else:return 'Post not found!', 404