def __init__(self, bot, tasks_raw):
self.bot = bot
self.tasks_raw = tasks_raw
self.plugin_loader = PluginLoader()
def init_config():
parser = argparse.ArgumentParser()
config_file = os.path.join(_base_dir, 'configs', 'config.json')
web_dir = "web"
# If config file exists, load variables from json
load = {}
def _json_loader(filename):
try:
with open(filename, 'rb') as data:
load.update(json.load(data))
except ValueError:
if jsonlint:
with open(filename, 'rb') as data:
lint = jsonlint()
rc = lint.main(['-v', filename])
logger.critical('Error with configuration file')
sys.exit(-1)
# Select a config file code
parser.add_argument("-cf", "--config", help="Config File to use")
parser.add_argument("-af", "--auth", help="Auth File to use")
for _config in ['auth', 'config']:
config_file = os.path.join(_base_dir, 'configs', _config + '.json')
config_arg = parser.parse_known_args() and parser.parse_known_args(
)[0].__dict__[_config] or None
if config_arg and os.path.isfile(config_arg):
_json_loader(config_arg)
config_file = config_arg
elif os.path.isfile(config_file):
logger.info('No ' + _config +
' argument specified, checking for ' + config_file)
_json_loader(config_file)
else:
logger.info('Error: No /configs/' + _config + '.json')
# Read passed in Arguments
required = lambda x: x not in load
add_config(parser,
load,
short_flag="-a",
long_flag="--auth_service",
help="Auth Service ('ptc' or 'google')",
required=required("auth_service"),
default=None)
add_config(parser,
load,
short_flag="-u",
long_flag="--username",
help="Username",
default=None)
add_config(parser,
load,
short_flag="-capi",
long_flag="--check_niantic_api",
help="Enable killswitch on API Change",
type=bool,
default=True)
add_config(parser,
load,
short_flag="-ws",
long_flag="--websocket.server_url",
help="Connect to websocket server at given url",
default=False)
add_config(parser,
load,
short_flag="-wss",
long_flag="--websocket.start_embedded_server",
help="Start embedded websocket server",
default=False)
add_config(
parser,
load,
short_flag="-wsr",
long_flag="--websocket.remote_control",
help=
"Enable remote control through websocket (requires websocket server url)",
default=False)
add_config(parser,
load,
short_flag="-p",
long_flag="--password",
help="Password",
default=None)
add_config(parser,
load,
short_flag="-l",
long_flag="--location",
help="Location",
type=parse_unicode_str,
default='')
add_config(parser,
load,
short_flag="-lc",
long_flag="--location_cache",
help="Bot will start at last known location",
type=bool,
default=False)
add_config(
parser,
load,
long_flag="--forts.spin",
help="Enable Spinning Pokestops",
type=bool,
default=True,
)
add_config(parser,
load,
short_flag="-wmax",
long_flag="--walk_max",
help="Walk instead of teleport with given speed",
type=float,
default=2.5)
add_config(parser,
load,
short_flag="-wmin",
long_flag="--walk_min",
help="Walk instead of teleport with given speed",
type=float,
default=2.5)
add_config(parser,
load,
short_flag="-k",
long_flag="--gmapkey",
help="Set Google Maps API KEY",
type=str,
default=None)
add_config(parser,
load,
short_flag="-hk",
long_flag="--hashkey",
help="Set Bossland hashing key",
type=str,
default=None)
add_config(parser,
load,
short_flag="-e",
long_flag="--show_events",
help="Show events",
type=bool,
default=False)
add_config(parser,
load,
short_flag="-d",
long_flag="--debug",
help="Debug Mode",
type=bool,
default=False)
add_config(parser,
load,
short_flag="-t",
long_flag="--test",
help="Only parse the specified location",
type=bool,
default=False)
add_config(
parser,
load,
short_flag="-du",
long_flag="--distance_unit",
help=
"Set the unit to display distance in (e.g, km for kilometers, mi for miles, ft for feet)",
type=str,
default='km')
add_config(
parser,
load,
short_flag="-rt",
long_flag="--reconnecting_timeout",
help=
"Timeout between reconnecting if error occured (in minutes, e.g. 15)",
type=float,
default=15.0)
add_config(
parser,
load,
short_flag="-hr",
long_flag="--health_record",
help=
"Send anonymous bot event to GA for bot health record. Set \"health_record\":false if you need disable it.",
type=bool,
default=True)
add_config(
parser,
load,
short_flag="-ac",
long_flag="--forts.avoid_circles",
help=
"Avoids circles (pokestops) of the max size set in max_circle_size flag",
type=bool,
default=False,
)
add_config(
parser,
load,
short_flag="-mcs",
long_flag="--forts.max_circle_size",
help=
"If avoid_circles flag is set, this flag specifies the maximum size of circles (pokestops) avoided",
type=int,
default=10,
)
add_config(
parser,
load,
short_flag="-crf",
long_flag="--forts.cache_recent_forts",
help="Caches recent forts used by max_circle_size",
type=bool,
default=True,
)
add_config(
parser,
load,
long_flag="--map_object_cache_time",
help=
"Amount of seconds to keep the map object in cache (bypass Niantic throttling)",
type=float,
default=5.0)
add_config(
parser,
load,
long_flag="--logging.color",
help=
"If logging_color is set to true, colorized logging handler will be used",
type=bool,
default=True)
add_config(
parser,
load,
long_flag="--logging.clean",
help=
"If clean_logging is set to true, meta data will be stripped from the log messages",
type=bool,
default=False)
add_config(parser,
load,
long_flag="--heartbeat_threshold",
help="A threshold between each heartbeat sending to server",
type=int,
default=10)
add_config(parser,
load,
long_flag="--pokemon_bag.show_at_start",
help="Logs all pokemon in the bag at bot start",
type=bool,
default=False)
add_config(parser,
load,
long_flag="--pokemon_bag.show_count",
help="Shows the amount of which pokemon (minimum 1)",
type=bool,
default=False)
add_config(parser,
load,
long_flag="--pokemon_bag.show_candies",
help="Shows the amount of candies for each pokemon",
type=bool,
default=False)
add_config(parser,
load,
long_flag="--pokemon_bag.pokemon_info",
help="List with the info to show for each pokemon",
type=bool,
default=[])
add_config(parser,
load,
long_flag="--alt_min",
help="Minimum random altitude",
type=float,
default=500)
add_config(parser,
load,
long_flag="--alt_max",
help="Maximum random altitude",
type=float,
default=1000)
add_config(parser,
load,
long_flag="--replicate_gps_xy_noise",
help="Add noise to current position",
type=bool,
default=False)
add_config(parser,
load,
long_flag="--replicate_gps_z_noise",
help="Add noise to current position",
type=bool,
default=False)
add_config(
parser,
load,
long_flag="--gps_xy_noise_range",
help=
"Intensity of gps noise (unit is lat and lng,) high values may cause issues (default=0.000125)",
type=float,
default=0.000125)
add_config(parser,
load,
long_flag="--gps_z_noise_range",
help="Intensity of gps noise (unit is in meter, default=12.5)",
type=float,
default=12.5)
add_config(parser,
load,
long_flag="--gps_default_altitude",
help="Initial altitude (default=8.0)",
type=float,
default=8.0)
add_config(parser,
load,
long_flag="--enable_social",
help="Enable social event exchange between bot",
type=bool,
default=True)
add_config(
parser,
load,
long_flag="--walker_limit_output",
help=
"Limit output from walker functions (move_to_fort, position_update, etc)",
type=bool,
default=False)
# Start to parse other attrs
config = parser.parse_args()
if not config.username and 'username' not in load:
config.username = raw_input("Username: "******"Password: "******"--map_object_cache_time is out of range! (should be >= 0.0)")
return None
if len(config.raw_tasks) == 0:
logging.error(
"No tasks are configured. Did you mean to configure some behaviors? Read https://github.com/PokemonGoF/PokemonGo-Bot/wiki/Configuration-files#configuring-tasks for more information"
)
return None
if config.auth_service not in ['ptc', 'google']:
logging.error("Invalid Auth service specified! ('ptc' or 'google')")
return None
def task_configuration_error(flag_name):
parser.error("""
\"{}\" was removed from the configuration options.
You can now change the behavior of the bot by modifying the \"tasks\" key.
Read https://github.com/PokemonGoF/PokemonGo-Bot/wiki/Configuration-files#configuring-tasks for more information.
""".format(flag_name))
old_flags = [
'mode', 'catch_pokemon', 'spin_forts', 'forts_spin', 'hatch_eggs',
'release_pokemon', 'softban_fix', 'longer_eggs_first', 'evolve_speed',
'use_lucky_egg', 'item_filter', 'evolve_all', 'evolve_cp_min',
'max_steps', 'catch_throw_parameters.excellent_rate',
'catch_throw_parameters.great_rate',
'catch_throw_parameters.nice_rate',
'catch_throw_parameters.normal_rate',
'catch_throw_parameters.spin_success_rate'
]
for flag in old_flags:
if flag in load:
task_configuration_error(flag)
return None
nested_old_flags = [('forts', 'spin'), ('forts', 'move_to_spin'),
('navigator', 'path_mode'), ('navigator', 'path_file'),
('navigator', 'type')]
for outer, inner in nested_old_flags:
if load.get(outer, {}).get(inner, None):
task_configuration_error('{}.{}'.format(outer, inner))
return None
if "evolve_captured" in load:
logger.warning(
'The evolve_captured argument is no longer supported. Please use the EvolvePokemon task instead'
)
if "walk" in load:
logger.warning(
'The walk argument is no longer supported. Please use the walk_max and walk_min variables instead'
)
if "daily_catch_limit" in load:
logger.warning(
'The daily_catch_limit argument has been moved into the CatchPokemon Task'
)
if "logging_color" in load:
logger.warning(
'The logging_color argument has been moved into the logging config section'
)
if config.walk_min < 1:
parser.error("--walk_min is out of range! (should be >= 1.0)")
return None
if config.alt_min < -413.0:
parser.error("--alt_min is out of range! (should be >= -413.0)")
return None
if not (config.location or config.location_cache):
parser.error("Needs either --use-location-cache or --location.")
return None
plugin_loader = PluginLoader()
for plugin in config.plugins:
plugin_loader.load_plugin(plugin)
# create web dir if not exists
try:
os.makedirs(web_dir)
except OSError:
if not os.path.isdir(web_dir):
raise
fix_nested_config(config)
return config, config_file
def setUp(self):
self.plugin_loader = PluginLoader()
