class QueueConfig(ConfigSection):
def __init__(self, *args, **kwargs):
ConfigSection.__init__(self, *args, **kwargs)
self.queue_name = None
self.config = Config(self.global_options.config_dir, False)
@staticmethod
def get_description():
return 'add, edit, or remove queues'
@staticmethod
def get_commands():
return (('add', 'create a new queue'),
('edit', 'edit an existing queue'))
def getopt_add(self, parser):
parser.add_option('--account-pool',
dest = 'account_pool',
metavar = 'STRING',
help = 'the account pool that is used')
parser.add_option('--max-threads',
dest = 'max_threads',
metavar = 'INT',
default = 5,
help = 'the name of the new queue')
def prepare_add(self, parser, queue_name):
self.queue_name = queue_name
if self.config.has_queue(self.queue_name):
parser.error('queue already exists')
def start_add(self):
self.config.add_queue(self.queue_name,
self.options.account_pool,
self.options.max_threads)
print 'Queue added.'
def getopt_edit(self, parser):
self.getopt_add(parser)
def prepare_edit(self, parser, queue_name):
self.queue_name = queue_name
if not self.config.has_queue(self.queue_name):
parser.error('queue not found')
def start_edit(self):
if self.config.add_queue(self.queue_name,
self.options.account_pool,
self.options.max_threads):
print 'Queue configured.'
else:
print 'No changes were made.'
class QueueConfig(ConfigSection):
def __init__(self, *args, **kwargs):
ConfigSection.__init__(self, *args, **kwargs)
self.queue_name = None
self.config = Config(self.global_options.config_dir, False)
@staticmethod
def get_description():
return 'add, edit, or remove queues'
@staticmethod
def get_commands():
return (('add', 'create a new queue'), ('edit',
'edit an existing queue'))
def getopt_add(self, parser):
parser.add_option('--account-pool',
dest='account_pool',
metavar='STRING',
help='the account pool that is used')
parser.add_option('--max-threads',
dest='max_threads',
metavar='INT',
default=5,
help='the name of the new queue')
def prepare_add(self, parser, queue_name):
self.queue_name = queue_name
if self.config.has_queue(self.queue_name):
parser.error('queue already exists')
def start_add(self):
self.config.add_queue(self.queue_name, self.options.account_pool,
self.options.max_threads)
print 'Queue added.'
def getopt_edit(self, parser):
self.getopt_add(parser)
def prepare_edit(self, parser, queue_name):
self.queue_name = queue_name
if not self.config.has_queue(self.queue_name):
parser.error('queue not found')
def start_edit(self):
if self.config.add_queue(self.queue_name, self.options.account_pool,
self.options.max_threads):
print 'Queue configured.'
else:
print 'No changes were made.'
