Page MenuHomePhabricator
Files F7159564

No OneTemporary
Actions

View Options

diff --git a/media-dispatcher b/media-dispatcher
index 3b38cea..347bba8 100644
--- a/media-dispatcher
+++ b/media-dispatcher
@@ -1,65 +1,80 @@
#!/usr/bin/env python
if __name__ == '__main__':
+ import mediaproxy
import sys
- from optparse import OptionParser
- from application.process import process, ProcessError
- from application.configuration import ConfigFile, datatypes
+
from application import log
- import mediaproxy
+ from application.configuration import ConfigFile, datatypes
+ from application.process import process, ProcessError
+ from argparse import ArgumentParser
name = 'media-dispatcher'
fullname = 'MediaProxy Dispatcher'
description = 'MediaProxy Dispatcher component'
process.configuration.user_directory = None
process.configuration.subdirectory = mediaproxy.mediaproxy_subdirectory
process.runtime.subdirectory = mediaproxy.mediaproxy_subdirectory
- default_pid = process.runtime.file('{}.pid'.format(name))
+ parser = ArgumentParser(usage='%(prog)s [options]')
+ parser.add_argument('--version', action='version', version='%(prog)s {}'.format(mediaproxy.__version__))
+ parser.add_argument('--systemd', action='store_true', help='run as a systemd simple service and log to journal')
+ parser.add_argument('--no-fork', action='store_false', dest='fork', help='run in the foreground and log to the terminal')
+ parser.add_argument('--config-dir', dest='config_directory', default=None, help='the configuration directory ({})'.format(process.configuration.system_directory), metavar='PATH')
+ parser.add_argument('--runtime-dir', dest='runtime_directory', default=None, help='the runtime directory ({})'.format(process.runtime.directory), metavar='PATH')
+ parser.add_argument('--debug', action='store_true', help='enable verbose logging')
+ parser.add_argument('--debug-memory', action='store_true', help='enable memory debugging')
- parser = OptionParser(version="%%prog %s" % mediaproxy.__version__)
- parser.add_option("--no-fork", action="store_false", dest="fork", default=1, help="run the process in the foreground (for debugging)")
- parser.add_option("--pid", dest="pid_file", default=default_pid, help="pid file (%s)" % default_pid, metavar="File")
- (options, args) = parser.parse_args()
+ options = parser.parse_args()
log.Formatter.prefix_format = '{record.levelname:<8s} '
+ if options.config_directory is not None:
+ process.configuration.local_directory = options.config_directory
+ if options.runtime_directory is not None:
+ process.runtime.directory = options.runtime_directory
+
config_file = ConfigFile(mediaproxy.configuration_file)
log.level.current = config_file.get_setting('Dispatcher', 'log_level', type=datatypes.LogLevel, default=log.level.DEBUG)
try:
process.runtime.create_directory()
except ProcessError as e:
log.critical('Cannot start %s: %s' % (fullname, e))
sys.exit(1)
- if options.fork:
+ if options.systemd:
+ from systemd.journal import JournalHandler
+ log.set_handler(JournalHandler(SYSLOG_IDENTIFIER=name))
+ log.capture_output()
+ elif options.fork:
try:
process.daemonize(options.pid_file)
except ProcessError as e:
log.critical('Cannot start %s: %s' % (fullname, e))
sys.exit(1)
log.use_syslog(name)
log.info('Starting %s %s' % (fullname, mediaproxy.__version__))
from mediaproxy.dispatcher import Dispatcher
- if not options.fork:
+ if options.debug:
+ log.level.current = log.level.DEBUG
+ if options.debug_memory:
from application.debug.memory import memory_dump
try:
dispatcher = Dispatcher()
except Exception as e:
log.critical('Failed to create %s: %s' % (fullname, e))
if type(e) is not RuntimeError:
log.exception()
sys.exit(1)
dispatcher.run()
- if not options.fork:
- #from application.debug.memory import memory_dump
+ if options.debug_memory:
memory_dump()
diff --git a/media-relay b/media-relay
index a02f86e..e10c60d 100644
--- a/media-relay
+++ b/media-relay
@@ -1,101 +1,116 @@
#!/usr/bin/env python
if __name__ == '__main__':
+ import mediaproxy
import errno
import sys
import subprocess
- from optparse import OptionParser
+
from application import log
from application.configuration import ConfigFile, datatypes
from application.process import process, ProcessError
from application.version import Version
- import mediaproxy
+ from argparse import ArgumentParser
IP_FORWARD_FILE = '/proc/sys/net/ipv4/ip_forward'
CONNTRACK_ACCT_FILE = '/proc/sys/net/netfilter/nf_conntrack_acct'
KERNEL_VERSION_FILE = '/proc/sys/kernel/osrelease'
name = 'media-relay'
fullname = 'MediaProxy Relay'
description = 'MediaProxy Relay component'
process.configuration.user_directory = None
process.configuration.subdirectory = mediaproxy.mediaproxy_subdirectory
process.runtime.subdirectory = mediaproxy.mediaproxy_subdirectory
- default_pid = process.runtime.file('{}.pid'.format(name))
+ parser = ArgumentParser(usage='%(prog)s [options]')
+ parser.add_argument('--version', action='version', version='%(prog)s {}'.format(mediaproxy.__version__))
+ parser.add_argument('--systemd', action='store_true', help='run as a systemd simple service and log to journal')
+ parser.add_argument('--no-fork', action='store_false', dest='fork', help='run in the foreground and log to the terminal')
+ parser.add_argument('--config-dir', dest='config_directory', default=None, help='the configuration directory ({})'.format(process.configuration.system_directory), metavar='PATH')
+ parser.add_argument('--runtime-dir', dest='runtime_directory', default=None, help='the runtime directory ({})'.format(process.runtime.directory), metavar='PATH')
+ parser.add_argument('--debug', action='store_true', help='enable verbose logging')
+ parser.add_argument('--debug-memory', action='store_true', help='enable memory debugging')
- parser = OptionParser(version="%%prog %s" % mediaproxy.__version__)
- parser.add_option("--no-fork", action="store_false", dest="fork", default=1, help="run the process in the foreground (for debugging)")
- parser.add_option("--pid", dest="pid_file", default=default_pid, help="pid file (%s)" % default_pid, metavar="File")
- (options, args) = parser.parse_args()
+ options = parser.parse_args()
log.Formatter.prefix_format = '{record.levelname:<8s} '
+ if options.config_directory is not None:
+ process.configuration.local_directory = options.config_directory
+ if options.runtime_directory is not None:
+ process.runtime.directory = options.runtime_directory
+
if not sys.platform.startswith('linux'):
log.critical('Cannot start %s. A Linux host is required for operation.' % fullname)
sys.exit(1)
try:
subprocess.call(['modprobe', 'ip_tables'], env={'PATH': '/usr/sbin:/sbin:/usr/bin:/bin'})
- except OSError, e:
+ except OSError as e:
log.critical('Cannot start %s: failed to load the ip_tables kernel module: %s' % (fullname, e))
sys.exit(1)
try:
kernel_version = Version.parse(open(KERNEL_VERSION_FILE).read().strip())
except (OSError, IOError, ValueError):
log.critical('Could not determine Linux kernel version')
sys.exit(1)
if kernel_version < Version(2, 6, 18):
log.critical('Linux kernel version 2.6.18 or newer is required to run the media relay')
sys.exit(1)
try:
ip_forward = bool(int(open(IP_FORWARD_FILE).read()))
except (OSError, IOError, ValueError):
ip_forward = False
if not ip_forward:
log.critical('IP forwarding is not available or not enabled (check %s)' % IP_FORWARD_FILE)
sys.exit(1)
try:
with open(CONNTRACK_ACCT_FILE, 'w') as acct_file:
acct_file.write('1')
- except (IOError, OSError), e:
+ except (IOError, OSError) as e:
if e.errno != errno.ENOENT:
log.critical('Could not enable conntrack rule counters (check %s): %s' % (CONNTRACK_ACCT_FILE, e))
sys.exit(1)
config_file = ConfigFile(mediaproxy.configuration_file)
log.level.current = config_file.get_setting('Relay', 'log_level', type=datatypes.LogLevel, default=log.level.DEBUG)
- if options.fork:
+ if options.systemd:
+ from systemd.journal import JournalHandler
+ log.set_handler(JournalHandler(SYSLOG_IDENTIFIER=name))
+ log.capture_output()
+ elif options.fork:
try:
- process.daemonize(options.pid_file)
- except ProcessError, e:
+ process.daemonize(pidfile='{}.pid'.format(name))
+ except ProcessError as e:
log.critical('Cannot start %s: %s' % (fullname, e))
sys.exit(1)
log.use_syslog(name)
log.info('Starting %s %s' % (fullname, mediaproxy.__version__))
try:
from mediaproxy.relay import MediaRelay
- if not options.fork:
+ if options.debug:
+ log.level.current = log.level.DEBUG
+ if options.debug_memory:
from application.debug.memory import memory_dump
relay = MediaRelay()
- except Exception, e:
+ except Exception as e:
log.critical('Failed to create %s: %s' % (fullname, e))
- if e.__class__ is not RuntimeError:
+ if type(e) is not RuntimeError:
log.exception()
sys.exit(1)
relay.run()
- if not options.fork:
- #from application.debug.memory import memory_dump
+ if options.debug_memory:
memory_dump()

File Metadata

Mime Type
text/x-diff
Expires
Sat, Nov 23, 7:13 AM (1 d, 14 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3408310
Default Alt Text
(9 KB)

Event Timeline