X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/305485e53a72c278e2b24b754f68df2e4dd71f40..d66f9960d12d739c44a89d1d4e013f52783097e7:/src/references/models.py?ds=sidebyside

diff --git a/src/references/models.py b/src/references/models.py
new file mode 100644
index 000000000..70493d347
--- /dev/null
+++ b/src/references/models.py
@@ -0,0 +1,62 @@
+import json
+from django.db import models
+from wikidata.client import Client
+
+
+class Entity(models.Model):
+    WIKIDATA_PREFIX = 'https://www.wikidata.org/wiki/'
+    WIKIDATA_IMAGE = 'P18'
+    WIKIDATA_COORDINATE_LOCATION = 'P625'
+    WIKIDATA_EARTH = 'Q2'
+
+    uri = models.CharField(max_length=255, unique=True, db_index=True)
+    label = models.CharField(max_length=1024, blank=True)
+    description = models.CharField(max_length=2048, blank=True)
+    wikipedia_link = models.CharField(max_length=1024, blank=True)
+    lat = models.FloatField(null=True, blank=True)
+    lon = models.FloatField(null=True, blank=True)
+    images = models.TextField(blank=True)
+
+    @property
+    def is_interesting(self):
+        return self.wikipedia_link or (self.lat and self.lon) or len(self.images)>2
+    
+    def populate(self):
+        if self.uri.startswith(self.WIKIDATA_PREFIX):
+            self.populate_from_wikidata(
+                self.uri[len(self.WIKIDATA_PREFIX):]
+            )
+
+    def populate_from_wikidata(self, wikidata_id):
+        client = Client()
+        entity = client.get(wikidata_id)
+
+        self.label = entity.label.get('pl', entity.label) or ''
+        self.description = entity.description.get('pl', entity.description) or ''
+        sitelinks = entity.attributes.get('sitelinks', {})
+        self.wikipedia_link = sitelinks.get('plwiki', sitelinks.get('enwiki', {})).get('url') or ''
+
+        location_prop = client.get(self.WIKIDATA_COORDINATE_LOCATION)
+        location = entity.get(location_prop)
+        if location and location.globe.id == self.WIKIDATA_EARTH:
+            self.lat = location.latitude
+            self.lon = location.longitude
+
+        images = entity.getlist(client.get(self.WIKIDATA_IMAGE))
+        self.images = json.dumps([
+            {
+                "url": image.image_url,
+                "page": image.page_url,
+                "resolution": image.image_resolution,
+            } for image in images
+        ])
+
+
+class Reference(models.Model):
+    book = models.ForeignKey('catalogue.Book', models.CASCADE)
+    entity = models.ForeignKey(Entity, models.CASCADE)
+    first_section = models.CharField(max_length=255)
+
+    class Meta:
+        unique_together = (('book', 'entity'),)
+