def run():
class TemplateDaemon(Daemon):
stopProcessing = False
def __init__(self, *args, **kwargs):
self._args = kwargs.pop('args', None)
super(TemplateDaemon, self).__init__(*args, **kwargs)
def run(self):
"""
Main program process
:return: Exit code value
"""
# Set SIGTERM signal Handler
signal.signal(signal.SIGTERM, signal_term_handler)
_logger.info('signal handlers set.')
return 0
def signal_term_handler(_sig, _frame):
if not _daemon.stopProcessing:
_logger.warning('received SIGTERM signal.')
_daemon.stopProcessing = True
# Set global debug value and setup application logging.
setup_logging(_logger, progname, '--debug' in sys.argv,
'{0}.log'.format(progname))
setup_unicode()
# Setup program arguments.
parser = argparse.ArgumentParser(prog=progname)
# pylint: disable=E0602
parser.add_argument('--debug',
help=_('Enable debug output'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--root-only',
help=_('Must run as root user'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--nodaemon',
help=_('Do not daemonize process'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('action',
choices=('start', 'stop', 'restart'),
default='') # noqa
args = parser.parse_args()
if args.root_only is True and os.getuid() != 0:
_logger.warning('daemon must be run as root')
sys.exit(4)
# --nodaemon only valid with start action
if args.nodaemon and args.action != 'start':
print(
'{0}: error: --nodaemon option not valid with stop or restart action'
.format(progname))
sys.exit(1)
# Do not fork the daemon process for systemd service or debugging, run in foreground.
if args.nodaemon is True:
_logger.info('running daemon in foreground.')
_daemon = TemplateDaemon(args=args)
_daemon.run()
else:
pidpath = os.path.expanduser('~/.local/run')
pidfile = os.path.join(pidpath, '{0}.pid'.format(progname))
logpath = os.path.expanduser('~/.local/log')
logfile = os.path.join(logpath, '{0}.log'.format(progname))
if args.action == 'start':
_logger.info('running daemon in background.')
# Setup daemon object
# make sure PID and Logging path exist
if not os.path.exists(pidpath):
os.makedirs(pidpath)
if not os.path.exists(logpath):
os.makedirs(logpath)
_daemon = TemplateDaemon(procbase='',
dirmask='0o700',
pidfile=pidfile,
uid='root',
gid='root',
stdin='/dev/null',
stdout=logfile,
stderr=logfile,
args=args)
if args.action == 'start':
_logger.debug('Starting daemon.')
_daemon.start()
elif args.action == 'stop':
_logger.debug('Stopping daemon.')
_daemon.stop()
elif args.action == 'restart':
_logger.debug('Restarting daemon.')
_daemon.restart()
return 0
def run():
class TemplateDaemon(Daemon):
stopProcessing = False
def __init__(self, *args, **kwargs):
self._args = kwargs.pop('args', None)
super(TemplateDaemon, self).__init__(*args, **kwargs)
def run(self):
"""
Main program process
:return: Exit code value
"""
# Set SIGTERM signal Handler
signal.signal(signal.SIGTERM, signal_term_handler)
_logger.debug('signal handlers set.')
result = gcp_set_account(args.account)
if result is not True:
_logger.error(
'failed to set authentication account, aborting.')
return 1
po_proxy = gcp_activate_proxy(args.enable_sandbox,
args.enable_test)
if not po_proxy:
_logger.error('cloud_sql_proxy process failed, aborting.')
return 1
while self.stopProcessing is False:
time.sleep(0.5)
_logger.debug('stopping cloud_sql_proxy process...')
po_proxy.kill()
_logger.debug('stopped')
return 0
def signal_term_handler(_sig, _frame):
if not _daemon.stopProcessing:
_logger.warning('received SIGTERM signal.')
_daemon.stopProcessing = True
# Set global debug value and setup application logging.
setup_logging(_logger, progname, '--debug' in sys.argv)
setup_unicode()
# Setup program arguments.
parser = argparse.ArgumentParser(prog=progname)
# pylint: disable=E0602
parser.add_argument('--debug',
help=_('Enable debug output'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--root-only',
help=_('Must run as root user'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--nodaemon',
help=_('Do not daemonize process'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--account', help=_('Security account'))
# pylint: disable=E0602
parser.add_argument('--enable-sandbox',
help=_('Add proxy to all-of-us-rdr-sandbox'),
default=False,
action='store_true') # noqa
# pylint: disable=E0602
parser.add_argument('--enable-test',
help=_('Add proxy to pmi-drc-api-test'),
default=False,
action='store_true') # noqa
parser.add_argument('action',
choices=('start', 'stop', 'restart'),
default='') # noqa
args = parser.parse_args()
if args.root_only is True and os.getuid() != 0:
_logger.warning('daemon must be run as root')
sys.exit(4)
if is_valid_email(args.account) is False:
if 'RDR_ACCOUNT' not in os.environ:
_logger.error(
'account parameter is invalid and RDR_ACCOUNT shell var is not set.'
)
return 1
else:
args.account = os.environ['RDR_ACCOUNT']
if which('cloud_sql_proxy') is None:
_logger.error(
'cloud_sql_proxy executable not found, ' +
'create symlink to cloud_sql_proxy in /usr/local/bin/ directory')
# --nodaemon only valid with start action
if args.nodaemon and args.action != 'start':
print(
'{0}: error: --nodaemon option not valid with stop or restart action'
.format(progname))
sys.exit(1)
_logger.info(' account: {0}'.format(args.account))
if args.action == 'start':
_logger.info(' tcp: 127.0.0.1:9900 -> all-of-us-rdr-prod')
_logger.info(' tcp: 127.0.0.1:9910 -> all-of-us-rdr-stable')
_logger.info(' tcp: 127.0.0.1:9920 -> all-of-us-rdr-staging')
if args.enable_sandbox is True:
_logger.info(
' tcp: 127.0.0.1:9930 -> all-of-us-rdr-sandbox')
if args.enable_test:
_logger.info(' tcp: 127.0.0.1:9940 -> all-of-us-rdr-test')
# Do not fork the daemon process for systemd service or debugging, run in foreground.
if args.nodaemon is True:
if args.action == 'start':
_logger.info('running daemon in foreground.')
_daemon = TemplateDaemon(args=args)
_daemon.run()
else:
pidpath = os.path.expanduser('~/.local/run')
pidfile = os.path.join(pidpath, '{0}.pid'.format(progname))
logpath = os.path.expanduser('~/.local/log')
logfile = os.path.join(logpath, '{0}.log'.format(progname))
if args.action == 'start':
_logger.info('running daemon in background.')
# Setup daemon object
# make sure PID and Logging path exist
if not os.path.exists(pidpath):
os.makedirs(pidpath)
if not os.path.exists(logpath):
os.makedirs(logpath)
_daemon = TemplateDaemon(procbase='',
dirmask='0o700',
pidfile=pidfile,
uid='root',
gid='root',
stdin='/dev/null',
stdout=logfile,
stderr=logfile,
args=args)
if args.action == 'start':
_logger.debug('Starting daemon.')
_daemon.start()
elif args.action == 'stop':
_logger.debug('Stopping daemon.')
_daemon.stop()
elif args.action == 'restart':
_logger.debug('Restarting daemon.')
_daemon.restart()
return 0