class PluginLoaderTest(unittest.TestCase): def setUp(self): self.plugin_loader = PluginLoader() def test_load_namespace_class(self): package_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'resources', 'plugin_fixture') self.plugin_loader.load_path(package_path) loaded_class = self.plugin_loader.get_class('plugin_fixture.FakeTask') self.assertEqual(loaded_class({}, {}).work(), 'FakeTask') self.plugin_loader.remove_path(package_path) def test_load_zip(self): package_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'resources', 'plugin_fixture_test.zip') self.plugin_loader.load_path(package_path) loaded_class = self.plugin_loader.get_class('plugin_fixture_test.FakeTask') self.assertEqual(loaded_class({}, {}).work(), 'FakeTask') self.plugin_loader.remove_path(package_path)
def init_config(): parser = argparse.ArgumentParser() config_file = "configs/config.json" web_dir = "web" # If config file exists, load variables from json load = {} # Select a config file code parser.add_argument("-cf", "--config", help="Config File to use") config_arg = parser.parse_known_args() and parser.parse_known_args()[0].config or None if config_arg and os.path.isfile(config_arg): with open(config_arg) as data: load.update(json.load(data)) elif os.path.isfile(config_file): logger.info('No config argument specified, checking for /configs/config.json') with open(config_file) as data: load.update(json.load(data)) else: logger.info('Error: No /configs/config.json or specified config') # Read passed in Arguments required = lambda x: not x 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="-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 websocekt 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="-w", long_flag="--walk", help= "Walk instead of teleport with given speed (meters per second, e.g. 2.5)", 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="-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="-ec", long_flag="--evolve_captured", help="(Ad-hoc mode) Pass \"all\" or a list of pokemon to evolve (e.g., \"Pidgey,Weedle,Caterpie\"). Bot will attempt to evolve all the pokemon captured!", type=str, default=[] ) 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, long_flag="--catch_randomize_reticle_factor", help="Randomize factor for pokeball throwing accuracy (DEFAULT 1.0 means no randomize: always 'Excellent' throw. 0.0 randomizes between normal and 'Excellent' throw)", type=float, default=1.0 ) add_config( parser, load, long_flag="--catch_randomize_spin_factor", help="Randomize factor for pokeball curve throwing (DEFAULT 1.0 means no randomize: always perfect 'Super Spin' curve ball. 0.0 randomizes between normal and 'Super Spin' curve ball)", type=float, default=1.0 ) 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 ) # 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'] 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 (config.evolve_captured and (not isinstance(config.evolve_captured, str) or str(config.evolve_captured).lower() in ["true", "false"])): parser.error('"evolve_captured" should be list of pokemons: use "all" or "none" to match all ' + 'or none of the pokemons, or use a comma separated list such as "Pidgey,Weedle,Caterpie"') return None if not (config.location or config.location_cache): parser.error("Needs either --use-location-cache or --location.") return None if config.catch_randomize_reticle_factor < 0 or 1 < config.catch_randomize_reticle_factor: parser.error("--catch_randomize_reticle_factor is out of range! (should be 0 <= catch_randomize_reticle_factor <= 1)") return None if config.catch_randomize_spin_factor < 0 or 1 < config.catch_randomize_spin_factor: parser.error("--catch_randomize_spin_factor is out of range! (should be 0 <= catch_randomize_spin_factor <= 1)") return None plugin_loader = PluginLoader() for plugin in config.plugins: plugin_loader.load_path(plugin) # create web dir if not exists try: os.makedirs(web_dir) except OSError: if not os.path.isdir(web_dir): raise if config.evolve_captured and isinstance(config.evolve_captured, str): config.evolve_captured = [str(pokemon_name).strip() for pokemon_name in config.evolve_captured.split(',')] fix_nested_config(config) return config