test fix
[wolnelektury.git] / 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
14 # Use the system (hardware-based) random number generator if it exists.
15 if hasattr(random, 'SystemRandom'):
16     randrange = random.SystemRandom().randrange
17 else:
18     randrange = random.randrange
19 MAX_SESSION_KEY = 18446744073709551616L     # 2 << 63
20
21
22 def get_random_hash(seed):
23     sha_digest = sha_constructor('%s%s%s%s' %
24         (randrange(0, MAX_SESSION_KEY), time.time(), unicode(seed).encode('utf-8', 'replace'),
25         settings.SECRET_KEY)).digest()
26     return urlsafe_b64encode(sha_digest).replace('=', '').replace('_', '-').lower()
27
28
29 def split_tags(tags):
30     result = {}
31     for tag in tags:
32         result.setdefault(tag.category, []).append(tag)
33     return result
34
35
36 class ExistingFile(UploadedFile):
37
38     def __init__(self, path, *args, **kwargs):
39         self.path = path
40         return super(ExistingFile, self).__init__(*args, **kwargs)
41
42     def temporary_file_path(self):
43         return self.path
44
45     def close(self):
46         pass