3

我有一个已经在这里提到的问题 https://youtrack.jetbrains.com/issue/PY-44557

但我还没有找到解决方案。我正在使用 Python Flask 和 SQLAlchemy 来创建 API 服务。

这是我的进口：

from flask import Flask, jsonify, request
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import Column, Integer, String, Float
import os
from flask_marshmallow import Marshmallow


app = Flask(__name__)
basedir = os.path.abspath(os.path.dirname(__file__))
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///'+os.path.join(basedir, 'dealers.db')


db = SQLAlchemy(app)
ma = Marshmallow(app)

这是我的数据模型：

class Car(db.Model):
    __tablename__ = 'cars'
    car_id = Column(Integer, primary_key=True)
    car_name = Column(String)
    car_type = Column(String)
 

class CarSchema(ma.Schema):
    class Meta:
        fields = ('car_id', 'car_name', 'car_type')

car_schema = CarSchema()  
cars_schema = CarSchema(many=True)

使用.query时，我的 UDF 出现问题/警告

def cars():
    car_list = Car.query.all()  # *Unresolved attribute reference 'query' for class 'Car'
    result = cars_schema.dump(car_list)
    return jsonify(result)
4

2 回答 2

3

我发现了如何使用type-hinting stubs为这个问题制作一个 hacky 和部分解决方案。

创建派生自db.Model一个单独文件的抽象 python 类和一个具有相同名称的“存根”文件

例子：

抽象.py：

class AbstractBaseModel(db.Model):
    __abstract__ = True

摘要.pyi：

from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import orm
from sqlalchemy.ext.declarative import DeclarativeMeta


db: SQLAlchemy = SQLAlchemy()


class _FBase(db.Model):
    query: orm.Query


class AbstractBaseModel(_FBase, DeclarativeMeta):
    ...

然后制作从 abstract.AbstractBaseModel 而不是 db.Model 派生的模型，例如：

from abstract import AbstractBaseModel

class Car(AbstractBaseModel):
    __tablename__ = 'cars'
    car_id = Column(Integer, primary_key=True)
    car_name = Column(String)
    car_type = Column(String)

我还创建了一个示例项目来测试此解决方法

于 2021-04-08T14:48:04.147 回答
0

修改class Car(db.Model):为：

class Car(db.Model):
    __tablename__ = 'cars'
    car_id = db.Column(db.Integer, primary_key=True)
    car_name = db.Column(db.String) # it is better to give string length for example db.String(100)
    car_type = db.Column(db.String)

您应该在哪里定义Columns 与db

于 2021-01-01T15:33:08.010 回答