fastreload.py

from importlib import reload

from examples.zerg import zerg_rush
from sc2 import maps
from sc2.data import Difficulty, Race
from sc2.main import _host_game_iter
from sc2.player import AbstractPlayer, Bot, Computer


def main():
    player_config: list[AbstractPlayer] = [
        Bot(Race.Zerg, zerg_rush.ZergRushBot()),
        Computer(Race.Terran, Difficulty.Medium),
    ]

    gen = _host_game_iter(maps.get("Abyssal Reef LE"), player_config, realtime=False)

    _r = next(gen)
    while True:
        input("Press enter to reload ")

        reload(zerg_rush)
        if isinstance(player_config[0], Bot):
            player_config[0].ai = zerg_rush.ZergRushBot()
        gen.send(player_config)


if __name__ == "__main__":
    main()