SQLALchemy:根据列键/名称获取行元素

问题描述 投票:0回答:1

我想做什么

使用SQLAlchemy(使用ORM)我正在尝试创建一个类似的函数:

export_database_table_to_excel_file( database_orm_object )

与database_orm_object一样

class database_orm_object(db.Model):
    id              = db.Column(db.Integer, primary_key=True)
    some_string     = db.Column(db.String(30), unique=True)

我们的想法是生成一个excel表,将列键作为标题行打印,然后为每个数据库条目输出一行。由于可以为该函数提供任何数据库表,因此列数和名称是未知的。

什么有效

到目前为止,我能够创建一个带有一行标题的表,对于上面的示例,输出将是一个包含两个单元格的excel文件:

| id | some_string |

这是通过获取所有database_orm_object列及其键来完成的。然后将密钥附加到内存中的(空)excel文件。

def create_download_file(database_orm_object):
    book = Workbook()
    sheet = book.active

    col_headers = database_orm_object.__table__.columns.keys()
    sheet.append(col_headers) # add column keys as header row
    book.save("generated.xlsx")

什么不起作用

我不知道如何访问数据库条目的元素。我的计划是创建两个for循环,一个遍历query.all()结果中的所有行的外部循环和一个遍历database_orm_object值的内部循环。内部循环将值附加到数组,外部循环然后将此数组添加为excel文件上的表行。

有些代码已经在提及的内容中提到了。添加了两个for循环,用于迭代查询结果(外部)和每个结果值(内部)。 col_id和query_result.col_id是不同的类型,用于说明我想要实现的目标:

def create_download_file(database_orm_object):
    book = Workbook()
    sheet = book.active

    col_headers = database_orm_object.__table__.columns.keys()
    sheet.append(col_headers) # add column keys as header row

    for item in database_orm_class.query.all():
        row = []                      # create an empty row
        for col_id in col_headers:    # ??? iterate over columns
            row.append(item.col_id)   # ??? add table row's value for this column
        sheet.append(row)             # append all the tables row values to excel file

    book.save("generated.xlsx")

如果我将列密钥存储在数组或列表中,我如何访问查询结果行的元素?

python mysql database orm sqlalchemy
1个回答
1
投票

你可以在新版本的sqlAlchemy中试试这个:

for item in database_orm_class.query.all():    
    item = dict(item)
    # your code

或者对于旧版本:

for item in database_orm_class.query.all():    
        item = item.__dict__
        # your code
© www.soinside.com 2019 - 2024. All rights reserved.