daemon: simplify get_fd_or_server

This commit is contained in:
ThomasV 2019-08-15 09:58:23 +02:00
parent a9239bd40f
commit b81feb6550
2 changed files with 10 additions and 7 deletions

View file

@ -62,7 +62,7 @@ def remove_lockfile(lockfile):
os.unlink(lockfile)
def get_fd_or_server(config: SimpleConfig):
def get_file_descriptor(config: SimpleConfig):
'''Tries to create the lockfile, using O_EXCL to
prevent races. If it succeeds it returns the FD.
Otherwise try and connect to the server specified in the lockfile.
@ -71,12 +71,12 @@ def get_fd_or_server(config: SimpleConfig):
lockfile = get_lockfile(config)
while True:
try:
return os.open(lockfile, os.O_CREAT | os.O_EXCL | os.O_WRONLY, 0o644), None
return os.open(lockfile, os.O_CREAT | os.O_EXCL | os.O_WRONLY, 0o644)
except OSError:
pass
server = get_server(config)
if server is not None:
return None, server
return
# Couldn't connect; remove lockfile and try again.
remove_lockfile(lockfile)
@ -170,8 +170,9 @@ class Daemon(DaemonThread):
DaemonThread.__init__(self)
self.config = config
if fd is None and listen_jsonrpc:
fd, server = get_fd_or_server(config)
if fd is None: raise Exception('failed to lock daemon; already running?')
fd = get_file_descriptor(config)
if fd is None:
raise Exception('failed to lock daemon; already running?')
self.asyncio_loop, self._stop_loop, self._loop_thread = create_and_start_event_loop()
if config.get('offline'):
self.network = None

View file

@ -357,13 +357,14 @@ if __name__ == '__main__':
if cmdname == 'gui':
configure_logging(config)
fd, server = daemon.get_fd_or_server(config)
fd = daemon.get_file_descriptor(config)
if fd is not None:
plugins = init_plugins(config, config.get('gui', 'qt'))
d = daemon.Daemon(config, fd)
d.init_gui(config, plugins)
sys.exit(0)
else:
server = daemon.get_server()
result = server.gui(config_options)
elif cmdname == 'daemon':
@ -373,7 +374,7 @@ if __name__ == '__main__':
if subcommand in [None, 'start']:
configure_logging(config)
fd, server = daemon.get_fd_or_server(config)
fd = daemon.get_file_descriptor(config)
if fd is not None:
if subcommand == 'start':
pid = os.fork()
@ -404,6 +405,7 @@ if __name__ == '__main__':
d.join()
sys.exit(0)
else:
server = daemon.get_server(config)
result = server.daemon(config_options)
else:
server = daemon.get_server(config)