730f9772f9
Also fixed LogPrinter regressions. Sorry for not doing that in a separate commit. Also made 'plum logs' show backlog. Yep, rolled that right in too. Gonna go whip myself now.
153 lines
3.9 KiB
Python
153 lines
3.9 KiB
Python
import datetime
|
|
import logging
|
|
import sys
|
|
import os
|
|
import re
|
|
|
|
from docopt import docopt
|
|
from inspect import getdoc
|
|
|
|
from .. import __version__
|
|
from ..service_collection import ServiceCollection
|
|
from .command import Command
|
|
from .log_printer import LogPrinter
|
|
|
|
from .errors import UserError
|
|
from .docopt_command import NoSuchCommand
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
console_handler = logging.StreamHandler()
|
|
console_handler.setFormatter(logging.Formatter())
|
|
console_handler.setLevel(logging.INFO)
|
|
root_logger = logging.getLogger()
|
|
root_logger.addHandler(console_handler)
|
|
root_logger.setLevel(logging.DEBUG)
|
|
|
|
# Disable requests logging
|
|
logging.getLogger("requests").propagate = False
|
|
|
|
try:
|
|
command = TopLevelCommand()
|
|
command.sys_dispatch()
|
|
except KeyboardInterrupt:
|
|
log.error("\nAborting.")
|
|
exit(1)
|
|
except UserError, e:
|
|
log.error(e.msg)
|
|
exit(1)
|
|
except NoSuchCommand, e:
|
|
log.error("No such command: %s", e.command)
|
|
log.error("")
|
|
log.error("\n".join(parse_doc_section("commands:", getdoc(e.supercommand))))
|
|
exit(1)
|
|
|
|
|
|
# stolen from docopt master
|
|
def parse_doc_section(name, source):
|
|
pattern = re.compile('^([^\n]*' + name + '[^\n]*\n?(?:[ \t].*?(?:\n|$))*)',
|
|
re.IGNORECASE | re.MULTILINE)
|
|
return [s.strip() for s in pattern.findall(source)]
|
|
|
|
|
|
class TopLevelCommand(Command):
|
|
""".
|
|
|
|
Usage:
|
|
plum [options] [COMMAND] [ARGS...]
|
|
plum -h|--help
|
|
|
|
Options:
|
|
--verbose Show more output
|
|
--version Print version and exit
|
|
|
|
Commands:
|
|
ps List services and containers
|
|
run Run a one-off command
|
|
start Start services
|
|
stop Stop services
|
|
|
|
"""
|
|
def ps(self, options):
|
|
"""
|
|
List services and containers.
|
|
|
|
Usage: ps
|
|
"""
|
|
for container in self._get_containers(all=False):
|
|
print container.name
|
|
|
|
def run(self, options):
|
|
"""
|
|
Run a one-off command.
|
|
|
|
Usage: run SERVICE COMMAND [ARGS...]
|
|
"""
|
|
service = self.service_collection.get(options['SERVICE'])
|
|
if service is None:
|
|
raise UserError("No such service: %s" % options['SERVICE'])
|
|
container_options = {
|
|
'command': [options['COMMAND']] + options['ARGS'],
|
|
}
|
|
container = service.create_container(**container_options)
|
|
stream = self.client.logs(container, stream=True)
|
|
for data in stream:
|
|
if data is None:
|
|
break
|
|
print data
|
|
|
|
def start(self, options):
|
|
"""
|
|
Start all services
|
|
|
|
Usage: start [-d]
|
|
"""
|
|
if options['-d']:
|
|
self.service_collection.start()
|
|
return
|
|
|
|
running = []
|
|
unstarted = []
|
|
|
|
for s in self.service_collection:
|
|
if len(s.containers()) == 0:
|
|
unstarted.append((s, s.create_container()))
|
|
else:
|
|
running += s.get_containers(all=False)
|
|
|
|
log_printer = LogPrinter(running + [c for (s, c) in unstarted])
|
|
|
|
for (s, c) in unstarted:
|
|
s.start_container(c)
|
|
|
|
try:
|
|
log_printer.run()
|
|
finally:
|
|
self.service_collection.stop()
|
|
|
|
def stop(self, options):
|
|
"""
|
|
Stop all services
|
|
|
|
Usage: stop
|
|
"""
|
|
self.service_collection.stop()
|
|
|
|
def logs(self, options):
|
|
"""
|
|
View containers' output
|
|
|
|
Usage: logs
|
|
"""
|
|
containers = self._get_containers(all=False)
|
|
print "Attaching to", list_containers(containers)
|
|
LogPrinter(containers, attach_params={'logs': True}).run()
|
|
|
|
def _get_containers(self, all):
|
|
return [c for s in self.service_collection for c in s.containers(all=all)]
|
|
|
|
|
|
def list_containers(containers):
|
|
return ", ".join(c.name for c in containers)
|