Python 数据库操作：使用 try-except 和 rollback 保证事务完整性

在 Python 中使用数据库进行操作时，为了保证数据的一致性和完整性，通常需要使用事务。事务可以保证一组操作要么全部成功，要么全部失败，从而避免数据出现不一致的情况。

在进行数据库操作时，如果出现异常，可能会导致部分操作成功，部分操作失败，从而导致数据不一致。为了避免这种情况，可以使用 try-except 语句块和 rollback 函数来回滚事务。

以下是一个示例代码，展示了如何使用 try-except 语句块和 rollback 函数来保证事务的完整性。

try:
    file = AnalysisFile(
        filename=filename,
        user_id=g.user.user_id,
        date_guance=ceduan_mes[0]['日期'],
    )
    db.session.add(file)
    db.session.flush()
    for index1, i in enumerate(ceduan_mes):
        ceduan = AnalysisFileCeduan(
            file_id=file.id,
            ceduan_num=index1+1,
            ceduan_start=i['起始点'],
            ceduan_end=i['结束点'],
            date=i['日期'],
            guance_mode=i['观测模式'],
            temp=i['温度'],
            ijiao=i['I角'],
            gaocheng_start=i['起始高程'],
            gaocheng_end=i['结束高程'],
            gaocha=i['累计高差'],
            juli=i['视距累计'],
            time_start=i['起始时间'],
            time_end=i['结束时间'],
            juli_b=i['后视距累计'],
            juli_f=i['前视距累计'],
        )
        db.session.add(ceduan)
        db.session.flush()
        for index2, j in enumerate(i['测段信息']):
            if '测站信息' in j.keys():
                cezhan = AnalysisFileCezhan(
                    ceduan_id=ceduan.id,
                    cezhan_num=index2+1,
                    cezhan_dianhao_b1=j['测站信息']['B1'][0],
                    cezhan_dianhao_b2=j['测站信息']['B2'][0],
                    cezhan_dianhao_f1=j['测站信息']['F1'][0],
                    cezhan_dianhao_f2=j['测站信息']['F2'][0],
                    cezhan_shiju_b1=j['测站信息']['B1'][1],
                    cezhan_shiju_b2=j['测站信息']['B2'][1],
                    cezhan_shiju_f1=j['测站信息']['F1'][1],
                    cezhan_shiju_f2=j['测站信息']['F2'][1],
                    cezhan_dushu_b1=j['测站信息']['B1'][2],
                    cezhan_dushu_b2=j['测站信息']['B2'][2],
                    cezhan_dushu_f1=j['测站信息']['F1'][2],
                    cezhan_dushu_f2=j['测站信息']['F2'][2],
                    gaocha=j['高差'],
                    juli=j['距离累计'],
                    gaocheng=j['测站高程'],
                    time_end=j['时间'],  # 结束时间
                )
                db.session.add(cezhan)
                db.session.flush()
                for index3, k in enumerate(j['测点信息']):
                    point = AnalysisFilePoint(
                        cezhan_id=cezhan.id,
                        dianhao=k[0],
                        dushu=k[1],
                        shiju=k[2],
                        gaocheng=k[3],
                    )
                    db.session.add(point)
    db.session.commit()
except:
    db.session.rollback()
    raise

在上面的代码中，所有数据库操作都放在了 try 块中。如果在执行过程中出现异常，则会执行 except 块中的代码。在 except 块中，调用 db.session.rollback() 函数来回滚事务，并将异常重新抛出。

这样，即使在数据库操作过程中出现异常，也能保证事务的完整性，避免数据不一致。

注意：

• 为了保证事务的完整性，在 except 块中应该重新抛出异常，以便程序可以正常处理异常。
• 在使用 db.session.rollback() 函数回滚事务后，应该重新引发异常，以确保程序可以正常处理异常。
• 在实际应用中，可以根据需要在 except 块中添加其他处理代码，例如记录日志或发送通知等。

使用 try-except 语句块和 rollback 函数，可以有效地保证数据库操作的完整性和一致性，提高程序的健壮性和可靠性。