Import INP(X)

[m_librarian.git] / m_librarian / db.py

diff --git a/m_librarian/db.py b/m_librarian/db.py
index 98e541e95732378344f2e6f47b6b5f2cb1af2f3c..ef0e3bd350d52b0b96b8fdae3b5f3832d72708d2 100755 (executable)
--- 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: