Added profile support for tagging manga.

[automanga.git] / manga / profile.py

diff --git a/manga/profile.py b/manga/profile.py
index 2ff6be5..85e47b3 100644 (file)
--- a/manga/profile.py
+++ b/manga/profile.py
@@ -60,6 +60,13 @@ def splitline(line):
         ret.append(buf)
     return ret
 
+def splitlines(fp):
+    for line in fp:
+        cur = splitline(line)
+        if len(cur) < 1:
+            continue
+        yield cur
+
 def consline(*words):
     buf = ""
     for w in words:
@@ -79,42 +86,99 @@ def consline(*words):
     return buf
 
 class manga(object):
-    def __init__(self, profile, libnm, id, path):
+    def __init__(self, profile, libnm, id):
         self.profile = profile
         self.libnm = libnm
         self.id = id
-        self.path = path
         self.props = self.loadprops()
 
+    def open(self):
+        import lib
+        return lib.findlib(self.libnm).byid(self.id)
+
+    def save(self):
+        pass
+
+class memmanga(manga):
+    def __init__(self, profile, libnm, id):
+        super(memmanga, self).__init__(profile, libnm, id)
+
+    def loadprops(self):
+        return {}
+
+class tagview(object):
+    def __init__(self, manga):
+        self.manga = manga
+        self.profile = manga.profile
+
+    def add(self, *tags):
+        mt = self.getall(self.profile)
+        ctags = mt.setdefault((self.manga.libnm, self.manga.id), set())
+        ctags |= set(tags)
+        self.save(self.profile, mt)
+
+    def remove(self, *tags):
+        mt = self.getall(self.profile)
+        ctags = mt.get((self.manga.libnm, self.manga.id), set())
+        ctags -= set(tags)
+        if len(ctags) < 1:
+            try:
+                del mt[self.manga.libnm, self.manga.id]
+            except KeyError:
+                pass
+        self.save(self.profile, mt)
+
+    def __iter__(self):
+        return iter(self.getall(self.profile).get((self.manga.libnm, self.manga.id), set()))
+
+    @staticmethod
+    def getall(profile):
+        ret = {}
+        try:
+            with profile.file("tags") as fp:
+                for words in splitlines(fp):
+                    libnm, id = words[0:2]
+                    tags = set(words[2:])
+                    ret[libnm, id] = tags
+        except IOError:
+            pass
+        return ret
+
+    @staticmethod
+    def save(profile, m):
+        with profile.file("tags", "w") as fp:
+            for (libnm, id), tags in m.iteritems():
+                fp.write(consline(libnm, id, *tags) + "\n")
+
+    @staticmethod
+    def bytag(profile, tag):
+        try:
+            with profile.file("tags") as fp:
+                for words in splitlines(fp):
+                    libnm, id = words[0:2]
+                    tags = words[2:]
+                    if tag in tags:
+                        yield profile.getmanga(libnm, id)
+        except IOError:
+            pass
+
+class filemanga(manga):
+    def __init__(self, profile, libnm, id, path):
+        self.path = path
+        super(filemanga, self).__init__(profile, libnm, id)
+        self.tags = tagview(self)
+
     def loadprops(self):
         ret = {}
         with openwdir(self.path) as f:
-            for line in f:
-                words = splitline(line)
-                if len(words) < 1: continue
+            for words in splitlines(f):
                 if words[0] == "set" and len(words) > 2:
                     ret[words[1]] = words[2]
                 elif words[0] == "lset" and len(words) > 1:
                     ret[words[1]] = words[2:]
         return ret
 
-    def prop(self, key, default=KeyError):
-        if key not in self.props:
-            if default is KeyError:
-                raise KeyError(key)
-            return default
-        return self.props[key]
-
-    def __getitem__(self, key):
-        return self.props[key]
-
-    def __contains__(self, key):
-        return key in self.props
-
-    def setprop(self, key, val):
-        self.props[key] = val
-
-    def saveprops(self):
+    def save(self):
         with openwdir(self.path, "w") as f:
             for key, val in self.props.iteritems():
                 if isinstance(val, str):
@@ -122,10 +186,6 @@ class manga(object):
                 else:
                     f.write(consline("lset", key, *val) + "\n")
 
-    def open(self):
-        import lib
-        return lib.findlib(self.libnm).byid(self.id)
-
 class profile(object):
     def __init__(self, dir):
         self.dir = dir
@@ -136,10 +196,7 @@ class profile(object):
         ret = {}
         if os.path.exists(pj(self.dir, "map")):
             with openwdir(pj(self.dir, "map")) as f:
-                for ln in f:
-                    words = splitline(ln)
-                    if len(words) < 1:
-                        continue
+                for words in splitlines(f):
                     if words[0] == "seq" and len(words) > 1:
                         try:
                             seq = int(words[1])
@@ -161,7 +218,7 @@ class profile(object):
     def getmanga(self, libnm, id, creat=False):
         seq, m = self.getmapping()
         if (libnm, id) in m:
-            return manga(self, libnm, id, pj(self.dir, "%i.manga" % m[(libnm, id)]))
+            return filemanga(self, libnm, id, pj(self.dir, "%i.manga" % m[(libnm, id)]))
         if not creat:
             raise KeyError("no such manga: (%s, %s)" % (libnm, id))
         while True:
@@ -174,7 +231,7 @@ class profile(object):
         fp.close()
         m[(libnm, id)] = seq
         self.savemapping(seq, m)
-        return manga(self, libnm, id, pj(self.dir, "%i.manga" % seq))
+        return filemanga(self, libnm, id, pj(self.dir, "%i.manga" % seq))
 
     def setlast(self):
         if self.name is None:
@@ -198,6 +255,9 @@ class profile(object):
             for nm, (libnm, id) in map.iteritems():
                 f.write(consline("alias", nm, libnm, id) + "\n")
 
+    def file(self, name, mode="r"):
+        return openwdir(pj(self.dir, name), mode)
+
     def getalias(self, nm):
         return self.getaliases()[nm]
 
@@ -206,6 +266,9 @@ class profile(object):
         aliases[nm] = libnm, id
         self.savealiases(aliases)
 
+    def bytag(self, tag):
+        return tagview.bytag(self, tag)
+
     @classmethod
     def byname(cls, name):
         if not name or name == "last" or name[0] == '.':