def main(argv=sys.argv): # pragma: no cover if len(argv) != 2: usage(argv) config_uri = argv[1] env = bootstrap(config_uri) worker = WorkerCommand(app=Celery(env)) worker.run()
def main(): usage = "usage: %prog [options] [broker-url]" epilog = """\ The worker needs Filetracker server configured. If no FILETRACKER_URL is present in the environment, a sensible default is generated, using the same host as the Celery broker uses, with default Filetracker port.""" parser = OptionParser(usage=usage, epilog=epilog) parser.disable_interspersed_args() os.environ.setdefault('CELERY_CONFIG_MODULE', 'sio.celery.default_config') app = Celery() cmd = WorkerCommand(app) for x in cmd.get_options(): parser.add_option(x) options, args = parser.parse_args() if len(args) > 1: parser.error("Unexpected arguments: " + ' '.join(args[1:])) if args: broker_url = args[0] os.environ['CELERY_BROKER_URL'] = args[0] if 'FILETRACKER_URL' not in os.environ: default_filetracker_host = None if 'CELERY_BROKER_URL' in os.environ: default_filetracker_host = \ _host_from_url(os.environ['CELERY_BROKER_URL']) if not default_filetracker_host: default_filetracker_host = '127.0.0.1' os.environ['FILETRACKER_URL'] = 'http://%s:%d' \ % (default_filetracker_host, DEFAULT_FILETRACKER_PORT) return cmd.run(**vars(options))
def main(): usage = "usage: %prog [options] [broker-url]" epilog = """\ The worker needs Filetracker server configured. If no FILETRACKER_URL is present in the environment, a sensible default is generated, using the same host as the Celery broker uses, with default Filetracker port.""" parser = OptionParser(usage=usage, epilog=epilog) parser.disable_interspersed_args() app = Celery(loader=Loader) cmd = WorkerCommand(app) for x in cmd.get_options(): parser.add_option(x) options, args = parser.parse_args() if len(args) > 1: parser.error("Unexpected arguments: " + " ".join(args[1:])) if args: broker_url = args[0] os.environ["CELERY_BROKER_URL"] = args[0] if "FILETRACKER_URL" not in os.environ: default_filetracker_host = None if "CELERY_BROKER_URL" in os.environ: default_filetracker_host = _host_from_url(os.environ["CELERY_BROKER_URL"]) if not default_filetracker_host: default_filetracker_host = "127.0.0.1" os.environ["FILETRACKER_URL"] = "http://%s:%d" % (default_filetracker_host, DEFAULT_FILETRACKER_PORT) return cmd.run(**vars(options))
def test_setup_concurrency_very_early(self): x = WorkerCommand() x.run = Mock() with self.assertRaises(ImportError): x.execute_from_commandline(['celeryd', '-P', 'xyzybox'])
def test_invalid_loglevel_gives_error(self): x = WorkerCommand(app=Celery(set_as_current=False)) with self.assertRaises(SystemExit): x.run(loglevel='GRIM_REAPER')