import sys

#Find out if customer is renting budget, daily, or weekly and prompt user to input their response.
rentalCode = input("(B)udget, (D)aily, or (W)eekly rental?\n")
#Use if and elif statement to determine whether to print days or weeks rented and prompt user to input their response.
if rentalCode == 'B' or rentalCode == 'D':
  rentalPeriod = int(input("Number of Days Rented:\n"))
else:
  rentalPeriod = int(input("Number of Weeks Rented:\n"))
'''
I thought an elif statement would be more appropriate
but I couldn't get it to work in codio, so I'm
putting it in the comments. I think using the else
statement would allow the customer to input
anything else and python would interpret it as a W

elif rentalCode == 'W':
  rentalPeriod = int(input("Number of Weeks Rented:\n))
  '''
##Prompt user to input their starting and ending mileage.
odoStart = int(input("Starting Odometer Reading:\n"))
odoEnd = int(input("Ending Odometer Reading:\n"))
#Calculate user's mileage
totalMiles = (odoEnd - odoStart)
#Output user retal summary
print('Rental Summary')
print('Rental Code:\t', rentalCode)
print('Rental Period:\t', rentalPeriod)
print('Starting Odometer:\t', odoStart)
print('Ending Odometer:\t', odoEnd)
print('Miles Driven:\t', totalMiles)

#These are the base charges for rental code
budgetCharge = 40.00
dailyCharge = 60.00
weeklyCharge = 190.00
#Use conditional statement to calculate base charges
if rentalCode == 'B':
  baseCharge = rentalPeriod * budgetCharge
elif rentalCode == 'D':
  baseCharge = rentalPeriod * dailyCharge
elif rentalCode == 'W':
  baseCharge = rentalPeriod * weeklyCharge
'''
Use condition statements to calculate the mileage
charge based on how many miles they drove for
rental code.
'''
if rentalCode == 'B':
  mileCharge = totalMiles * 0.25
elif rentalCode == 'D':
  averageDayMiles = totalMiles/rentalPeriod
  if averageDayMiles <= 100:
    extraMiles = 0
  elif averageDayMiles > 100:
    extraMiles = averageDayMiles - 100
    mileCharge = 0.25 * extraMiles
elif rentalCode == 'W':
  averageWeekMiles = totalMiles/rentalPeriod
  if averageWeekMiles <= 900:
    mileCharge = 0
  elif averageWeekMiles > 900:
    mileCharge = 100.00 * rentalPeriod
#Calculate the amount due.
amtDue = baseCharge + mileCharge
#Display this info to user.
print('Amount Due:\t', '$', '%.2f'%amtDue)