Fix wikidata language fallback

[wolnelektury.git] / src / references / models.py

diff --git a/src/references/models.py b/src/references/models.py
index e8813a0..fb249e1 100644 (file)
--- a/src/references/models.py
+++ b/src/references/models.py
@@ -36,8 +36,8 @@ class Entity(models.Model):
         )]
         entity = client.get(wikidata_id)
 
-        self.label = entity.label.get('pl', entity.label) or ''
-        self.description = entity.description.get('pl', entity.description) or ''
+        self.label = entity.label.get('pl', entity.label.get('mul', entity.label)) or ''
+        self.description = entity.description.get('pl', entity.description.get('mul', entity.description)) or ''
         sitelinks = entity.attributes.get('sitelinks', {})
         self.wikipedia_link = sitelinks.get('plwiki', sitelinks.get('enwiki', {})).get('url') or ''
 
@@ -107,7 +107,18 @@ class Entity(models.Model):
             *to_minutes(lon),
             lonm,
         )
-        
+
+    def get_json(self):
+        data = {
+            "id": self.id,
+            "images": self.get_images(),
+            "label": self.label,
+            "description": self.description,
+            "wikipedia_link": self.wikipedia_link,
+        }
+        if self.lat and self.lon:
+            data['location'] = [self.lat, self.lon]
+        return json.dumps(data)
     
 
 class Reference(models.Model):