如何将SqlAlchemy连接查询序列化为JSON?

时间:2016-03-12 01:36:58

标签: python json sqlalchemy flask-sqlalchemy

任何人都可以帮我设计/找到一个JSON编码器,它在我的sqlalchemy查询中使用连接时会起作用,并会序列化我指定的结果。

查询如下:data = db.session.query(Post).join(Users).filter(Post.area == id, Users.id == Post.user_id).all()

然后我按照这样做:json_object = json.dumps(data, cls=AlchemyEncoder)

传统上我使用自己的JSON编码器,可以在这里看到:

class AlchemyEncoder(json.JSONEncoder):
    def default(self, o):
        if isinstance(o.__class__, DeclarativeMeta):
            data = {}
            fields = o.__json__() if hasattr(o, '__json__') else dir(o)
            for field in [f for f in fields if not f.startswith('_') and f not in ['metadata', 'query', 'query_class']]:
                value = o.__getattribute__(field)
                try:
                    json.dumps(value)
                    data[field] = value
                except TypeError:
                    data[field] = None
            return data
        return json.JSONEncoder.default(self, o)

它引用的函数如下所示:

 def __json__(self):
        return ['id', 'user_id', 'message', 'media', 'time_uploaded', 'lifetime', 'votes', 'area']

但是我在表中加入了一些列,我希望将其包含在我创建的JSON对象中。我的JSON编码器显然只能使用我查询的第一个表中的列。

编辑:

以下是我的两个SQLAlchemy类:

class Post(Base):
    __tablename__ = "post"

    id = db.Column('id', UUIDType(binary=False), primary_key=True)
    user_id = db.Column('user_id', UUIDType(binary=False), db.ForeignKey('users.id'))
    votes = db.Column('votes', db.Integer)
    message = db.Column('message', db.Unicode)
    media = db.Column('media', db.Unicode)
    lifetime = db.Column('lifetime', db.Integer)
    time_uploaded = db.Column('time_uploaded', db.DateTime, server_default=func.now())
    area = db.Column('area', db.Unicode)

    user = db.relationship('Users', foreign_keys=user_id)

    def __json__(self):
        return ['id', 'username', 'name', 'profile_picture', 'message', 'media', 'time_uploaded', 'lifetime', 'votes', 'area']


class Users(Base):
    __tablename__ = "users"

    id = db.Column('id', UUIDType(binary=False), primary_key=True)
    username = db.Column('username', db.Unicode, unique=True)
    email = db.Column('email', db.Unicode, unique=True)
    password = db.Column('password', db.Unicode)
    name = db.Column('name', db.Unicode)
    bio = db.Column('bio', db.Unicode)
    profile_picture = db.Column('profile_picture', db.Unicode) 

1 个答案:

答案 0 :(得分:1)

无论你传递给select方法,

sqlalchemy都会在查询的query部分放置,所以如果你想获得2个课程,你可以这样做:

db.session.query(Post, Users).join(Users).filter(Post.area == id, Users.id == Post.user_id).all()

这将返回Post和Users类的元组数组(因此您必须更改编码器以便能够接收元组)

编辑:

以下是如何更改编码器以接受元组的示例:

class AlchemyEncoder(json.JSONEncoder):
    def default(self, o):
        if isinstance(o, tuple):
            data = {}
            for obj in o:
                data.update(self.parse_sqlalchemy_object(obj))
            return data
        if isinstance(o.__class__, DeclarativeMeta):
            return self.parse_sqlalchemy_object(o)
        return json.JSONEncoder.default(self, o)

    def parse_sqlalchemy_object(self, o):
        data = {}
        fields = o.__json__() if hasattr(o, '__json__') else dir(o)
        for field in [f for f in fields if not f.startswith('_') and f not in ['metadata', 'query', 'query_class']]:
            value = o.__getattribute__(field)
            try:
                json.dumps(value)
                data[field] = value
            except TypeError:
                data[field] = None
        return data
相关问题