X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/2caa6415139fe58938a62dca695639ba0cc86dda..24ae804e7a88a17dcefd0e671eb08e548de14ae8:/src/catalogue/wikidata.py
diff --git a/src/catalogue/wikidata.py b/src/catalogue/wikidata.py
index c6885705..b5f3e94f 100644
--- a/src/catalogue/wikidata.py
+++ b/src/catalogue/wikidata.py
@@ -1,4 +1,8 @@
+# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
from datetime import date
+from django.conf import settings
from django.db import models
from django.db.models.signals import m2m_changed
from django.utils.html import format_html
@@ -6,71 +10,112 @@ from django.utils.translation import gettext_lazy as _
from django.dispatch import receiver
from wikidata.client import Client
from wikidata.datavalue import DatavalueError
+from modeltranslation.translator import translator
+from modeltranslation.settings import AVAILABLE_LANGUAGES
-class WikidataMixin(models.Model):
+class WikidataModel(models.Model):
wikidata = models.CharField(
max_length=255,
- null=True,
blank=True,
- unique=True,
help_text=_('If you have a Wikidata ID, like "Q1337", enter it and save.'),
)
class Meta:
abstract = True
+
+ def wikidata_populate_field(self, client, entity, attname, wd, save, lang):
+ model_field = self._meta.get_field(attname)
+ if isinstance(model_field, models.ManyToManyField):
+ if getattr(self, attname).all().exists():
+ return
+ else:
+ if getattr(self, attname):
+ return
+
+ wdvalue = None
+ if wd == "description":
+ wdvalue = entity.description.get(lang, str(entity.description))
+ elif wd == "label":
+ wdvalue = entity.label.get(lang, str(entity.label))
+ else:
+ try:
+ # TODO: lang?
+ wdvalue = entity.get(client.get(wd))
+ except DatavalueError:
+ pass
+
+ self.set_field_from_wikidata(attname, wdvalue, save=save)
+
+ def wikidata_populate(self, save=True):
+ Wikidata = type(self).Wikidata
+ client = Client()
+ # Probably should getlist
+ entity = client.get(self.wikidata)
+ for attname in dir(Wikidata):
+ if attname.startswith("_"):
+ continue
+ wd = getattr(Wikidata, attname)
+
+ self.wikidata_populate_attribute(client, entity, attname, wd, save=save)
+ if hasattr(Wikidata, '_supplement'):
+ for attname, wd in Wikidata._supplement(self):
+ self.wikidata_populate_attribute(client, entity, attname, wd, save=save)
+
+ def wikidata_fields_for_attribute(self, attname):
+ field = getattr(type(self), attname)
+ if type(self) in translator._registry:
+ opts = translator.get_options_for_model(type(self))
+ if attname in opts.fields:
+ tfields = opts.fields[attname]
+ for tf in tfields:
+ yield tf.name, tf.language
+ return
+ yield attname, settings.LANGUAGE_CODE
+
+ def wikidata_populate_attribute(self, client, entity, attname, wd, save):
+ for fieldname, lang in self.wikidata_fields_for_attribute(attname):
+ self.wikidata_populate_field(client, entity, fieldname, wd, save, lang)
+
def save(self, **kwargs):
+ am_new = self.pk is None
+
super().save()
- if self.wikidata and hasattr(self, "Wikidata"):
- Wikidata = type(self).Wikidata
- client = Client()
- # Probably should getlist
- entity = client.get(self.wikidata)
- for attname in dir(Wikidata):
- if attname.startswith("_"):
- continue
- wd = getattr(Wikidata, attname)
-
- model_field = self._meta.get_field(attname)
- if isinstance(model_field, models.ManyToManyField):
- if getattr(self, attname).all().exists():
- continue
- else:
- if getattr(self, attname):
- continue
-
- wdvalue = None
- if wd == "description":
- wdvalue = entity.description.get("pl", str(entity.description))
- elif wd == "label":
- wdvalue = entity.label.get("pl", str(entity.label))
- else:
- try:
- wdvalue = entity.get(client.get(wd))
- except DatavalueError:
- pass
-
- self.set_field_from_wikidata(attname, wdvalue)
+ if am_new and self.wikidata and hasattr(self, "Wikidata"):
+ self.wikidata_populate()
kwargs.update(force_insert=False, force_update=True)
super().save(**kwargs)
- def set_field_from_wikidata(self, attname, wdvalue):
+ def set_field_from_wikidata(self, attname, wdvalue, save, language='pl'):
if not wdvalue:
return
# Find out what this model field is
model_field = self._meta.get_field(attname)
if isinstance(model_field, models.ForeignKey):
rel_model = model_field.related_model
- if issubclass(rel_model, WikidataMixin):
- # welp, we can try and find by WD identifier.
- wdvalue, created = rel_model.objects.get_or_create(wikidata=wdvalue.id)
+ if issubclass(rel_model, WikidataModel):
+ label = wdvalue.label.get(language, str(wdvalue.label))
+ try:
+ wdvalue = rel_model.objects.get(wikidata=wdvalue.id)
+ except rel_model.DoesNotExist:
+ wdvalue = rel_model(wikidata=wdvalue.id)
+ if save:
+ wdvalue.save()
+ wdvalue._wikidata_label = label
setattr(self, attname, wdvalue)
elif isinstance(model_field, models.ManyToManyField):
rel_model = model_field.related_model
- if issubclass(rel_model, WikidataMixin):
- wdvalue, created = rel_model.objects.get_or_create(wikidata=wdvalue.id)
+ if issubclass(rel_model, WikidataModel):
+ label = wdvalue.label.get(language, str(wdvalue.label))
+ try:
+ wdvalue = rel_model.objects.get(wikidata=wdvalue.id)
+ except rel_model.DoesNotExist:
+ wdvalue = rel_model(wikidata=wdvalue.id)
+ if save:
+ wdvalue.save()
+ wdvalue._wikidata_label = label
getattr(self, attname).set([wdvalue])
else:
# How to get original title?
@@ -78,18 +123,26 @@ class WikidataMixin(models.Model):
if isinstance(model_field, models.IntegerField):
wdvalue = wdvalue.year
elif not isinstance(wdvalue, str):
- wdvalue = wdvalue.label.get("pl", str(wdvalue.label))
+ wdvalue = wdvalue.label.get(language, str(wdvalue.label))
setattr(self, attname, wdvalue)
+ def wikidata_link(self):
+ if self.wikidata:
+ return format_html(
+ '{wd}',
+ wd=self.wikidata,
+ )
+ else:
+ return ""
+
+ wikidata_link.admin_order_field = "wikidata"
+
class WikidataAdminMixin:
+ class Media:
+ css = {"screen": ("catalogue/wikidata_admin.css",)}
+ js = ("catalogue/wikidata_admin.js",)
+
def save_related(self, request, form, formsets, change):
super().save_related(request, form, formsets, change)
form.instance.save()
-
- def wikidata_link(self, obj):
- if obj.wikidata:
- return format_html('{wd}', wd=obj.wikidata)
- else:
- return ''
- wikidata_link.admin_order_field = 'wikidata'