]> code.delx.au - bg-scripts/commitdiff
Fixed the oneshot mode
authorGreg Darke <greg@tsukasa.net.au>
Fri, 28 Nov 2008 02:22:05 +0000 (13:22 +1100)
committerGreg Darke <greg@tsukasa.net.au>
Fri, 28 Nov 2008 02:22:05 +0000 (13:22 +1100)
randombg.py

index 360c0966746d38730c15ad081fdf0a892e1cf547..fff408f3b6d6cacf6fe77487986d5dc1551294b7 100755 (executable)
@@ -245,7 +245,7 @@ class FolderRandomFileList(BaseFileList):
 
 
 class Cycler(object):
-       def init(self, options, paths):
+       def init(self, options, paths, oneshot=False):
                self.cycle_time = options.cycle_time
                self.history_filename = options.history_filename
 
@@ -274,7 +274,10 @@ class Cycler(object):
                        sys.exit(1)
        
                self.task = None
-               self.cmd_reload()
+               if oneshot:
+                       self.cmd_next()
+               else:
+                       self.cmd_reload()
        
        def finish(self):
                self.filelist.store_cache(self.history_filename)
@@ -397,7 +400,7 @@ def do_client(options, args):
 
 def do_oneshot(options, paths):
        cycler = Cycler()
-       cycler.init(options, paths)
+       cycler.init(options, paths, oneshot=True)
 
 def build_parser():
        parser = OptionParser(version="%prog " + VERSION, 
@@ -450,11 +453,11 @@ def main():
        
        if options.oneshot:
                do_oneshot(options, args)
-
-       if os.path.exists(options.socket_filename):
-               do_client(options, args)
        else:
-               do_server(options, args)
+               if os.path.exists(options.socket_filename):
+                       do_client(options, args)
+               else:
+                       do_server(options, args)
 
 
 if __name__ == "__main__":