ride.py

from datetime import datetime
from vehicle import Car, Bike
class RideSharing:
    def __init__(self, company_name) -> None:
        self.company_name = company_name
        self.riders = []
        self.drivers = []
        self.rides = []
    def add_rider(self, rider):
        self.riders.append(rider)
    def add_driver(self, driver):
        self.drivers.append(driver)
    def __str__(self):
        return f"Company Name {self.company_name} with riders : {len(self.riders)} and Drivers : {len(self.drivers)}"

class Ride:
    def __init__(self, start_location, end_location, vehicle) -> None:
        self.start_location = start_location
        self.end_location = end_location
        self.driver = None
        self.rider = None
        self.start_time = None
        self.end_time = None
        self.estimated_fare = self.calculate_fare(vehicle.vehicle_type)
        self.vehicle = vehicle
    
    def set_driver(self, driver):
        self.driver = driver
    
    def start_ride(self):
        self.start_time = datetime.now()
    def end_ride(self):
        self.end_time = datetime.now()
        self.rider.wallet -= self.estimated_fare
        self.driver.wallet += self.estimated_fare
    
    # Total cost - Fare
    # 20 km gechi 
    # 1 ---> 30 taka
    # 20 ---> 30*20 = 600
    def calculate_fare(self, vehicle):
        print(vehicle)
        distance = 10
        fare_per_km = {
            'car' : 30,
            'bike' : 20,
            'cng' : 25
        }
        return distance*fare_per_km.get(vehicle)
    
    def __repr__(self):
        return f"Ride details. Started {self.start_location} to {self.end_location}"

# Dhaka --> Sylhet 
# Ride request e ki ki thake
# 1. Ke request korteche -- rider
# 2. se kothay jabe -- endlocation


class RideRequest:
    def __init__(self, rider, end_location) -> None:
        self.rider = rider
        self.end_location = end_location
        
class RideMatching:
    def __init__(self, drivers) -> None:
        self.available_drivers = drivers
    
    def find_driver(self, ride_request, vehicle_type):
        if len(self.available_drivers) > 0:
            print("Looking for drivers.....")
            driver = self.available_drivers[0]
            
            if vehicle_type == 'car':
                vechicle = Car('car', "ABC456", 30)
            elif vehicle_type == 'bike':
                vechicle = Bike("bike", "1234BH", 50)
            ride = Ride(ride_request.rider.current_location,
                        ride_request.end_location, vechicle)
            driver.accept_ride(ride)
            return ride