from CvPythonExtensions import *
import CvEventInterface
import CvUtil
import BugUtil
import PyHelpers
gc = CyGlobalContext()
localText = CyTranslator()
PyPlayer = PyHelpers.PyPlayer
PyInfo = PyHelpers.PyInfo
PyCity = PyHelpers.PyCity
PyGame = PyHelpers.PyGame
class Militia:
def __init__(self, eventManager):
eventManager.addEventHandler("ImprovementBuilt", self.onImprovementBuilt)
def onImprovementBuilt(self, argsList):
'Improvement Built'
iImprovement, iX, iY = argsList
###Militia Start ###
pPlot = CyMap().plot(iX, iY)
if iImprovement == gc.getInfoTypeForString('IMPROVEMENT_FARM'):
charity = gc.getInfoTypeForString( "CIVIC_CHARITY" )
iPlayer = pPlot.getOwner()
pPlayer = gc.getPlayer(iPlayer)
if (pPlayer.isCivic(charity)==True):
era = pPlayer.getCurrentEra ()
if (era == gc.getInfoTypeForString( "ERA_MEDIEVAL" )):
militia = gc.getInfoTypeForString( 'UNIT_MILITIA_MEDIEVAL' )
pNewUnit = pPlayer.initUnit( militia, iX, iY, UnitAITypes.UNITAI_RESERVE, DirectionTypes.NO_DIRECTION )
elif (era == gc.getInfoTypeForString( "ERA_RENAISSANCE" )):
militia = gc.getInfoTypeForString( 'UNIT_MILITIA_RENAISSANCE' )
pNewUnit = pPlayer.initUnit( militia, iX, iY, UnitAITypes.UNITAI_RESERVE, DirectionTypes.NO_DIRECTION )
elif (era == gc.getInfoTypeForString( "ERA_INDUSTRIAL" )):
militia = gc.getInfoTypeForString( 'UNIT_MILITIA_INDUSTRIAL' )
pNewUnit = pPlayer.initUnit( militia, iX, iY, UnitAITypes.UNITAI_RESERVE, DirectionTypes.NO_DIRECTION )
elif (era == gc.getInfoTypeForString( "ERA_MODERN" )):
militia = gc.getInfoTypeForString( 'UNIT_MILITIA_MODERN' )
pNewUnit = pPlayer.initUnit( militia, iX, iY, UnitAITypes.UNITAI_RESERVE, DirectionTypes.NO_DIRECTION )
CyInterface().addMessage(iPlayer,False,15,CyTranslator().getText("TXT_RECRUITED",()),'',0,'Art/Interface/Buttons/Civics/Serfdom.dds',ColorTypes(44), iX, iY, True,True)
###Militia End###