uber-oop/uber.py at main · nothando-debug/uber-oop · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
from driver import Driver
from rider import Rider
from ride import Ride
import random


class Uber:
    def __init__(self):
        self.available_drivers = []
        self.active_rides = []
        self.completed_rides = []


    def add_driver(self, driver):
        self.available_drivers.append(driver)


    def request_ride(self, rider, distance):
        random_driver = self.available_drivers.pop()
        ride1 = Ride(random_driver,rider, distance)
        self.active_rides.append(ride1)
        random_driver.available = False
        return ride1

    def mark_ride_completed(self, ride):
        ride.driver.available = True
        self.completed_rides.append(ride)
        self.active_rides.remove(ride)
        return ride


    def __str__(self):
        result = ""
        for i in self.available_drivers:
            result += str(i) + "\n"
        for k in self.active_rides:
            result+= str(k) + "\n"
        for j in self.completed_rides:
            result += str(j) + "\n"
        return result


# d1 = Driver("Drake","Bajaj")
# d2 = Driver("Beyonce", "Fiat")
# d3 = Driver("Nelly", "Bus")
# r1 = Rider("Father Elvis")
# r2 = Rider("Gerald")

# # ride1 = Ride("Pharsa", "Hanabi", 8 )
# # ride2 = Ride("No", "jay", 9)
# app = Uber()
# app.add_driver(d1)
# app.add_driver(d2)
# app.add_driver(d3)


# ride1 = app.request_ride(r1,6)

# ride1.start_ride()


# ride1.complete_ride()

# app.mark_ride_completed(ride1)

# print(ride1.driver.available == True)

# print(ride1 in app.completed_rides)
# print(ride1 not in app.active_rides)