9a6d66dfea
git-svn-id: svn://svn.cy55.de/Zope3/src/cybertools/trunk@2924 fd906abe-77d9-0310-91a1-e0d9ade77398
121 lines
3.7 KiB
Python
121 lines
3.7 KiB
Python
#
|
|
# Copyright (c) 2008 Helmut Merz helmutm@cy55.de
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation; either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
#
|
|
|
|
"""
|
|
Agent base and sample classes.
|
|
|
|
$Id$
|
|
"""
|
|
|
|
from zope.interface import implements
|
|
|
|
from cybertools.agent.common import states
|
|
from cybertools.agent.components import agents, controllers, jobs
|
|
from cybertools.agent.components import loggers, schedulers
|
|
from cybertools.agent.components import servers, clients
|
|
from cybertools.agent.interfaces import IAgent
|
|
from cybertools.util.config import Configurator
|
|
|
|
|
|
class Agent(object):
|
|
|
|
implements(IAgent)
|
|
|
|
name = '???'
|
|
master = None
|
|
config = None
|
|
logger = None
|
|
|
|
def __init__(self, master):
|
|
self.master = master
|
|
self.config = master.config
|
|
self.logger = master.logger
|
|
|
|
def send(self, job):
|
|
self.execute(job)
|
|
|
|
def execute(self, job):
|
|
pass
|
|
|
|
def log(self, job, result='OK'):
|
|
self.logger.log(dict(message='job execution', job=job.identifier,
|
|
agent=self.name, result=result))
|
|
|
|
|
|
class Master(Agent):
|
|
|
|
name = 'master'
|
|
scheduler = None
|
|
logger = None
|
|
|
|
def __init__(self, configuration):
|
|
if isinstance(configuration, Configurator):
|
|
self.config = configuration
|
|
else: # configuration is path to config file
|
|
self.config = Configurator()
|
|
self.config.load(configuration)
|
|
self.master = self
|
|
self.controllers = []
|
|
self.children = {}
|
|
self.servers = []
|
|
|
|
def setup(self):
|
|
config = self.config
|
|
self.logger = loggers(self, name=config.logger.name)
|
|
print 'Starting agent application...'
|
|
for n in config.controller.names:
|
|
self.controllers.append(controllers(self, n))
|
|
self.scheduler = schedulers(self, name=config.scheduler.name)
|
|
for cont in self.controllers:
|
|
cont.setup()
|
|
print 'Using controllers %s.' % ', '.join(config.controller.names)
|
|
for n in config.talk.server.names:
|
|
server = servers(self, n)
|
|
self.servers.append(server)
|
|
server.setup()
|
|
|
|
def setupAgents(self, controller, agentSpecs):
|
|
for spec in agentSpecs:
|
|
agent = agents(self, spec.type)
|
|
agent.name = spec.name
|
|
self.children[spec.name] = agent
|
|
|
|
def setupJobs(self, controller, jobSpecs):
|
|
for spec in jobSpecs:
|
|
job = jobs(self.scheduler, spec.type)
|
|
job.agent = self.children[spec.agent]
|
|
job.identifier = spec.identifier
|
|
job.controller = controller
|
|
job.params = spec.params
|
|
self.scheduler.schedule(job, spec.startTime)
|
|
|
|
def notify(self, job, result=None, message=''):
|
|
if job.state.hasFinished():
|
|
job.controller.notify(job.identifier, job.state, result, message)
|
|
|
|
|
|
class SampleAgent(Agent):
|
|
|
|
def execute(self, job):
|
|
job.state = states.running
|
|
print 'Job %s on agent %s has been executed.' % (job.identifier, self.name)
|
|
self.log(job)
|
|
job.state = states.completed
|
|
self.master.notify(job)
|
|
|
|
agents.register(SampleAgent, Master, name='base.sample')
|