drivers.py

from reportlab.pdfgen import canvas
from openpyxl import Workbook

class Driver:
    drivers = {}
    id_data = {}

    def __init__(self, id, name, age, state, phn):
        self.id = id
        self.name = name
        self.age = age
        self.state = state
        self.phn = phn
        self.createProfile()

    def addDriver(self, id):
        return f"id{id}"

    def profile(self):
        print(f"ID: {self.id}")
        print(f"Name: {self.name}")
        print(f"Age: {self.age}")
        print(f"State: {self.state}")
        print(f"PHN: {self.phn}")

    def createProfile(self):
        key = self.addDriver(self.id)
        Driver.drivers[key] = {
            "name": self.name,
            "age": self.age,
            "state": self.state,
            "phn": self.phn
        }
        Driver.id_data[self.id] = {
            "id": self.id,
            "name": self.name,
            "age": self.age,
            "state": self.state,
            "phn": self.phn
        }

    @classmethod
    def searchById(cls, id):
        key = f"id{id}"
        if key in cls.drivers:
            data = cls.drivers[key]
            print(f"Driver Found:\nID: {id}\nName: {data['name']}\nAge: {data['age']}\nState: {data['state']}\nPHN: {data['phn']}")
        else:
            print(f"No driver found with ID: {id}")

    @classmethod
    def update_by_id(cls, id):
        key = f"id{id}"
        if key in cls.drivers:
            data = cls.drivers[key]
            print("Leave blank to keep current value.")
            name = input(f"NAME [{data['name']}]: ") or data['name']
            age = input(f"AGE [{data['age']}]: ") or data['age']
            state = input(f"STATE [{data['state']}]: ") or data['state']
            phn = input(f"PHN [{data['phn']}]: ") or data['phn']
            cls.drivers[key] = {
                "name": name,
                "age": age,
                "state": state,
                "phn": phn
            }
            cls.id_data[id] = {
                "id": id,
                "name": name,
                "age": age,
                "state": state,
                "phn": phn
            }
            print(f"Driver with ID {id} updated.")
        else:
            print(f"No driver found with ID: {id}")

    @classmethod
    def export_to_pdf(cls, filename="drivers.pdf"):
        pdf = canvas.Canvas(filename)
        pdf.setFont("Helvetica", 12)
        pdf.drawString(200, 800, "Registered Drivers")
        pdf.line(50, 790, 550, 790)
        y = 770
        if not cls.drivers:
            pdf.drawString(50, y, "No drivers registered.")
        else:
            for key, data in cls.drivers.items():
                pdf.drawString(50, y, f"ID: {key[2:]}, Name: {data['name']}, Age: {data['age']}, State: {data['state']}, PHN: {data['phn']}")
                y -= 20
        pdf.save()
        print(f"PDF exported as {filename}")

    @classmethod
    def printToPDF(cls, id):
        key = f"id{id}"
        if key in cls.drivers:
            data = cls.drivers[key]
            filename = f"Driver_{id}.pdf"
            c = canvas.Canvas(filename)
            c.setFont("Helvetica", 14)
            c.drawString(100, 750, f"Driver Profile - ID: {id}")
            c.drawString(100, 720, f"Name: {data['name']}")
            c.drawString(100, 700, f"Age: {data['age']}")
            c.drawString(100, 680, f"State: {data['state']}")
            c.drawString(100, 660, f"PHN: {data['phn']}")
            c.save()
            print(f"Driver profile saved as '{filename}'")
        else:
            print(f"No driver found with ID: {id}")

    @classmethod
    def export_to_excel(cls, filename="drivers.xlsx"):
        wb = Workbook()
        ws = wb.active
        ws.title = "Drivers"
        ws.append(["ID", "Name", "Age", "State", "PHN"])
        for key, data in cls.drivers.items():
            ws.append([
                key[2:],
                data["name"],
                data["age"],
                data["state"],
                data["phn"]
            ])
        wb.save(filename)
        print(f"Excel file exported as {filename}")

    @classmethod
    def delete_by_id(cls, id):
        key = f"id{id}"
        if key in cls.drivers:
            del cls.drivers[key]
            if id in cls.id_data:
                del cls.id_data[id]
            print(f"Driver with ID {id} deleted.")
        else:
            print(f"No driver found with ID: {id}")

