def update_package_status(self, package: Package, status: PackageStatus, time: datetime) -> None: if status == PackageStatus.ON_TRUCK: package = self.packages.search(package) package.status = status package.time_on_truck = time self.packages.update(package) if status == PackageStatus.DELIVERED: package = self.packages.search(package) package.status = status package.delivery_time = time self.packages.update(package)