from t9a import  mainBranch

from t9b import fritidsboligBranch, secondaryHouseBranch, safeQuestion

choices = ['egen bolig', 'sekundærbolig', 'fritidsbolig']
choices.extend([choice.capitalize() for choice in choices])

choice = safeQuestion(
  question='Skriv inn hustype for skatteutregning (egen bolig, sekundærbolig, fritidsbolig): ',
  choices=choices
)

if choice in ['egen bolig', 'Egen bolig']:
  mainBranch()
elif choice in ['sekundærbolig', 'Sekundærbolig']:
  secondaryHouseBranch()
else:
  fritidsboligBranch()