X-Git-Url: https://code.delx.au/bg-scripts/blobdiff_plain/e82e47afd1d108241367327cad3f7637cd9b5d3c..0736e53c6d6e9f192c8cbf68602f4a18c0264501:/randombg.py

diff --git a/randombg.py b/randombg.py
index 9a587bb..9ff4642 100755
--- a/randombg.py
+++ b/randombg.py
@@ -1,10 +1,15 @@
 #!/usr/bin/env python
 
-VERSION = "2.0"
+VERSION = "2.1"
 
 
-import asyncore, asynchat, socket
-import os, os.path, random, sys, time
+import asyncore
+import asynchat
+import socket
+import os
+import random
+import sys
+import time
 from optparse import OptionParser
 import logging
 try:
@@ -231,7 +236,7 @@ class FolderRandomFileList(BaseFileList):
 class Cycler(object):
 	def init(self, options, paths, oneshot=False):
 		self.cycle_time = options.cycle_time
-		self.history_filename = options.history_filename
+		self.cache_filename = options.cache_filename
 
 		logging.debug("Initialising wallchanger")
 		wallchanger.init(options.background_colour, options.permanent, options.convert)
@@ -247,7 +252,7 @@ class Cycler(object):
 		for path in paths:
 			self.filelist.add_path(path)
 
-		if self.filelist.load_cache(self.history_filename):
+		if self.filelist.load_cache(self.cache_filename):
 			logging.debug("Loaded cache successfully")
 		else:
 			logging.debug("Could not load cache")
@@ -264,7 +269,7 @@ class Cycler(object):
 			self.cmd_reload()
 	
 	def finish(self):
-		self.filelist.store_cache(self.history_filename)
+		self.filelist.store_cache(self.cache_filename)
 
 	def find_files(self, options, paths):
 		return filelist
@@ -280,7 +285,7 @@ class Cycler(object):
 			self.task.cancel()
 		self.task = asyncsched.schedule(self.cycle_time, next)
 		logging.debug("Reset timer for %s seconds" % self.cycle_time)
-		self.filelist.store_cache(self.history_filename)
+		self.filelist.store_cache(self.cache_filename)
 	
 	def cmd_reload(self):
 		image = self.filelist.get_current_image()
@@ -361,6 +366,19 @@ class Listener(asyncore.dispatcher):
 		
 
 def do_server(options, paths):
+	try:
+		sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+		sock.connect(options.socket_filename)
+		print >>sys.stderr, "Server is already running! Will do nothing"
+		return
+	except:
+		pass
+
+	try:
+		os.unlink(options.socket_filename)
+	except OSError:
+		pass
+
 	try:
 		cycler = Cycler()
 		listener = Listener(options.socket_filename, cycler)
@@ -375,7 +393,7 @@ def do_server(options, paths):
 		# Make sure that the socket is cleaned up
 		try:
 			os.unlink(options.socket_filename)
-		except:
+		except OSError:
 			pass
 
 def do_client(options, args):
@@ -429,9 +447,12 @@ def build_parser():
 	parser.add_option("--socket",
 		action="store", type="string", dest="socket_filename", default=os.path.expanduser('~/.randombg_socket'),
 		help="Location of the command/control socket.")
-	parser.add_option("--history-file",
-		action="store", type="string", dest="history_filename", default=os.path.expanduser('~/.randombg_historyfile'),
+	parser.add_option("--cache-file",
+		action="store", type="string", dest="cache_filename", default=os.path.expanduser('~/.randombg_cache'),
 		help="Stores the location of the last image to be loaded.")
+	parser.add_option("--server",
+		action="store_true", dest="server", default=False,
+		help="Run in server mode to listen for clients.")
 	return parser
 
 def main():
@@ -443,13 +464,19 @@ def main():
 	elif options.verbose >= 2:
 		logging.getLogger().setLevel(logging.DEBUG)
 	
+	if options.server:
+		do_server(options, args)
+		return
+
 	if options.oneshot:
 		do_oneshot(options, args)
-	else:
-		if os.path.exists(options.socket_filename):
-			do_client(options, args)
-		else:
-			do_server(options, args)
+		return
+
+	try:
+		do_client(options, args)
+		return
+	except Exception, e:
+		print >>sys.stderr, "Failed to connect to server:", e
 
 
 if __name__ == "__main__":