查找库存差额的起止日期

import datetime

# 模拟库存数据
inventory_data = [
    {"item": "A", "quantity": 100, "date": datetime.datetime(2023, 1, 1)},
    {"item": "A", "quantity": -50, "date": datetime.datetime(2023, 1, 5)},
    {"item": "A", "quantity": -30, "date": datetime.datetime(2023, 1, 10)},
    {"item": "A", "quantity": 20, "date": datetime.datetime(2023, 1, 15)},
]

# 计算库存差额的起止日期
def find_inventory_discrepancy_dates(data):
    discrepancy_start_date = None
    discrepancy_end_date = None
    current_quantity = 0

    for entry in data:
        current_quantity += entry["quantity"]
        if current_quantity != 0 and discrepancy_start_date is None:
            discrepancy_start_date = entry["date"]
        if current_quantity == 0 and discrepancy_end_date is None:
            discrepancy_end_date = entry["date"]

    return discrepancy_start_date, discrepancy_end_date

start_date, end_date = find_inventory_discrepancy_dates(inventory_data)
print(f"库存差额起止日期: {start_date} 到 {end_date}")