import sys import os import math n_nights = int(input('Type number of nights: ')) room_type = input('Type the type of room: ') price = 0.0 if room_type == 'apartment': price = n_nights * 70 else: price = n_nights * 125 #code doesn't account for writing mistakes if n_nights < 10: if room_type == 'apartment': price *= 0.7 else: price *= 0.9 elif n_nights > 15: if room_type == 'apartment': price *= 0.5 else: price *= 0.8 else: if room_type == 'apartment': price *= 0.65 else: price *= 0.85 print('Total price: {}'.format(round(price, 2)))