如何在 Flask 中设置继承的方法视图以对 sqlalchemy 模型执行 CRUD 操作



为了避免为我的模型编写 api 方法视图。 我希望创建一种更优雅的继承方式来做到这一点。

class ModelCrudAPI(MethodView):
    def __init__(self, model):
      self.model = model
    def get(self):
        store_id = request.args.get("store_id", None, int)
        waiters = self.model.query.filter_by(store_id=store_id).all()
        if waiters is None or store_id is None:
            return jsonify({'items': [], 'store_id': store_id}), 204
        # TODO: change personnel_data to list
        else:
            items = []
            for waiter in waiters:
                items.append(row2dict(waiter))
            return jsonify({"items": items, 'store_id': store_id})

我之前在这个问题中设置了类似的继承。上一个关于方法视图中的继承和修饰器的问题

请注意,我想将 sqlalcdhemy 模型传递给我的类。

第一种情况

class Test(ModelCrudAPI):
    def __init__(self, model):
        super(Test, self).__init__(model)
api.add_url_rule("/test", view_func=Test(MYMODEL).as_view("test"), methods=['GET','POST','PUT','DELETE'])

错误:

flask debuger :TypeError: __init__() takes exactly 2 arguments (1 given)

第二种情况(

class Test(ModelCrudAPI):
    def __init__(self, MYMODEL):
        super(Test, self).__init__(MYMODEL)
api.add_url_rule("/test", view_func=Test.as_view("test"), methods=['GET','POST','PUT','DELETE'])

错误:

flask debuger :TypeError: __init__() takes exactly 2 arguments (1 given)

我真的很感激任何帮助。 这将清理我的代码。

编辑此作品

class Test(ModelCrudAPI):
    def __init__(self):
        ModelCrudAPI.__init__(self, Personnel)

api.add_url_rule("/test", view_func=Test.as_view("test"), methods=['GET','POST','PUT','DELETE'])

as_view是一个classmethod,它将提供给它的参数转发给类的构造函数。 所以你所要做的就是将你的模型作为第二个参数传递给as_view

class Test(ModelCrudAPI):
    def __init__(self, model):
        super(Test, self).__init__(model)
api.add_url_rule("/test", view_func=Test.as_view("test", MYMODEL), methods=['GET','POST','PUT','DELETE'])

最新更新