X-Git-Url: https://git.phdru.name/?a=blobdiff_plain;f=m_librarian%2Fdb.py;h=ef0e3bd350d52b0b96b8fdae3b5f3832d72708d2;hb=15065e84b53d63bbde0cbda272c18865e4932aee;hp=98e541e95732378344f2e6f47b6b5f2cb1af2f3c;hpb=e94c6883c5c71b05b82db32583cb05dac4091907;p=m_librarian.git

diff --git a/m_librarian/db.py b/m_librarian/db.py
index 98e541e..ef0e3bd 100755
--- a/m_librarian/db.py
+++ b/m_librarian/db.py
@@ -114,6 +114,20 @@ def insert_name(table, name, **kw):
         return table(name=name, count=0, **kw)
 
 
+def update_counters():
+    for author in Author.select():
+        author.count = len(author.books)
+
+    for ext in Extension.select():
+        ext.count = Book.select(Book.q.extension == ext.name).count()
+
+    for genre in Genre.select():
+        genre.count = len(genre.books)
+
+    for language in Language.select():
+        language.count = Book.select(Book.q.language == language.name).count()
+
+
 def test():
     print "DB dirs:", db_dirs
     if db_uri: