src/catalogue/wikimedia.py

   1 from urllib.parse import unquote
   2 from django.core.files.base import ContentFile
   3 from cover.utils import get_wikimedia_data, URLOpener
   4
   5
   6 class WikiMedia:
   7     def get_description_url(imgdata):
   8         if imgdata is None:
   9             return None
  10         return imgdata.attributes['imageinfo'][0]['descriptionurl']
  11
  12     @classmethod
  13     def descriptionurl(cls, arg):
  14         def transform(get_value):
  15             value = get_value(arg)
  16             if value is None:
  17                 return None
  18             return cls.get_description_url(value)
  19         return transform
  20
  21     @classmethod
  22     def attribution(cls, arg):
  23         def transform(get_value):
  24             value = get_value(arg)
  25             if value is None:
  26                 return None
  27             media_data = get_wikimedia_data(
  28                 cls.get_description_url(value)
  29             )
  30             parts = [
  31                 media_data['title'],
  32                 media_data['author'],
  33                 media_data['license_name'],
  34             ]
  35             parts = [p for p in parts if p]
  36             attribution = ', '.join(parts)
  37             return attribution
  38         return transform
  39
  40     @classmethod
  41     def download(cls, arg):
  42         def transform(get_value):
  43             value = get_value(arg)
  44             if value is None:
  45                 return None
  46             media_data = get_wikimedia_data(
  47                 cls.get_description_url(value)
  48             )
  49             download_url = media_data['download_url']
  50             return Downloadable(download_url)
  51         return transform
  52
  53
  54 class Downloadable:
  55     def __init__(self, url):
  56         self.url = url
  57
  58     def apply_to_field(self, obj, attname):
  59         t = URLOpener().open(self.url).read()
  60         getattr(obj, attname).save(
  61             unquote(self.url.rsplit('/', 1)[-1]),
  62             ContentFile(t),
  63             save=False
  64         )
  65
  66     def as_hint_json(self):
  67         return {
  68             'download': self.url,
  69             'img': self.url,
  70         }