18 lines
459 B
Python
18 lines
459 B
Python
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)
|