python - 使用 sqlalchemy 获取插入行的数量

Question

我正在编写代码来更新我的数据库，我想获取提交后插入的行数，我该怎么做？

for table, endpoint in ENDPOINTS.items():
    url = BASE_URL + endpoint
    response = get_response(url)

    table_object = getattr(Base.classes, table)

    self.add_missing_data(response, table_object, session)

session.commit()
print('Total rows added: {}'.format(xxxxx))

Answer 1

引用this Q/A :

这是你的代码:

from sqlalchemy import create_engine

def add_mising_data(self, response, table_object, session):
    con = create_engine(...)
    # insert some data using 
    # con, response, table_object and session
    result_proxy = con.execute("INSERT ...")
    return result_proxy.rowcount # this is the number of inserted rows

# in your loop
total_num_row = 0
for table, endpoint in ENDPOINTS.items():
    url = BASE_URL + endpoint
    response = get_response(url)

    table_object = getattr(Base.classes, table)

    num_row = self.add_missing_data(response, table_object, session)
    total_num_row = total_num_row + num_row

print (total_num_row) # <-- here you are