class Vehicle:
    vehicles = {}
    reg_data = {}

    def __init__(self, reg_no, model, type_, capacity, assigned_driver):
        self.reg_no = reg_no
        self.model = model
        self.type_ = type_
        self.capacity = capacity
        self.assigned_driver = assigned_driver
        self.createProfile()

    def addVehicle(self, reg_no):
        return f"reg{reg_no}"

    def createProfile(self):
        key = self.addVehicle(self.reg_no)
        Vehicle.vehicles[key] = {
            "model": self.model,
            "type": self.type_,
            "capacity": self.capacity,
            "assigned_driver": self.assigned_driver
        }
        Vehicle.reg_data[self.reg_no] = {
            "model": self.model,
            "type": self.type_,
            "capacity": self.capacity,
            "assigned_driver": self.assigned_driver
        }

    @classmethod
    def searchByReg(cls, reg_no):
        key = f"reg{reg_no}"
        if key in cls.vehicles:
            data = cls.vehicles[key]
            assigned_driver_id = data['assigned_driver']
            driver_name = Driver.id_data.get(assigned_driver_id, {}).get("name", "Unknown")
            print(f"Vehicle Found:\nReg No: {reg_no}\nModel: {data['model']}\nType: {data['type']}\nCapacity: {data['capacity']}\nAssigned Driver: {driver_name} (ID: {assigned_driver_id})")
        else:
            print(f"No vehicle found with Reg No: {reg_no}")

    @classmethod
    def update_by_reg(cls, reg_no):
        key = f"reg{reg_no}"
        if key in cls.vehicles:
            data = cls.vehicles[key]
            print("Leave blank to keep current value.")
            model = input(f"MODEL [{data['model']}]: ") or data['model']
            type_ = input(f"TYPE [{data['type']}]: ") or data['type']
            capacity = input(f"CAPACITY [{data['capacity']}]: ") or data['capacity']
            assigned_driver = input(f"ASSIGNED DRIVER ID [{data['assigned_driver']}]: ") or data['assigned_driver']
            cls.vehicles[key] = {
                "model": model,
                "type": type_,
                "capacity": capacity,
                "assigned_driver": assigned_driver
            }
            cls.reg_data[reg_no] = {
                "model": model,
                "type": type_,
                "capacity": capacity,
                "assigned_driver": assigned_driver
            }
            print(f"Vehicle with Reg No {reg_no} updated.")
        else:
            print(f"No vehicle found with Reg No: {reg_no}")

    @classmethod
    def export_to_pdf(cls, filename="vehicles.pdf"):
        pdf = canvas.Canvas(filename)
        pdf.setFont("Helvetica", 12)
        pdf.drawString(200, 800, "Registered Vehicles")
        pdf.line(50, 790, 550, 790)
        y = 770
        if not cls.vehicles:
            pdf.drawString(50, y, "No vehicles registered.")
        else:
            for key, data in cls.vehicles.items():
                pdf.drawString(50, y, f"Reg No: {key[3:]}, Model: {data['model']}, Type: {data['type']}, Capacity: {data['capacity']}, Assigned Driver: {data['assigned_driver']}")
                y -= 20
        pdf.save()
        print(f"PDF exported as {filename}")

    @classmethod
    def export_to_excel(cls, filename="vehicles.xlsx"):
        wb = Workbook()
        ws = wb.active
        ws.title = "Vehicles"
        ws.append(["Reg No", "Model", "Type", "Capacity", "Assigned Driver"])
        for key, data in cls.vehicles.items():
            ws.append([
                key[3:],
                data["model"],
                data["type"],
                data["capacity"],
                data["assigned_driver"]
            ])
        wb.save(filename)
        print(f"Excel file exported as {filename}")

    @classmethod
    def delete_by_reg(cls, reg_no):
        key = f"reg{reg_no}"
        if key in cls.vehicles:
            del cls.vehicles[key]
            if reg_no in cls.reg_data:
                del cls.reg_data[reg_no]
            print(f"Vehicle with Reg No {reg_no} deleted.")
        else:
            print(f"No vehicle found with Reg No: {reg_no}")

