Notifications

2011-04-09 20:18:51 +02:00
parent 288aa8c0a1
commit 85fc1c01ee
67 changed files with 3270 additions and 400 deletions
@@ -0,0 +1,68 @@
+from couchpotato import get_session
+from couchpotato.core.event import addEvent
+from couchpotato.core.helpers.encoding import toUnicode
+from couchpotato.core.logger import CPLog
+from couchpotato.core.settings.model import Status
+
+log = CPLog(__name__)
+
+class StatusPlugin:
+
+    statuses = {
+        'active': 'Active',
+        'done': 'Done',
+        'downloaded': 'Downloaded',
+        'wanted': 'Wanted',
+        'deleted': 'Deleted',
+    }
+
+    def __init__(self):
+        addEvent('status.add', self.add)
+        addEvent('status.all', self.all)
+        addEvent('app.load', self.fill)
+
+    def all(self):
+
+        db = get_session()
+
+        statuses = db.query(Status).all()
+
+        temp = []
+        for status in statuses:
+            s = status.to_dict()
+            temp.append(s)
+
+        return temp
+
+    def add(self, identifier):
+
+        db = get_session()
+
+        s = db.query(Status).filter_by(identifier = identifier).first()
+        if not s:
+            s = Status(
+                identifier = identifier,
+                label = identifier.capitalize()
+            )
+            db.add(s)
+            db.commit()
+
+        #db.remove()
+        return s
+
+    def fill(self):
+
+        db = get_session()
+
+        for identifier, label in self.statuses.iteritems():
+            s = db.query(Status).filter_by(identifier = identifier).first()
+            if not s:
+                log.info('Creating status: %s' % label)
+                s = Status(
+                    identifier = identifier,
+                    label = toUnicode(label)
+                )
+                db.add(s)
+
+            s.label = label
+            db.commit()