def main(args=sys.argv): parser = argparse.ArgumentParser(description='Run an agent.') parser.add_argument('--broker', dest='broker', default=DEFAULT_FRONTEND, help="ZMQ socket to the broker.") parser.add_argument('--receiver', dest='receiver', default=DEFAULT_AGENT_RECEIVER, help="ZMQ socket to get results from workers.") parser.add_argument('--debug', action='store_true', default=False, help="Debug mode") parser.add_argument('--logfile', dest='logfile', default='stdout', help="File to log in to.") parser.add_argument('--params', dest='params', default=None, help='The parameters to be used by the agent.') parser.add_argument('--timeout', dest='timeout', type=float, default=DEFAULT_TIMEOUT_MOVF, help=('The maximum time allowed before the thread ' 'stacks is dump and the message result not sent ' 'back.')) parser.add_argument('--max-age', dest='max_age', type=float, default=DEFAULT_MAX_AGE, help=('The maximum age for a worker in seconds. ' 'After that delay, the worker will simply quit. ' 'When set to -1, never quits.')) parser.add_argument('--max-age-delta', dest='max_age_delta', type=int, default=DEFAULT_MAX_AGE_DELTA, help='The maximum value in seconds added to max_age') args = parser.parse_args() set_logger(args.debug, logfile=args.logfile) sys.path.insert(0, os.getcwd()) # XXX if args.params is None: params = {} else: params = decode_params(args.params) logger.info('Connecting to %s' % args.broker) agent = Agent(broker=args.broker, receiver=args.receiver, params=params, timeout=args.timeout, max_age=args.max_age, max_age_delta=args.max_age_delta) try: agent.start() except KeyboardInterrupt: return 1 finally: agent.stop() return 0
def main(args=sys.argv): parser = argparse.ArgumentParser(description='Run an agent.') parser.add_argument('--broker', dest='broker', default=DEFAULT_FRONTEND, help="ZMQ socket to the broker.") parser.add_argument('--debug', action='store_true', default=False, help="Debug mode") parser.add_argument('--logfile', dest='logfile', default='stdout', help="File to log in to.") parser.add_argument('--params', dest='params', default=None, help='The parameters to be used by the agent.') parser.add_argument('--timeout', dest='timeout', type=float, default=DEFAULT_TIMEOUT_MOVF, help=('The maximum time allowed before the thread ' 'stacks is dump and the message result not sent ' 'back.')) parser.add_argument('--max-age', dest='max_age', type=float, default=DEFAULT_MAX_AGE, help=('The maximum age for a worker in seconds. ' 'After that delay, the worker will simply quit. ' 'When set to -1, never quits.')) parser.add_argument('--max-age-delta', dest='max_age_delta', type=int, default=DEFAULT_MAX_AGE_DELTA, help='The maximum value in seconds added to max_age') args = parser.parse_args() set_logger(args.debug, logfile=args.logfile) sys.path.insert(0, os.getcwd()) # XXX if args.params is None: params = {} else: params = decode_params(args.params) logger.info('Connecting to %s' % args.broker) agent = Agent(broker=args.broker, params=params, timeout=args.timeout, max_age=args.max_age, max_age_delta=args.max_age_delta) try: agent.start() except KeyboardInterrupt: return 1 finally: agent.stop() return 0
def test_decode_multiple_colons(self): params = decode_params('one:tcp://foo|two:tcp://blah') items = params.items() items.sort() self.assertEqual(items, [('one', 'tcp://foo'), ('two', 'tcp://blah')])
def test_decode(self): params = decode_params('one:1|two:2') items = params.items() items.sort() self.assertEqual(items, [('one', '1'), ('two', '2')])
def main(args=sys.argv): parser = argparse.ArgumentParser(description='Run some watchers.') parser.add_argument('--backend', dest='backend', default=DEFAULT_BACKEND, help="ZMQ socket to the broker.") parser.add_argument('--register', dest='register', default=DEFAULT_REG, help="ZMQ socket for the registration.") parser.add_argument('target', help="Fully qualified name of the callable.") parser.add_argument('--debug', action='store_true', default=False, help="Debug mode") parser.add_argument('--logfile', dest='logfile', default='stdout', help="File to log in to.") parser.add_argument('--heartbeat', dest='heartbeat', default=DEFAULT_HEARTBEAT, help="ZMQ socket for the heartbeat.") parser.add_argument('--params', dest='params', default=None, help='The parameters to be used in the worker.') parser.add_argument('--timeout', dest='timeout', type=float, default=DEFAULT_TIMEOUT_MOVF, help=('The maximum time allowed before the thread ' 'stacks is dump and the message result not sent ' 'back.')) parser.add_argument('--max-age', dest='max_age', type=float, default=DEFAULT_MAX_AGE, help=('The maximum age for a worker in seconds. ' 'After that delay, the worker will simply quit. ' 'When set to -1, never quits.')) parser.add_argument('--max-age-delta', dest='max_age_delta', type=int, default=DEFAULT_MAX_AGE_DELTA, help='The maximum value in seconds added to max_age') args = parser.parse_args() set_logger(args.debug, logfile=args.logfile) sys.path.insert(0, os.getcwd()) # XXX target = resolve_name(args.target) if args.params is None: params = {} else: params = decode_params(args.params) logger.info('Worker registers at %s' % args.backend) logger.info('The heartbeat socket is at %r' % args.heartbeat) worker = Worker(target, backend=args.backend, heartbeat=args.heartbeat, register=args.register, params=params, timeout=args.timeout, max_age=args.max_age, max_age_delta=args.max_age_delta) try: worker.start() except KeyboardInterrupt: return 1 finally: worker.stop() return 0