From 149903ae7da0968ef3b1b0dc0f329afeaa5a7007 Mon Sep 17 00:00:00 2001
From: Radek Czajka <rczajka@rczajka.pl>
Date: Mon, 5 Apr 2021 15:03:23 +0200
Subject: [PATCH] Fetch thumbnails from wikidata.

---
 src/catalogue/models/book.py                  |  4 +++
 .../templates/catalogue/book_text.html        |  2 +-
 src/references/models.py                      | 28 ++++++++++++++++---
 .../static/js/book_text/references.js         |  6 +++-
 .../static/scss/book_text/references.scss     |  6 +++-
 5 files changed, 39 insertions(+), 7 deletions(-)

diff --git a/src/catalogue/models/book.py b/src/catalogue/models/book.py
index 5a0f1696f..978bca4f2 100644
--- a/src/catalogue/models/book.py
+++ b/src/catalogue/models/book.py
@@ -677,6 +677,10 @@ class Book(models.Model):
                 entity.save()
         Reference.objects.filter(book=self).exclude(entity__uri__in=found).delete()
     
+    @property
+    def references(self):
+        return self.reference_set.all().select_related('entity')
+
     @classmethod
     @transaction.atomic
     def repopulate_ancestors(cls):
diff --git a/src/catalogue/templates/catalogue/book_text.html b/src/catalogue/templates/catalogue/book_text.html
index 27601d87d..e10bc9689 100644
--- a/src/catalogue/templates/catalogue/book_text.html
+++ b/src/catalogue/templates/catalogue/book_text.html
@@ -182,7 +182,7 @@
   {% localize off %}
   <script type="application/json" id="interesting-references">
    {
-       {% for ref in book.reference_set.all %}
+       {% for ref in book.references %}
        {% if ref.entity.is_interesting %}
        "{{ ref.entity.uri }}": {
            {% if ref.entity.lat and ref.entity.lon %}
diff --git a/src/references/models.py b/src/references/models.py
index 70493d347..c98647191 100644
--- a/src/references/models.py
+++ b/src/references/models.py
@@ -1,4 +1,5 @@
 import json
+import urllib.parse
 from django.db import models
 from wikidata.client import Client
 
@@ -8,6 +9,7 @@ class Entity(models.Model):
     WIKIDATA_IMAGE = 'P18'
     WIKIDATA_COORDINATE_LOCATION = 'P625'
     WIKIDATA_EARTH = 'Q2'
+    WIKIDATA_IMAGE_QUERY = './w/api.php?action=query&titles={}&format=json&prop=imageinfo&iiprop=url&iiurlwidth=240&iiurlheight=200'
 
     uri = models.CharField(max_length=255, unique=True, db_index=True)
     label = models.CharField(max_length=1024, blank=True)
@@ -43,13 +45,31 @@ class Entity(models.Model):
             self.lon = location.longitude
 
         images = entity.getlist(client.get(self.WIKIDATA_IMAGE))
-        self.images = json.dumps([
-            {
+        image_data_list = []
+        for image in images:
+            image_data = {
                 "url": image.image_url,
                 "page": image.page_url,
                 "resolution": image.image_resolution,
-            } for image in images
-        ])
+            }
+
+            result = client.request(
+                self.WIKIDATA_IMAGE_QUERY.format(
+                    urllib.parse.quote(image.title)
+                )
+            )
+
+            result_data = next(iter(result['query']['pages'].values()))['imageinfo'][0]
+            image_data['thumburl'] = result_data['thumburl']
+            image_data['thumbresolution'] = [
+                result_data['thumbwidth'],
+                result_data['thumbheight']
+            ]
+            image_data['responsiveUrls'] = result_data['responsiveUrls']
+
+            image_data_list.append(image_data)
+
+        self.images = json.dumps(image_data_list)
 
 
 class Reference(models.Model):
diff --git a/src/wolnelektury/static/js/book_text/references.js b/src/wolnelektury/static/js/book_text/references.js
index 4c72fe60b..b75a4daf3 100644
--- a/src/wolnelektury/static/js/book_text/references.js
+++ b/src/wolnelektury/static/js/book_text/references.js
@@ -78,7 +78,11 @@
             $.each(ref.images, function(i, e) {
                 $i = $("<a target='_blank'><img></a>");
                 $i.attr('href', e.page);
-                $('img', $i).attr('src', e.url);
+                $('img', $i).attr('src', e.thumburl || e.url);
+                if (e.thumbresolution) {
+                    $('img', $i).attr('width', e.thumbresolution[0]).attr('height', e.thumbresolution[1]);
+                }
+
                 $("#reference-images").append($i);
             })
         }
diff --git a/src/wolnelektury/static/scss/book_text/references.scss b/src/wolnelektury/static/scss/book_text/references.scss
index e188025af..3ef469a45 100644
--- a/src/wolnelektury/static/scss/book_text/references.scss
+++ b/src/wolnelektury/static/scss/book_text/references.scss
@@ -34,7 +34,11 @@ a.reference.interesting::after {
             vertical-align: middle;
             margin: 0 10px 0 0;
             img {
-                height: 100px;
+		width: auto;
+		border: 5px solid white;
+		box-shadow: 4px 4px 4px #666;
+		max-width: 240px;
+		max-height: 200px;
             }
         }
     }
-- 
2.20.1