dddea2747d
Also fix other problems due to flake8 upgrades. Change-Id: I5cb6194e48c66e40ec7aa9739eb37550d6bf3e2a
80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
# Copyright (c) 2013 Rackspace Hosting, Inc.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
|
|
# implied.
|
|
#
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from __future__ import print_function
|
|
import atexit
|
|
import functools
|
|
import sys
|
|
import termios
|
|
|
|
from marconi.common import config
|
|
from marconi.openstack.common import log as logging
|
|
|
|
PROJECT_CFG = config.project('marconi')
|
|
LOG = logging.getLogger(__name__)
|
|
|
|
|
|
def _fail(returncode, ex):
|
|
"""Handles terminal errors.
|
|
|
|
:param returncode: process return code to pass to sys.exit
|
|
:param ex: the error that occurred
|
|
"""
|
|
|
|
print(ex, file=sys.stderr)
|
|
LOG.exception(ex)
|
|
sys.exit(returncode)
|
|
|
|
|
|
def _enable_echo(enable):
|
|
"""Enables or disables terminal echo.
|
|
|
|
:param enable: pass True to enable echo, False to disable
|
|
"""
|
|
|
|
fd = sys.stdin.fileno()
|
|
new_attr = termios.tcgetattr(fd)
|
|
if enable:
|
|
new_attr[3] |= termios.ECHO
|
|
else:
|
|
new_attr[3] &= ~termios.ECHO
|
|
|
|
termios.tcsetattr(fd, termios.TCSANOW, new_attr)
|
|
|
|
|
|
def runnable(func):
|
|
"""Entry point wrapper.
|
|
|
|
Note: This call blocks until the process is killed
|
|
or interrupted.
|
|
"""
|
|
|
|
@functools.wraps(func)
|
|
def _wrapper():
|
|
atexit.register(_enable_echo, True)
|
|
_enable_echo(False)
|
|
|
|
try:
|
|
logging.setup('marconi')
|
|
PROJECT_CFG.load(args=sys.argv[1:])
|
|
func()
|
|
except KeyboardInterrupt:
|
|
LOG.info('Terminating')
|
|
except Exception as ex:
|
|
_fail(1, ex)
|
|
|
|
return _wrapper
|