if __name__ == "__main__":
    while True:
        print("\nOPTIONS: DRIVERS / VEHICLES / EXIT")
        ans = input("COMMAND: ").strip().upper()

        if ans == "DRIVERS":
            while True:
                print("\nDRIVER OPTIONS: ADD / SEARCH / UPDATE / DB / EXPORT_PDF / EXPORT_EXCEL / DELETE / BACK")
                d_ans = input("DRIVER COMMAND: ").strip().upper()
                if d_ans == "ADD":
                    id = input("ID: ")
                    name = input("NAME: ")
                    age = input("AGE: ")
                    state = input("STATE: ")
                    phn = input("PHN: ")
                    if f"id{id}" in Driver.drivers:
                        print("A driver with this ID already exists.")
                        continue
                    Driver(id, name, age, state, phn)
                    print("Driver added successfully.")
                elif d_ans == "SEARCH":
                    search_id = input("Enter the ID to search: ")
                    Driver.searchById(search_id)
                elif d_ans == "UPDATE":
                    update_id = input("Enter the ID to update: ")
                    Driver.update_by_id(update_id)
                elif d_ans == "DB":
                    if not Driver.drivers:
                        print("No drivers registered yet.")
                    else:
                        print("All Registered Drivers:")
                        for key, data in Driver.drivers.items():
                            print(f"ID: {key[2:]}, Name: {data['name']}, Age: {data['age']}, State: {data['state']}, PHN: {data['phn']}")
                elif d_ans == "EXPORT_PDF":
                    filename = input("Enter the filename to export (default 'drivers.pdf'): ")
                    if not filename:
                        filename = "drivers.pdf"
                    Driver.export_to_pdf(filename)
                elif d_ans == "EXPORT_EXCEL":
                    filename = input("Enter the filename to export (default 'drivers.xlsx'): ")
                    if not filename:
                        filename = "drivers.xlsx"
                    Driver.export_to_excel(filename)
                elif d_ans == "DELETE":
                    del_id = input("Enter the ID to delete: ")
                    Driver.delete_by_id(del_id)
                elif d_ans == "BACK":
                    break
                else:
                    print("Invalid DRIVER command.")

        elif ans == "VEHICLES":
            while True:
                print("\nVEHICLE OPTIONS: ADD / SEARCH / UPDATE / DB / EXPORT_PDF / EXPORT_EXCEL / DELETE / BACK")
                v_ans = input("VEHICLE COMMAND: ").strip().upper()
                if v_ans == "ADD":
                    reg_no = input("REG NO: ")
                    model = input("MODEL: ")
                    type_ = input("TYPE: ")
                    capacity = input("CAPACITY: ")
                    assigned_driver = input("ASSIGNED DRIVER ID: ")
                    if f"reg{reg_no}" in Vehicle.vehicles:
                        print("A vehicle with this Reg No already exists.")
                        continue
                    Vehicle(reg_no, model, type_, capacity, assigned_driver)
                    print("Vehicle added successfully.")
                elif v_ans == "SEARCH":
                    search_reg = input("Enter the Reg No to search: ")
                    Vehicle.searchByReg(search_reg)
                elif v_ans == "UPDATE":
                    update_reg = input("Enter the Reg No to update: ")
                    Vehicle.update_by_reg(update_reg)
                elif v_ans == "DB":
                    if not Vehicle.vehicles:
                        print("No vehicles registered yet.")
                    else:
                        print("All Registered Vehicles:")
                        for key, data in Vehicle.vehicles.items():
                            assigned_driver_id = data['assigned_driver']
                            driver_name = Driver.id_data.get(assigned_driver_id, {}).get("name", "Unknown")
                            print(f"Reg No: {key[3:]}, Model: {data['model']}, Type: {data['type']}, Capacity: {data['capacity']}, Assigned Driver: {driver_name} (ID: {assigned_driver_id})")
                elif v_ans == "EXPORT_PDF":
                    filename = input("Enter the filename to export (default 'vehicles.pdf'): ")
                    if not filename:
                        filename = "vehicles.pdf"
                    Vehicle.export_to_pdf(filename)
                elif v_ans == "EXPORT_EXCEL":
                    filename = input("Enter the filename to export (default 'vehicles.xlsx'): ")
                    if not filename:
                        filename = "vehicles.xlsx"
                    Vehicle.export_to_excel(filename)
                elif v_ans == "DELETE":
                    del_reg = input("Enter the Reg No to delete: ")
                    Vehicle.delete_by_reg(del_reg)
                elif v_ans == "BACK":
                    break
                else:
                    print("Invalid VEHICLE command.")

        elif ans == "EXIT":
            print("Exiting program. Goodbye!")
            break

        else:
            print("Invalid command. Please enter DRIVERS, VEHICLES, or EXIT.")