src/bookmarks/models.py

   1 import uuid
   2 from django.db import models
   3
   4
   5 class Bookmark(models.Model):
   6     uuid = models.UUIDField(unique=True, default=uuid.uuid4, editable=False)
   7     user = models.ForeignKey('auth.User', models.CASCADE)
   8     book = models.ForeignKey('catalogue.Book', models.CASCADE)
   9     anchor = models.CharField(max_length=100, blank=True)
  10     created_at = models.DateTimeField(auto_now_add=True)
  11     note = models.TextField(blank=True)
  12
  13     def __str__(self):
  14         return str(self.uuid)
  15
  16     def location(self):
  17         return f'{self.book.slug}/{self.anchor}'
  18
  19     def get_for_json(self):
  20         return {
  21             'uuid': self.uuid,
  22             'anchor': self.anchor,
  23             'note': self.note,
  24             'created_at': self.created_at,
  25         }
  26
  27
  28 class Quote(models.Model):
  29     uuid = models.UUIDField(unique=True, default=uuid.uuid4, editable=False)
  30     user = models.ForeignKey('auth.User', models.CASCADE)
  31     book = models.ForeignKey('catalogue.Book', models.CASCADE)
  32     created_at = models.DateTimeField(auto_now_add=True)
  33     start_elem = models.CharField(max_length=100, blank=True)
  34     end_elem = models.CharField(max_length=100, blank=True)
  35     start_offset = models.IntegerField(null=True, blank=True)
  36     end_offset = models.IntegerField(null=True, blank=True)
  37     text = models.TextField(blank=True)
  38
  39     def __str__(self):
  40         return str(self.uuid)
  41
  42     def get_for_json(self):
  43         return {
  44             'uuid': self.uuid,
  45             'startElem': self.start_elem,
  46             'endElem': self.end_elem,
  47             'startOffset': self.start_offset,
  48             'startOffset': self.end_offset,
  49             'created_at': self.created_at,
  50         }
  51
  52 def from_path(elem, path):
  53     def child_nodes(e):
  54         if e.text: yield (e, 'text')
  55         for child in e:
  56             if child.attrib.get('id') != 'toc':
  57                 yield (child, None)
  58             if child.tail:
  59                 yield (child, 'tail')
  60     while len(path) > 1:
  61         n = path.pop(0)
  62         elem = list(child_nodes(elem))[n]
  63     return elem
  64
  65
  66