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 ') 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)))