awesome-low-level-design/solutions/python/concertticketbookingsystem/concertticketbookingsystemdemo.py at 5f8b5df1fbdadf9bb68d6081258689132e297915 · ashishps1/awesome-low-level-design · 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
75
76
77
78
79
80
81
from datetime import datetime, timedelta
from typing import List
from concertticketbookingsystem import ConcertTicketBookingSystem
from concert import Concert
from seat import Seat, SeatType, SeatStatus
from user import User

class ConcertTicketBookingSystemDemo:
    @staticmethod
    def run():
        # Create concert ticket booking system instance
        booking_system = ConcertTicketBookingSystem()

        # Create concerts
        concert1_seats = ConcertTicketBookingSystemDemo._generate_seats(10)
        concert1 = Concert("C001", "Artist 1", "Venue 1", datetime.now().replace(day=10, hour=20, minute=0, second=0, microsecond=0), concert1_seats)
        booking_system.addConcert(concert1)

        concert2_seats = ConcertTicketBookingSystemDemo._generate_seats(5)
        concert2 = Concert("C002", "Artist 2", "Venue 2", datetime.now().replace(day=15, hour=19, minute=30, second=0, microsecond=0), concert2_seats)
        booking_system.addConcert(concert2)

        # Create users
        user1 = User("U001", "John Doe", "[email protected]")
        user2 = User("U002", "Jane Smith", "[email protected]")

        # Search concerts
        search_results = booking_system.searchConcerts("Artist 1", "Venue 1", concert1.getDateTime())
        print("Search Results:")
        for concert in search_results:
            print(f"Concert: {concert.getArtist()} at {concert.getVenue()}")

        # Book tickets
        print("Booking 3 tickets for concert 1")
        selected_seats1 = ConcertTicketBookingSystemDemo._select_seats(concert1, 3)
        booking1 = booking_system.bookTickets(user1, concert1, selected_seats1)
        seats = concert1.getSeats()
        for seat in seats:
            print(seat)

        print("Booking 2 tickets for concert2")
        selected_seats2 = ConcertTicketBookingSystemDemo._select_seats(concert2, 2)
        booking2 = booking_system.bookTickets(user2, concert2, selected_seats2)
        seats = concert2.getSeats()
        for seat in seats:
            print(seat)

        # Cancel booking
        print("Cancelling Booking1 with 3 seats")
        booking_system.cancelBooking(booking1.id)
        print("Seats after cancelling booking 1 - all are available")
        seats = concert1.getSeats()
        for seat in seats:
            print(seat)

        # Book tickets again
        print("Booking 2 tickets for concert 1")
        selected_seats3 = ConcertTicketBookingSystemDemo._select_seats(concert1, 2)
        booking3 = booking_system.bookTickets(user2, concert1, selected_seats3)
        print("Seats after booking with booking 3")
        seats = concert1.getSeats()
        for seat in seats:
            print(seat)

    @staticmethod
    def _generate_seats(number_of_seats: int) -> List[Seat]:
        seats = []
        for i in range(1, number_of_seats + 1):
            seat_number = f"S{i}"
            seat_type = SeatType.VIP if i <= 1 else SeatType.PREMIUM if i <= 3 else SeatType.REGULAR
            price = 100.0 if seat_type == SeatType.VIP else 75.0 if seat_type == SeatType.PREMIUM else 50.0
            seats.append(Seat(seat_number, seat_number, seat_type, price))
        return seats

    @staticmethod
    def _select_seats(concert: Concert, number_of_seats: int) -> List[Seat]:
        available_seats = [seat for seat in concert.seats if seat.seatStatus == SeatStatus.AVAILABLE]
        return available_seats[:number_of_seats]

if __name__ == "__main__":
    ConcertTicketBookingSystemDemo.run()