src/librarian/meta/types/wluri.py

   1 import re
   2 from .base import MetaValue
   3
   4
   5 class WLURI(MetaValue):
   6     """Represents a WL URI. Extracts slug from it."""
   7     slug = None
   8
   9     example = 'http://wolnelektury.pl/katalog/lektura/template/'
  10     _re_wl_uri = re.compile(
  11         r'http://(www\.)?wolnelektury.pl/katalog/lektur[ay]/'
  12         '(?P<slug>[-a-z0-9]+)/?$'
  13     )
  14
  15     def __init__(self, slug, uri=None):
  16         """Contructs an URI from slug.
  17
  18         >>> print(WLURI.from_slug('a-slug').uri)
  19         http://wolnelektury.pl/katalog/lektura/a-slug/
  20
  21         """
  22         if uri is None:
  23             uri = 'http://wolnelektury.pl/katalog/lektura/%s/' % slug
  24         self.uri = uri
  25         return super().__init__(slug)
  26
  27
  28     @classmethod
  29     def from_text(cls, uri):
  30         slug = uri.rstrip('/').rsplit('/', 1)[-1]
  31         return cls(slug, uri)
  32
  33     def validate(self):
  34         match = self._re_wl_uri.match(self.uri)
  35         if not match:
  36             raise ValidationError('Invalid URI (%s). Should match: %s' % (
  37                         self.uri, self._re_wl_uri.pattern))
  38         return cls(uri)
  39
  40     def __str__(self):
  41         return self.uri
  42
  43     def __eq__(self, other):
  44         return self.slug == other.slug