You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
209 lines
5.9 KiB
209 lines
5.9 KiB
# python 3 oop - chapter 3 case study
|
|
# simple real estate application
|
|
|
|
|
|
class Property:
|
|
|
|
def __init__(self, square_feet='', beds='', baths='', **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.square_feet = square_feet
|
|
self.num_bedrooms = beds
|
|
self.num_baths = baths
|
|
|
|
def display(self):
|
|
print("PROPERTY DETAILS")
|
|
print("================")
|
|
print("square footage: {}".format(self.square_feet))
|
|
print("bedrooms: {}".format(self.num_bedrooms))
|
|
print("bathrooms: {}".format(self.num_baths))
|
|
print()
|
|
|
|
@staticmethod # decorator form of the staticmethod constructor function
|
|
def prompt_init():
|
|
return dict(square_feet=input("Enter the square feet: "),
|
|
beds=input("Enter number of bedrooms: "),
|
|
baths=input("Enter number of baths: "))
|
|
|
|
|
|
class Apartment(Property):
|
|
|
|
# class variables
|
|
valid_laundries = ("coin", "ensuite", "none")
|
|
valid_balconies = ("yes", "no", "solarium")
|
|
|
|
def __init__(self, balcony="", laundry="", **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.balcony = balcony
|
|
self.laundry = laundry
|
|
|
|
def display(self):
|
|
super().display()
|
|
print("APARTMENT DETAILS")
|
|
print("laundry: %s" % self.laundry)
|
|
print("has balcony: %s" % self.balcony)
|
|
print()
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
parent_init = Property.prompt_init()
|
|
laundry = get_valid_input("What laundry facilities does the property "
|
|
"have?", Apartment.valid_laundries)
|
|
balcony = get_valid_input("Does the property have a balcony?",
|
|
Apartment.valid_balconies)
|
|
parent_init.update({
|
|
"laundry": laundry,
|
|
"balcony": balcony
|
|
})
|
|
return parent_init
|
|
|
|
|
|
class House(Property):
|
|
|
|
# class variables
|
|
valid_garage = ("attached", "detached", "none")
|
|
valid_fenced = ("yes", "no")
|
|
|
|
def __init__(self, num_stories="", garage="", fenced="", **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.garage = garage
|
|
self.fenced = fenced
|
|
self.num_stories = num_stories
|
|
|
|
def display(self):
|
|
super().display()
|
|
print("HOUSE DETAILS")
|
|
print("# of stories: {}".format(self.num_stories))
|
|
print("garage: {}".format(self.garage))
|
|
print("fenced yard: {}".format(self.fenced))
|
|
print()
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
parent_init = Property.prompt_init()
|
|
fenced = get_valid_input("Is the yard fenced?", House.valid_fenced)
|
|
garage = get_valid_input("Is there a garage?", House.valid_garage)
|
|
num_stories = input("How many stories? ")
|
|
parent_init.update({
|
|
"fenced": fenced,
|
|
"garage": garage,
|
|
"num_stories": num_stories
|
|
})
|
|
return parent_init
|
|
|
|
|
|
class Purchase:
|
|
|
|
def __init__(self, price="", taxes="", **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.price = price
|
|
self.taxes = taxes
|
|
|
|
def display(self):
|
|
super().display()
|
|
print("PURCHASE DETAILS")
|
|
print("selling price: {}".format(self.price))
|
|
print("estimated taxes: {}".format(self.taxes))
|
|
print()
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
return dict(
|
|
price=input("What's the selling price? "),
|
|
taxes=input("What are the estimated taxes? ")
|
|
)
|
|
|
|
|
|
class Rental:
|
|
|
|
def __init__(self, furnished="", utilities="", rent="", **kwargs):
|
|
super().__init__(**kwargs)
|
|
self.furnished = furnished
|
|
self.utilities = utilities
|
|
self.rent = rent
|
|
|
|
def display(self):
|
|
super().display()
|
|
print("RENTAL DETAILS")
|
|
print("rent: {}".format(self.rent))
|
|
print("estimated utilities: {}".format(self.utilities))
|
|
print("furnished: {}".format(self.furnished))
|
|
print()
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
return dict(
|
|
rent = input("What is the monthly rent? "),
|
|
utilities=input("What are the estimated utilities? "),
|
|
furnished=get_valid_input("Is the property furnished?",
|
|
("yes", "no"))
|
|
)
|
|
|
|
|
|
class HouseRental(Rental, House):
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
init = House.prompt_init()
|
|
init.update(Rental.prompt_init())
|
|
return init
|
|
|
|
|
|
class ApartmentRental(Rental, Apartment):
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
init = Apartment.prompt_init()
|
|
init.update(Rental.prompt_init())
|
|
return init
|
|
|
|
|
|
class HousePurchase(Purchase, House):
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
init = House.prompt_init()
|
|
init.update(Purchase.prompt_init())
|
|
return init
|
|
|
|
|
|
class ApartmentPurchase(Purchase, Apartment):
|
|
|
|
@staticmethod
|
|
def prompt_init():
|
|
init = Apartment.prompt_init()
|
|
init.update(Purchase.prompt_init())
|
|
return init
|
|
|
|
|
|
class Agent:
|
|
|
|
type_map = {
|
|
("house", "rental"): HouseRental,
|
|
("house", "purchase"): HousePurchase,
|
|
("apartment", "rental"): ApartmentRental,
|
|
("apartment", "purchase"): ApartmentPurchase
|
|
}
|
|
|
|
def __init__(self):
|
|
self.property_list = []
|
|
|
|
def display_properties(self):
|
|
for property in self.property_list:
|
|
property.display()
|
|
|
|
def add_property(self):
|
|
property_type = get_valid_input("What type of property?",
|
|
("house", "apartment")).lower()
|
|
payment_type = get_valid_input("What payment type?",
|
|
("purchase", "rental")).lower()
|
|
property_class = self.type_map[(property_type, payment_type)]
|
|
init_args = property_class.prompt_init()
|
|
self.property_list.append(property_class(**init_args))
|
|
|
|
|
|
def get_valid_input(input_string, valid_options):
|
|
input_string += " ({}) ".format(", ".join(valid_options))
|
|
response = ""
|
|
while response.lower() not in valid_options:
|
|
response = input(input_string)
|
|
return response
|