我想做什么
使用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")
题
如果我将列密钥存储在数组或列表中,我如何访问查询结果行的元素?
你可以在新版本的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