Automated merge with ssh://hg@kagami.tsukasa.net.au/bg_scripts/

[bg-scripts] / randombg.py

diff --git a/randombg.py b/randombg.py
index a259e1e0d4c251cf1e844ec70f93c701466acfac..0bc4b7973610495f8ac665bc82c11a4d21b72406 100755 (executable)
--- a/randombg.py
+++ b/randombg.py
@@ -40,6 +40,7 @@ class BaseFileList(object):
        """Base file list implementation"""
        def __init__(self):
                self.paths = []
+               self.favourites = []
 
        def add_path(self, path):
                self.paths.append(path)
@@ -48,7 +49,7 @@ class BaseFileList(object):
                try:
                        logging.debug("Attempting to store cache")
                        fd = open(filename, 'wb')
-                       pickle.dump(obj = self, file = fd, protocol = 2)
+                       pickle.dump(self, fd, 2)
                        logging.debug("Cache successfully stored")
                except Exception, e:
                        warning("Storing cache: %s" % e)
@@ -77,6 +78,10 @@ class BaseFileList(object):
                        logging.warning("Loading cache: %s" % e)
                        return False
 
+       def add_to_favourites(self):
+               '''Adds the current image to the list of favourites'''
+               self.favourites.append(self.get_current_image())
+
        def scan_paths(self):
                raise NotImplementedError()
 
@@ -166,7 +171,7 @@ class AllRandomFileList(BaseFileList):
                        if self.paths == tmp.paths:
                                logging.debug("Path lists match, copying properties")
                                # Overwrite this object with the other
-                               for attr in ('list', 'imagePointer'):
+                               for attr in ('list', 'imagePointer', 'favourites'):
                                        setattr(self, attr, getattr(tmp, attr))
                        else:
                                logging.debug("Ignoring cache, path lists do not match")
@@ -315,6 +320,9 @@ class Cycler(object):
        def cmd_exit(self):
                asyncsched.exit()
 
+       def cmd_favourite(self):
+               self.filelist.add_to_favourites()
+
 class Server(asynchat.async_chat):
        def __init__(self, cycler, conn, addr):
                asynchat.async_chat.__init__(self, conn=conn)