import sys, getopt

def set_home_amount():
	arguments = sys.argv[1:]
	try:
		optlist, args = getopt.getopt(arguments, '', ['home-count='])
		for o, amount in optlist:
			if o == '--home-count':
				if int(amount) < 2:
					print('Home count too low - must be higher than 2')
					sys.exit(2)
				return int(amount)
		print('Missing argument: --home-count <amount>')
		sys.exit(2)
	except getopt.GetoptError as err:
		print(err)
		sys.exit(2)