22 lines
582 B
Python
22 lines
582 B
Python
import sys, getopt, random
|
|
|
|
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)
|
|
|
|
def generate_rand_coordinates(max_x, max_y):
|
|
return (random.randint(0, max_x), random.randint(0, (max_y)))
|
|
|