def test_returnCar_for_invalid_rentalTime(self): # create a shop and a customer shop = CarRental(10) customer = Customer() # let the customer not rent a bike a try to return one. request = customer.returnCar() self.assertIsNone(shop.returnCar(request)) # manually check return function with error values self.assertIsNone(shop.returnCar((0, 0, 0)))
def test_returnCar_for_valid_credentials(self): # create a shop and a various customers shop = CarRental(50) customer1 = Customer() customer2 = Customer() customer3 = Customer() customer4 = Customer() customer5 = Customer() customer6 = Customer() # create valid rentalBasis for each customer customer1.rentalBasis = 1 # hourly customer2.rentalBasis = 1 # hourly customer3.rentalBasis = 2 # daily customer4.rentalBasis = 2 # daily customer5.rentalBasis = 3 # weekly customer6.rentalBasis = 3 # weekly # create valid bikes for each customer customer1.cars = 1 customer2.cars = 5 # eligible for family discount 30% customer3.cars = 2 customer4.cars = 8 customer5.cars = 15 customer6.cars = 30 # create past valid rental times for each customer customer1.rentalTime = datetime.now() + timedelta(hours=-4) customer2.rentalTime = datetime.now() + timedelta(hours=-23) customer3.rentalTime = datetime.now() + timedelta(days=-4) customer4.rentalTime = datetime.now() + timedelta(days=-13) customer5.rentalTime = datetime.now() + timedelta(weeks=-6) customer6.rentalTime = datetime.now() + timedelta(weeks=-12) # make all customers return their bikes request1 = customer1.returnCar() request2 = customer2.returnCar() request3 = customer3.returnCar() request4 = customer4.returnCar() request5 = customer5.returnCar() request6 = customer6.returnCar() # check if all of them get correct bill self.assertEqual(shop.returnCar(request1), 20) self.assertEqual(shop.returnCar(request2), 402.5) self.assertEqual(shop.returnCar(request3), 160) self.assertEqual(shop.returnCar(request4), 2080) self.assertEqual(shop.returnCar(request5), 5400) self.assertEqual(shop.returnCar(request6), 21600)
def test_returnCar_for_invalid_rentalBasis(self): # create a shop and a customer shop = CarRental(10) customer = Customer() # create valid rentalTime and cars customer.rentalTime = datetime.now() customer.cars = 3 # create invalid rentalbasis customer.rentalBasis = 8 request = customer.returnCar() self.assertEqual(shop.returnCar(request), 0)
def test_returnCar_for_invalid_numOfCars(self): # create a shop and a customer shop = CarRental(10) customer = Customer() # create valid rentalTime and rentalBasis customer.rentalTime = datetime.now() customer.rentalBasis = 1 # create invalid cars customer.cars = 0 request = customer.returnCar() self.assertIsNone(shop.returnCar(request))
def main(): shop = CarRental(100) customer = Customer() while True: print(""" ====== Car Rental Shop ======= 1. Display available cars 2. Request a car on hourly basis $5 3. Request a car on daily basis $20 4. Request a car on weekly basis $60 5. Return a car 6. Exit """) choice = input("Enter choice: ") try: choice = int(choice) except ValueError: print("That's not an int!") continue if choice == 1: shop.displaystock() elif choice == 2: customer.rentalTime = shop.rentCarOnHourlyBasis(customer.requestCar()) customer.rentalBasis = 1 elif choice == 3: customer.rentalTime = shop.rentCarOnDailyBasis(customer.requestCar()) customer.rentalBasis = 2 elif choice == 4: customer.rentalTime = shop.rentCarOnWeeklyBasis(customer.requestCar()) customer.rentalBasis = 3 elif choice == 5: customer.bill = shop.returnCar(customer.returnCar()) customer.rentalBasis, customer.rentalTime, customer.cars = 0,0,0 elif choice == 6: break else: print("Invalid input. Please enter number between 1-6 ") print("Thank you for using the car rental system.")
def test_rentCarOnWeeklyBasis_for_invalid_positive_number_of_cars(self): shop = CarRental(10) self.assertEqual(shop.rentOnWeeklyBasis(11), None)
def test_rentCarOnWeeklyBasis_for_valid_positive_number_of_cars(self): shop = CarRental(10) hour = datetime.now().hour self.assertEqual(shop.rentOnWeeklyBasis(2).hour, hour)
def test_rentCarOnWeeklyBasis_for_zero_number_of_cars(self): shop = CarRental(10) self.assertEqual(shop.rentOnWeeklyBasis(0), None)
def test_rentCarOnDailyBasis_for_negative_number_of_cars(self): shop = CarRental(10) self.assertEqual(shop.rentOnDailyBasis(-1), None)
def test_returnCar_for_valid_credentials(self): # create a shop and a various customers shop = CarRental(50) customer1 = Customer() customer2 = Customer() customer3 = Customer() customer4 = Customer() customer5 = Customer() customer6 = Customer() customer7 = Customer() customer8 = Customer() # create valid rentalBasis for each customer customer1.rentalBasis = 1 # hourly customer2.rentalBasis = 1 # hourly customer3.rentalBasis = 2 # daily customer4.rentalBasis = 2 # daily customer5.rentalBasis = 3 # weekly customer6.rentalBasis = 3 # weekly customer7.rentalBasis = 4 # monthly customer8.rentalBasis = 4 # monthly # create valid cars for each customer customer1.cars = 1 customer2.cars = 5 customer3.cars = 2 customer4.cars = 8 customer5.cars = 15 customer6.cars = 30 customer7.cars = 4 customer8.cars = 10 # create past valid rental times for each customer customer1.rentalTime = datetime.now() + timedelta(hours=-4) customer2.rentalTime = datetime.now() + timedelta(hours=-23) customer3.rentalTime = datetime.now() + timedelta(days=-4) customer4.rentalTime = datetime.now() + timedelta(days=-13) customer5.rentalTime = datetime.now() + timedelta(weeks=-6) customer6.rentalTime = datetime.now() + timedelta(weeks=-12) customer5.rentalTime = datetime.now() + timedelta(months=-2) customer5.rentalTime = datetime.now() + timedelta(months=-4) # make all customers return their cars request1 = customer1.returnCar() request2 = customer2.returnCar() request3 = customer3.returnCar() request4 = customer4.returnCar() request5 = customer5.returnCar() request6 = customer6.returnCar() request7 = customer7.returnCar() request8 = customer8.returnCar() # check if all of them get correct bill self.assertEqual(shop.returnCar(request1), 80) self.assertEqual(shop.returnCar(request2), 2300) self.assertEqual(shop.returnCar(request3), 1200) self.assertEqual(shop.returnCar(request4), 15600) self.assertEqual(shop.returnCar(request5), 58500) self.assertEqual(shop.returnCar(request6), 234000) self.assertEqual(shop.returnCar(request7), 58500) self.assertEqual(shop.returnCar(request8), 234000)
def __init__(self): self.__car_rental = CarRental()
class CarRentalApp(object): ## create car objects global car, engineCar, petrolCar, dieselCar, electricCar, hybridCar, carRental car = Car() engineCar = EngineCar() petrolCar = PetrolCar() dieselCar = DieselCar() electricCar = ElectricCar() hybridCar = HybridCar() carRental = CarRental() ## create customer object global customer customer = Customer() def fleetInfo(): ## construct car fleet carRental.setFleet(40) fleet = carRental.getFleet() availablePetrol = int(0.60 * fleet) availableDiesel = int(0.20 * fleet) availableElectric = int(0.10 * fleet) availableHybrid = int(0.10 * fleet) carRental.setAvailablePetrolCars([]) carRental.setAvailableDieselCars([]) carRental.setAvailableElectricCars([]) carRental.setAvailableHybridCars([]) for p in range(1, availablePetrol + 1): carRental.availablePetrolCars.append(petrolCar) for d in range(1, availableDiesel + 1): carRental.availableDieselCars.append(dieselCar) for e in range(1, availableElectric + 1): carRental.availableElectricCars.append(electricCar) for h in range(1, availableHybrid + 1): carRental.availableHybridCars.append(hybridCar) def customerDetails(): ## print quit message print('Type "q" to quit this program') ## obtain input form user while True: name = input('Please enter Customer name: ') if name == 'q' or name == 'Q': quit() ## check user did not just press return button ## and all characters are alphabetic ## note: unfortunately the isalpha method will not work in cases ## whereby some african names contain an exclamation mark if name.isalpha() == False: print('Sorry, name not recognised. Please try again.') continue customer.setName(name) licence = input('Please enter Customer driving licence number: ') if licence == 'q' or customer.licence == 'Q': quit() if licence == '': print('Sorry, licence not valid. Please try again.') continue customer.setLicence(licence) print('Type "H" if Customer wishes to hire a car from the fleet.') print('Type "R" if Customer is returning a car to the fleet.') global hireReturn hireReturn = input('Hire or Return?: ') if hireReturn == 'q': quit() hireReturn = hireReturn.lower() if (hireReturn != 'h') and (hireReturn != 'r'): print('Invalid input. Please try again.') continue break return hireReturn def carInfo(): ## obtain car choice while True: print('What type of car is being hired/returned?:') print('Type "P" for petrol') print('Type "D" for diesel') print('Type "E" for electric') print('Type "H" for hybrid') global carChoice carChoice = input('Enter car type now: ') carChoice = carChoice.lower() if carChoice == 'q': quit() carChoiceOptions = ['p', 'd', 'e', 'h'] ## check for valid carChoice if carChoice not in carChoiceOptions: print('Invalid car choice. Please try agsin.') continue ## check if car choice available if hireReturn == 'h': if ( (carChoice == 'p') and (len(carRental.getAvailablePetrolCars()) <= 0) ) \ or ( (carChoice == 'd') and (len(carRental.getAvailableDieselCars()) <= 0) ) \ or ( (carChoice == 'e') and (len(carRental.getAvailableElectricCars()) <= 0) ) \ or ( (carChoice == 'h') and (len(carRental.getAvailableHybridCars()) <= 0) ): print( 'Apologies, there are none of this car type currently in the fleet.' ) print('Please choose alternative car type.') continue break return carChoice def transmissionInfo(): ##obtain transmission info while True: if hireReturn == 'h': print( 'Does Customer prefer an Automatic or a Manual transmission?:' ) print('Type "A" for Automatic') print('Type "M" for Manual') transmissionChoice = input('Enter transmission preference: ') transmissionChoice = transmissionChoice.lower() if transmissionChoice == 'q': quit() transmissionChoiceOptions = ['a', 'm'] ## check for valid transmissionChoice if transmissionChoice not in transmissionChoiceOptions: print('Invalid transmission choice. Please try agsin.') continue if transmissionChoice == 'a': engineCar.setTransmission('Automatic') if transmissionChoice == 'm': engineCar.setTransmission('Manual') break def carMakeInfo(): ##obtain car make info while True: if hireReturn == 'h': print('Does Customer prefer a Ford or a Toyota?:') print('Type "F" for Ford') print('Type "T" for Toyota') makeChoice = input('Enter Make preference: ') makeChoice = makeChoice.lower() if makeChoice == 'q': quit() makeChoiceOptions = ['f', 't'] ## check for valid makeChoice if makeChoice not in makeChoiceOptions: print('Invalid Make choice. Please try agsin.') continue if makeChoice == 'f': car.setMake('Ford') if makeChoice == 't': car.setMake('Toyota') break def carModelInfo(): ##obtain car model info while True: if car.getMake() == 'Ford': print('Does Customer prefer a Focus or a Mondeo?:') print('Type "F" for Focus') print('Type "M" for Mondeo') modelChoice = input('Enter Model preference: ') modelChoice = modelChoice.lower() if modelChoice == 'q': quit() modelChoiceOptions = ['f', 'm'] ## check for valid modelChoice if modelChoice not in modelChoiceOptions: print('Invalid Model choice. Please try agsin.') continue if modelChoice == 'f': car.setModel('Focus') if modelChoice == 'm': car.setModel('Mondeo') break while True: if car.getMake() == 'Toyota': print('Does Customer prefer a Corolla or an Avensis?:') print('Type "C" for Corolla') print('Type "A" for Avensis') modelChoice = input('Enter Model preference: ') modelChoice = modelChoice.lower() if modelChoice == 'q': quit() modelChoiceOptions = ['c', 'a'] ## check for valid modelChoice if modelChoice not in modelChoiceOptions: print('Invalid make choice. Please try agsin.') continue if modelChoice == 'c': car.setModel('Corolla') if modelChoice == 'a': car.setModel('Avensis') break def petrolCarInfo(): ##obtain fuel info while True: if carChoice == 'p': print('Does Customer have a fuel preference?:') print('Type "L" for leaded') print('Type "U" for unleaded') fuelChoice = input('Enter Customer choice now: ') fuelChoice = fuelChoice.lower() if fuelChoice == 'q': quit() fuelChoiceOptions = ['l', 'u'] ## check for valid fuelChoice if fuelChoice not in fuelChoiceOptions: print('Invalid fuel choice. Please try agsin.') continue if fuelChoice == 'l': petrolCar.setPetrolType('Leaded') if fuelChoice == 'u': petrolCar.setPetrolType('Unleaded') break def dieselCarInfo(): ##obtain turbo info while True: if carChoice == 'd': print('Does Customer prefer standard or turbo?:') print('Type "S" for standard') print('Type "T" for turbo') turboChoice = input('Enter Customer choice now: ') turboChoice = turboChoice.lower() if turboChoice == 'q': quit() turboChoiceOptions = ['s', 't'] ## check for valid turboChoice if turboChoice not in turboChoiceOptions: print('Invalid turbo choice. Please try agsin.') continue if turboChoice == 's': dieselCar.setDieselTurbo(False) if turboChoice == 't': dieselCar.setDieselTurbo(True) break def electricCarInfo(): ##obtain fuelCells info while True: if carChoice == 'e': print('Does Customer prefer battery or capacitor?:') print('Type "B" for battery') print('Type "C" for capacitor') fuelCellsChoice = input('Enter Customer choice now: ') fuelCellsChoice = fuelCellsChoice.lower() if fuelCellsChoice == 'q': quit() fuelCellsChoiceOptions = ['b', 'c'] ## check for valid fuelCellsChoice if fuelCellsChoice not in fuelCellsChoiceOptions: print('Invalid fuel cell choice. Please try agsin.') continue if fuelCellsChoice == 'b': electricCar.setFuelCells('Battery') if fuelCellsChoice == 'c': electricCar.setFuelCells('Capacitor') break def hybridCarInfo(): ##obtain brake info while True: if carChoice == 'h': print( 'Does Customer prefer standard or regenerative braking system?:' ) print('Type "S" for standard brake system') print('Type "R" for regenerative brake system') brakeChoice = input('Enter Customer choice now: ') brakeChoice = brakeChoice.lower() if brakeChoice == 'q': quit() brakeChoiceOptions = ['s', 'r'] ## check for valid brakeChoice if brakeChoice not in brakeChoiceOptions: print('Invalid brake choice. Please try agsin.') continue if brakeChoice == 's': hybridCar.setHybridBrake(False) if brakeChoice == 'r': hybridCar.setHybridBrake(True) break def updateFleet(carChoice, hireReturn): ## remove/add one (1) car to/from fleet fleet = carRental.getFleet() if carChoice == 'p': if (hireReturn == 'h'): carRental.availablePetrolCars.pop() fleet -= 1 elif (hireReturn == 'r'): carRental.availablePetrolCars.append(petrolCar) fleet += 1 else: print('Hell has just frozen over. Petrol debug necessary!') quit() elif carChoice == 'd': if (hireReturn == 'h'): carRental.availableDieselCars.pop() fleet -= 1 elif (hireReturn == 'r'): carRental.availableDieselCars.append(dieselCar) fleet += 1 else: print('Hell has just frozen over. Diesel debug necessary!') quit() elif carChoice == 'e': if (hireReturn == 'h'): carRental.availableElectricCars.pop() fleet -= 1 elif (hireReturn == 'r'): carRental.availableElectricCars.append(electricCar) fleet += 1 else: print('Hell has just frozen over. Electric debug necessary!') quit() elif carChoice == 'h': if (hireReturn == 'h'): carRental.availableHybridCars.pop() fleet -= 1 elif (hireReturn == 'r'): carRental.availableHybridCars.append(hybridCar) fleet += 1 else: print('Hell has just frozen over. Hybrid debug necessary!') quit() else: print('Woah ... back up the apple-cart. Debug required now!') quit() def calculateHirePeriod(hireReturn): while True: print('How many days does Customer require?') print('Number must be between 1 and 28 (inclusive).') hirePeriod = input('Please enter number of days: ') if hirePeriod == 'q': quit() try: hirePeriod = int(hirePeriod) if (1 <= hirePeriod) and (hirePeriod <= 28): break else: print('Hire period must be between 1 and 28.') except: print("Invalid input. Try again.") return hirePeriod def calculateHireCharge(hireReturn, hirePeriod): ## hire rates for large cars (i.e. Avensis & Mondeo): ## are 50 euro per day for the first 10 days ## and 35 euro per day thereafter ## hire rates for small cars (i.e. Corolla & Focus): ## are 40 euro per day for the first 10 days ## and 25 euro per day thereafter if hirePeriod <= 10: if (car.getModel() == 'Avensis') or (car.getModel() == 'Mondeo'): hireCharge = hirePeriod * 50.00 if (car.getModel() == 'Corolla') or (car.getModel() == 'Focus'): hireCharge = hirePeriod * 40.00 elif (10 < hirePeriod) and (hirePeriod <= 28): if (car.getModel() == 'Avensis') or (car.getModel() == 'Mondeo'): hireCharge = 500 + ((hirePeriod - 10) * 35.00) if (car.getModel() == 'Corolla') or (car.getModel() == 'Focus'): hireCharge = 400 + ((hirePeriod - 10) * 25.00) else: print("Oops! ... we shouldn't be here!") quit() return hireCharge fleetInfo() hireReturn = customerDetails() carInfo() transmissionInfo() carMakeInfo() carModelInfo() petrolCarInfo() dieselCarInfo() electricCarInfo() hybridCarInfo() availableCars = updateFleet(carChoice, hireReturn) if hireReturn == 'h': customer.hirePeriod = calculateHirePeriod(hireReturn) customer.hireCharge = calculateHireCharge(hireReturn, customer.hirePeriod) print('Customer name:\t\t\t', customer.getName()) print('Customer licence:\t\t', customer.getLicence()) if hireReturn == 'r': print('Thanks for returning the car.') print('Hope to see you again soom.') if hireReturn == 'h': print('Make of car hired:\t\t', car.getMake()) print('Model of car hired:\t\t', car.getModel()) print('Transmission of car hired:\t', engineCar.getTransmission()) if carChoice == 'p': print('Petrol type:\t\t\t', petrolCar.getPetrolType()) if carChoice == 'd': print('Has turbo?:\t\t\t', dieselCar.getDieselTurbo()) if carChoice == 'e': print('Fuel cell type:\t\t\t', electricCar.getFuelCells()) if carChoice == 'h': print('Regenerative braking system:\t', hybridCar.getHybridBrake()) print('Hire period (days):\t\t', customer.hirePeriod) print('Hire charge (euro):\t\t', customer.hireCharge)
def test_car_rental(self): # Load car_rental = CarRental() car_rental.load_current_stock('test_CarRentalStock.csv') self.assertEqual(len(car_rental.get_diesel_cars()), 10) self.assertEqual(len(car_rental.get_electric_cars()), 6) self.assertEqual(len(car_rental.get_hybrid_cars()), 4) self.assertEqual(len(car_rental.get_petrol_cars()), 20) # Renting car_rental.rent_cars('p', 3) rented_cars = len([ 1 for car in car_rental.get_petrol_cars() if car.get_is_rented() == True ]) self.assertEqual( rented_cars, 3, 'The petrol cars list has not been updated correctly with the # of rented cars.' ) # Returns validation self.assertTrue(car_rental.is_rented_car_registration( 'p', '01 P 1002')) self.assertFalse( car_rental.is_rented_car_registration('p', '06 G 1002')) # Returns car_rental.return_validated_cars('p', ['01 P 1002', '01 P 1003']) rented_cars = len([ 1 for car in car_rental.get_petrol_cars() if car.get_is_rented() == True ]) self.assertEqual( rented_cars, 1, 'The returned cars have not been updated correctly in the petrol cars list.' )
def setUp(self): self.carRental = CarRental()
class TestCarRental(unittest.TestCase): def setUp(self): self.carRental = CarRental() def test_default_constructor(self): self.assertEquals(0, self.carRental.fleet) self.assertEquals([], self.carRental.availablePetrolCars) self.assertEquals([], self.carRental.availableDieselCars) self.assertEquals([], self.carRental.availableElectricCars) self.assertEquals([], self.carRental.availableHybridCars) def test_car_rental_get_methods(self): self.assertEquals(0, self.carRental.getFleet()) self.assertEquals([], self.carRental.getAvailablePetrolCars()) self.assertEquals([], self.carRental.getAvailableDieselCars()) self.assertEquals([], self.carRental.getAvailableElectricCars()) self.assertEquals([], self.carRental.getAvailableHybridCars()) def test_car_rental_set_methods(self): self.carRental.setFleet(40) self.assertEquals(40, self.carRental.getFleet()) petrolCar = PetrolCar() dieselCar = DieselCar() electricCar = ElectricCar() hybridCar = HybridCar() self.carRental.setAvailablePetrolCars([petrolCar, petrolCar]) self.assertEquals([petrolCar, petrolCar], self.carRental.getAvailablePetrolCars()) self.carRental.setAvailableDieselCars([dieselCar, dieselCar, dieselCar]) self.assertEquals([dieselCar, dieselCar, dieselCar], self.carRental.getAvailableDieselCars()) self.carRental.setAvailableElectricCars([electricCar, electricCar]) self.assertEquals([electricCar, electricCar], self.carRental.getAvailableElectricCars()) self.carRental.setAvailableHybridCars([hybridCar, hybridCar]) self.assertEquals([hybridCar, hybridCar], self.carRental.getAvailableHybridCars())
def test_Car_Rental_displays_correct_stock(self): shop1 = CarRental() shop2 = CarRental(10) self.assertEqual(shop1.displaystock(), 0) self.assertEqual(shop2.displaystock(), 10)
class TestCarRental(unittest.TestCase): def setUp(self): self.carRental = CarRental() def test_default_constructor(self): self.assertEquals(0, self.carRental.fleet) self.assertEquals([], self.carRental.availablePetrolCars) self.assertEquals([], self.carRental.availableDieselCars) self.assertEquals([], self.carRental.availableElectricCars) self.assertEquals([], self.carRental.availableHybridCars) def test_car_rental_get_methods(self): self.assertEquals(0, self.carRental.getFleet()) self.assertEquals([], self.carRental.getAvailablePetrolCars()) self.assertEquals([], self.carRental.getAvailableDieselCars()) self.assertEquals([], self.carRental.getAvailableElectricCars()) self.assertEquals([], self.carRental.getAvailableHybridCars()) def test_car_rental_set_methods(self): self.carRental.setFleet(40) self.assertEquals(40, self.carRental.getFleet()) petrolCar = PetrolCar() dieselCar = DieselCar() electricCar = ElectricCar() hybridCar = HybridCar() self.carRental.setAvailablePetrolCars([petrolCar, petrolCar]) self.assertEquals([petrolCar, petrolCar], self.carRental.getAvailablePetrolCars()) self.carRental.setAvailableDieselCars( [dieselCar, dieselCar, dieselCar]) self.assertEquals([dieselCar, dieselCar, dieselCar], self.carRental.getAvailableDieselCars()) self.carRental.setAvailableElectricCars([electricCar, electricCar]) self.assertEquals([electricCar, electricCar], self.carRental.getAvailableElectricCars()) self.carRental.setAvailableHybridCars([hybridCar, hybridCar]) self.assertEquals([hybridCar, hybridCar], self.carRental.getAvailableHybridCars())
def setUp(self): self.carRental = CarRental()
class CarRentalApp(): def __init__(self): self.__car_rental = CarRental() def __show_banner(self): # Display the application banner. print('#' * 30) print('Welcome to DBS Car Rental') print('#' * 30, '\n') def __show_main_options(self): # Show the first level user options print('Select from the following options ...') option_format = '\t{}\t{}' print(option_format.format(1, 'Rent')) print(option_format.format(2, 'Return')) print(option_format.format('q', 'Quit')) def __show_car_type_options(self): # Show the options of car types for selection. print('Select from the following options ...') option_format = '\t{}\t{}' print(option_format.format('d', 'Diesel')) print(option_format.format('e', 'Electric')) print(option_format.format('h', 'Hybrid')) print(option_format.format('p', 'Petrol')) print(option_format.format('q', 'Quit')) def __get_user_options(self, valid_options): # Validate user input is from a allowed set of characters. input_invalid = True valid_options = list(valid_options) while input_invalid: user_input = input('Enter your option: ').lower() if user_input in valid_options: input_invalid = False else: print("'{}' is not a valid option.".format(user_input)) return user_input def __get_cars_amount(self, car_type): # Get from user how many cars are rented. rentable_amount = self.__car_rental.get_rentable_amount_by_type( car_type) print('The # of cars available for rent - {}'.format(rentable_amount)) input_invalid = True while input_invalid: amount = input( 'Please enter a number for the amount or 0 to quit: ') if amount.isdigit(): amount = int(amount) if rentable_amount < amount: print('There are not enough cars to rent.') else: input_invalid = False else: print("'{}' is not a valid amount.".format(amount)) return amount def __show_registrations(self, registrations): # Show the list of rented/returned cars. for reg in registrations: print('\t{}'.format(reg)) def __do_renting(self): # Manages the rental process. print('Which type of cars do you want to rent?') self.__show_car_type_options() car_type = self.__get_user_options('dehpq') if car_type == 'q': return else: amount = self.__get_cars_amount(car_type) if amount == 0: return else: registrations = self.__car_rental.rent_cars(car_type, amount) print('The following cars have been rented.') self.__show_registrations(registrations) def __get_return_registrations(self, car_type): # Gets the registrations of returned cars from the user. # Registrations are checked for validity on input. print('Please enter the returned car registrations.') print('\tEnter d when Done or q to Quit.') registrations = [] get_registrations = True while get_registrations == True: registration = input('Car registration: ') if registration.lower() == 'q': registrations.clear() get_registrations = False elif registration.lower() == 'd': get_registrations = False elif self.__car_rental.is_rented_car_registration( car_type, registration): registrations.append(registration) else: print("'{}' is not a valid car registration.".format( registration)) return registrations def __do_returns(self): # Manages the returns process. print('Which type of cars do you want to return?') self.__show_car_type_options() car_type = self.__get_user_options('dehpq') if car_type == 'q': return else: registrations = self.__get_return_registrations(car_type) if len(registrations) > 0: self.__car_rental.return_validated_cars( car_type, registrations) print('The following cars have been returned.') self.__show_registrations(registrations) def __run_user_interface(self): # Run the inteeractive application. self.__show_banner() keep_running = True while keep_running: self.__show_main_options() option = self.__get_user_options('12q') if option == '1': self.__do_renting() elif option == '2': self.__do_returns() else: keep_running = False print('Sorry to see you leave. Goodbye.') def run_application(self): # Loads the CarRental class and on completion of application run # saves current status of stock back to the CSV file. csv_file_name = 'CarRentalStock.csv' self.__car_rental.load_current_stock(csv_file_name) self.__run_user_interface() self.__car_rental.save_current_stock(csv_file_name)