X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/f5a171520d0105e9316afdd42dfee295f09e57ed..9b31d2bc42b0e1239609cd92c532dfd345e48d0f:/src/references/models.py

diff --git a/src/references/models.py b/src/references/models.py
index e8813a016..fb249e1c8 100644
--- 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):