add status manager

tunasync.py

@@ -2,79 +2,8 @@
 # -*- coding:utf-8 -*-
 import os
 import argparse
-import json
-from datetime import datetime
 
 from tunasync import TUNASync
-from tunasync.hook import JobHook
-
-
-class IndexPageHook(JobHook):
-
-    def __init__(self, parent, dbfile):
-        self.parent = parent
-        self.dbfile = dbfile
-
-    @property
-    def mirrors(self):
-        mirrors = {}
-        try:
-            with open(self.dbfile) as f:
-                _mirrors = json.load(f)
-                for m in _mirrors:
-                    mirrors[m["name"]] = m
-        except:
-            for name, _ in self.parent.mirrors.iteritems():
-                mirrors[name] = {
-                    'name': name,
-                    'last_update': '-',
-                    'status': 'unknown',
-                }
-        return mirrors
-
-    def before_job(self, name=None, *args, **kwargs):
-        if name is None:
-            return
-        mirrors = self.mirrors
-        _m = mirrors.get(name, {
-            'name': name,
-            'last_update': '-',
-            'status': '-',
-        })
-
-        mirrors[name] = {
-            'name': name,
-            'last_update': _m['last_update'],
-            'status': 'syncing'
-        }
-        with open(self.dbfile, 'wb') as f:
-            _mirrors = sorted(
-                [m for _, m in mirrors.items()],
-                key=lambda x: x['name']
-            )
-
-            json.dump(_mirrors, f)
-
-    def after_job(self, name=None, status="unknown", *args, **kwargs):
-        if name is None:
-            return
-
-        print("Updating tunasync.json")
-        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        mirrors = self.mirrors
-        mirrors[name] = {
-            'name': name,
-            'last_update': now,
-            'status': status
-        }
-        with open(self.dbfile, 'wb') as f:
-
-            _mirrors = sorted(
-                [m for _, m in mirrors.items()],
-                key=lambda x: x['name']
-            )
-
-            json.dump(_mirrors, f)
 
 
 if __name__ == "__main__":
@@ -83,9 +12,6 @@ if __name__ == "__main__":
     parser = argparse.ArgumentParser(prog="tunasync")
     parser.add_argument("-c", "--config",
                         default="tunasync.ini", help="config file")
-    parser.add_argument("--dbfile",
-                        default="tunasync.json",
-                        help="mirror status db file")
     parser.add_argument("--pidfile", default="/var/run/tunasync.pid",
                         help="pidfile")
 
@@ -97,10 +23,6 @@ if __name__ == "__main__":
     tunaSync = TUNASync()
     tunaSync.read_config(args.config)
 
-    index_hook = IndexPageHook(tunaSync, args.dbfile)
-
-    tunaSync.add_hook(index_hook)
-
     tunaSync.run_jobs()
 
 # vim: ts=4 sw=4 sts=4 expandtab