Я хочу создать main внутри класса, используя Python 3.7.Я запутался, что как я могу написать основную функцию с помощью try и исключением, а также с get getRoom и def getCatagory.Я пытался создать def main (), но это не сработало.Любое предложение будет оценено.Мой код:
class Hotel:
def __init__(self,room,catagory):
if type(catagory) != str:
raise TypeError()
if room<0 or room >5:
raise ValueError()
self.room = room
self.catagory = catagory
self.catagories = {"A":"Elite","B":"Economy","C":"Regular"}
self.rooms = ["0","1","2","3","4","5"]
def getRoom(self):
return self.room
def getCatagory(self):
return self.catagories.get(self.catagory)
def __str__(self):
return "%s and %s"%(self.rooms[self.room],self.catagories.get(self.catagory))
if (__name__== "__ main__"):
r = Hotel ()
print (r)
try:
room1 = Hotel(2,1)
print (room1)
except TypeError:
print("Error")
try:
room2= Hotel(1,"A")
print (room2)
except ValueError:
print ("mistake")
Спасибо.