exampleai.py

from pybw_swig import * # import all constants and classes
import pybw

class ExampleAI(object):

    def __init__(self):
        # Add ourselves to the console manager as variable "m"
        pybw.consoleManager.locals.update({'m': self})

    def onConnect(self):
        # Get the game instance
        self.game = getGame() 

    def onMatchStart(self):
        if self.game.isReplay:
            return
        # Get all minerals on the map
        self.minerals = list(self.game.minerals)
        self.player = self.game.self
        self.race = self.player.race

        self.main_centers = [u for u in self.player.units if u.type == self.race.center]
        if self.main_centers:
            self.main_center = self.main_centers[0]

            self.mineral_queue = []
            for mineral in self.minerals:
                distance_to_center = mineral.position.getDistance( self.main_center.position )
                # Only queue workers to get minerals next to the main base
                if distance_to_center < 250:
                    self.mineral_queue.append(( distance_to_center, mineral ))
            self.mineral_queue.sort()

        else:
            self.main_center = None

    def onMatchFrame(self):
        if self.game.isReplay:
            return
        if self.main_center:
            self.main_center.train( self.race.worker )

            inactive_workers = [u for u in self.game.self.units if u.type.isWorker and u.order == PlayerGuard]

            for worker in inactive_workers:
                mineral = self.mineral_queue.pop(0)
                worker.rightClick( mineral[1] )
                self.mineral_queue.append( mineral )