From 7eb9967524a4f0d8ebd0c5fab4f1482a2a550a4e Mon Sep 17 00:00:00 2001
From: oysteini <oysteini@pvv.ntnu.no>
Date: Sat, 8 Oct 2011 12:22:40 +0000
Subject: [PATCH] =?UTF-8?q?Fikset=20litt=20p=C3=A5=20visning=20av=20b?=
 =?UTF-8?q?=C3=B8ker.=20=20F=C3=A5r=20med=20personer=20knyttet=20til=20bok?=
 =?UTF-8?q?en.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cli/worblehat.py | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/cli/worblehat.py b/cli/worblehat.py
index 8ee8d69..3d2cbf6 100644
--- a/cli/worblehat.py
+++ b/cli/worblehat.py
@@ -22,6 +22,11 @@ q_list_books = \
     'FROM book ' \
     'LEFT JOIN bookperson ON book.isbn=bookperson.book ' \
     'LEFT JOIN person ON bookperson.person=person.id'
+q_persons_for_book = \
+    'SELECT person.id, lastname, firstname, relation ' \
+    'FROM person ' \
+    'INNER JOIN bookperson ON person.id=bookperson.person ' \
+    'WHERE bookperson.book=%(isbn)s'
 
 def connect_to_db():
     connection = pgdb.connect(database='oysteini_pbb2',
@@ -45,6 +50,20 @@ def get_by_id(connection, id):
             return d
     return None
 
+def list_of_dicts_to_dict(lst, key_name, value_name):
+    res = {}
+    for d in lst:
+        if d[key_name] in res:
+            res[d[key_name]].append(d[value_name])
+        else:
+            res[d[key_name]] = [d[value_name]]
+    return res
+
+def get_persons_for_book(connection, isbn):
+    c = connection.cursor()
+    c.execute(q_persons_for_book, {'isbn': isbn})
+    return fetchall_dict(c)
+
 def show_book(book):
     s = ''
     if book['id']:
@@ -56,7 +75,11 @@ def show_book(book):
     	s += 'Subtitle: %s\n' % book['subtitle']
     s += 'ISBN: %s\n' % book['isbn']
     s += 'Persons:\n'
-    # TODO
+    for bp in book['persons_data']:
+        s += ' %s %s %s (%s)\n' % (bp['id'], bp['firstname'], bp['lastname'],
+                                   bp['relation'])
+    if len(book['persons_data']) == 0:
+        s += ' (no persons associated with this book)\n'
     if book['series']:
         s += 'Part of series: %s %s\n' % (book['series'], book['series_title'])
     s += 'Category: %s\n' % book['category']
@@ -83,16 +106,22 @@ def show_category(category):
 def show(connection, ids, commit_format=False, tmp_file=False):
     objects = map(lambda id: get_by_id(connection, id), ids)
     for i in range(len(ids)):
+        typ = objects[i]['type']
+        if typ == 'book':
+            persons = get_persons_for_book(connection, objects[i]['isbn'])
+            objects[i]['persons'] = list_of_dicts_to_dict(persons, 'relation', 'id')
+            objects[i]['persons_data'] = persons
         if not objects[i]:
             objects[i] = 'No object with id %s.\n' % ids[i]
         elif commit_format:
-            objects[i]['action'] = 'edit-%s' % objects[i]['type']
+            objects[i]['action'] = 'edit-%s' % typ
         else:
             show_funs = {'book': show_book,
                          'person': show_person,
                          'category': show_category}
             show_fun = show_funs[objects[i]['type']]
             objects[i] = show_fun(objects[i])
+
     if commit_format:
         output = write_actionlist(objects)
     else: