apps/catalogue/utils.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 import random
   6 import time
   7 from base64 import urlsafe_b64encode
   8
   9 from django.core.files.uploadedfile import UploadedFile
  10 from django.utils.hashcompat import sha_constructor
  11 from django.conf import settings
  12
  13 from librarian import DocProvider
  14
  15
  16 # Use the system (hardware-based) random number generator if it exists.
  17 if hasattr(random, 'SystemRandom'):
  18     randrange = random.SystemRandom().randrange
  19 else:
  20     randrange = random.randrange
  21 MAX_SESSION_KEY = 18446744073709551616L     # 2 << 63
  22
  23
  24 def get_random_hash(seed):
  25     sha_digest = sha_constructor('%s%s%s%s' %
  26         (randrange(0, MAX_SESSION_KEY), time.time(), unicode(seed).encode('utf-8', 'replace'),
  27         settings.SECRET_KEY)).digest()
  28     return urlsafe_b64encode(sha_digest).replace('=', '').replace('_', '-').lower()
  29
  30
  31 def split_tags(tags):
  32     result = {}
  33     for tag in tags:
  34         result.setdefault(tag.category, []).append(tag)
  35     return result
  36
  37
  38 class ExistingFile(UploadedFile):
  39
  40     def __init__(self, path, *args, **kwargs):
  41         self.path = path
  42         return super(ExistingFile, self).__init__(*args, **kwargs)
  43
  44     def temporary_file_path(self):
  45         return self.path
  46
  47     def close(self):
  48         pass
  49
  50
  51 class BookImportDocProvider(DocProvider):
  52     """Used for joined EPUB and PDF files."""
  53
  54     def __init__(self, book):
  55         self.book = book
  56
  57     def by_slug(self, slug):
  58         if slug == self.book.slug:
  59             return self.book.xml_file
  60         else:
  61             return type(self.book).objects.get(slug=slug).xml_file