def flight_booking(flightid=None): try: if(flightid==None): print("Enter Flight Id:") flightid=input() print("Enter date of travel:") date_of_travel=input() print("Enter number of children:") no_of_children=input() print("Enter number of adults") no_of_adults=input() print("Enter name of primary passenger") primary_passenger=input() fare=0 fare=flight_booking_validation.validate_flights(flightid,date_of_travel,no_of_children,no_of_adults,primary_passenger) print("Total fare:",fare) count=book_flightDB.retrieve_count(); bookingid="F"+str(count+1) booking=FlightBooking() booking.set_bookingid(bookingid) booking.set_flight_id(flightid) booking.set_date_of_travel(date_of_travel) booking.set_no_of_children(no_of_children) booking.set_no_of_adults(no_of_adults) booking.set_primary_passenger(primary_passenger) booking.set_fare(fare) book_flightDB.bookFlight(booking) print("Your ticket is successfully booked with booking id",bookingid) print("Do you wish to search a hotel? Enter 'Y'or 'N'") ch=input() if(ch=='Y'): search_hotel.search_hotel(bookingid); except CustomExceptions.InvalidDateException as e: print(e) except CustomExceptions.InvalidFlightIdException as e: print(e) except CustomExceptions.InvalidChildrenException as e: print(e) except CustomExceptions.InvalidAdultsException as e: print(e) except CustomExceptions.InvalidPrimaryPassenger as e: print(e) except Exception as e: print(e) finally: if(fare==0): flight_booking()
''' Created on Aug 13, 2015 @author: sadhna01 ''' from validations import flight_booking_validation from exceptions import CustomExceptions ''' positive test cases ''' fare=flight_booking_validation.validate_flights('SP101','01-01-16',2,3,"GOPAL") print(fare) fare=flight_booking_validation.validate_flights('AI101','28-02-16',1,4,"GOPAL") print(fare) ''' negetive test cases ''' try: fare=flight_booking_validation.validate_flights('SP10','01-01-16',2,3,"GOPAL") except CustomExceptions.InvalidFlightIdException as e: print(e) try: fare=flight_booking_validation.validate_flights('SP101','01-01-16',0,0,"GOPAL") except CustomExceptions.InvalidAdultsException as e: print(e) try: fare=flight_booking_validation.validate_flights('SP101','01-01-16',5,4,"GOPAL") except CustomExceptions.InvalidChildrenException as e: print(e) try: