Package.py

# Imports
from datetime import datetime


# Object class for packages
class Package:
    def __init__(self, ID, address, city, state, zip_code, deadline, weight, status):
        self.ID = ID # Package ID
        # Package address entered
        self.address = address
        self.original_address = address
        self.city = city
        self.state = state
        self.zip_code = zip_code
        # Package details
        self.deadline = deadline
        self.weight = weight
        self.status = status
        self.departure_time = None # Time package left the hub
        self.arrival_time = None # Time package arrived to customer
        self.in_truck_id = None # ID of truck that package is in

    # Package default string
    def __str__(self):
        return "%s, %s, %s, %s, %s, %s, %s %s" % (self.ID, self.address, self.city, self.state, self.zip_code, self.weight, self.deadline,
                                               self.status)

    # Helper method to inform user on status of package
    def update_status(self, current_time: datetime):
        if self.arrival_time is not None and self.arrival_time <= current_time:
            self.status = "Delivered"
        elif self.departure_time is not None and self.departure_time <= current_time:
            self.status = "On The Way"
        else:
            self.status = "At Hub"