e3b6dfd82ad12214b5a40d3b57ccdcacff88c0f8
[redakcja.git] / src / depot / models.py
1 import json
2 import os
3 import tempfile
4 import zipfile
5 from datetime import datetime
6 from django.db import models
7 from librarian.cover import make_cover
8 from librarian.builders import EpubBuilder, MobiBuilder
9
10
11 class Package(models.Model):
12     created_at = models.DateTimeField(auto_now_add=True)
13     placed_at = models.DateTimeField(null=True, blank=True)
14     finished_at = models.DateTimeField(null=True, blank=True)
15     definition_json = models.TextField(blank=True)
16     books = models.ManyToManyField('documents.Book')
17     status_json = models.TextField(blank=True)
18     logo = models.FileField(blank=True, upload_to='depot/logo')
19     file = models.FileField(blank=True, upload_to='depot/package/')
20
21     def save(self, *args, **kwargs):
22         try:
23             self.set_status(self.get_status())
24         except:
25             pass
26         
27         try:
28             self.set_definition(self.get_definition())
29         except:
30             pass
31
32         super().save(*args, **kwargs)
33     
34     def get_status(self):
35         return json.loads(self.status_json)
36
37     def set_status(self, status):
38         self.status_json = json.dumps(status, indent=4, ensure_ascii=False)
39
40     def get_definition(self):
41         return json.loads(self.definition_json)
42
43     def set_definition(self, definition):
44         self.definition_json = json.dumps(definition, indent=4, ensure_ascii=False)
45
46     def build(self):
47         f = tempfile.NamedTemporaryFile(prefix='depot-', suffix='.zip', mode='wb', delete=False)
48         book_count = self.books.all().count()
49         with zipfile.ZipFile(f, 'w') as z:
50             for i, book in enumerate(self.books.all()):
51                 print(f'{i}/{book_count} {book.slug}')
52                 self.build_for(book, z)
53         f.close()
54         with open(f.name, 'rb') as ff:
55             self.file.save('package-{}.zip'.format(datetime.now().isoformat(timespec='seconds')), ff)
56         os.unlink(f.name)
57
58     def build_for(self, book, z):
59         wldoc2 = book.wldocument(librarian2=True)
60         slug = wldoc2.meta.url.slug
61         for item in self.get_definition():
62             wldoc = book.wldocument()
63             wldoc2 = book.wldocument(librarian2=True)
64             base_url = 'file://' + book.gallery_path() + '/'
65
66             ext = item['type']
67
68             if item['type'] == 'cover':
69                 kwargs = {}
70                 if self.logo:
71                     kwargs['cover_logo'] = self.logo.path
72                 for k in 'format', 'width', 'height', 'cover_class':
73                     if k in item:
74                         kwargs[k] = item[k]
75                 cover = make_cover(wldoc.book_info, **kwargs)
76                 output = cover.output_file()
77                 ext = cover.ext()
78
79             elif item['type'] == 'pdf':
80                 cover_kwargs = {}
81                 if 'cover_class' in item:
82                     cover_kwargs['cover_class'] = item['cover_class']
83                 if self.logo:
84                     cover_kwargs['cover_logo'] = self.logo.path
85                 cover = lambda *args, **kwargs: make_cover(*args, **kwargs, **cover_kwargs)
86                 output = wldoc.as_pdf(cover=cover, base_url=base_url)
87
88             elif item['type'] == 'epub':
89                 cover_kwargs = {}
90                 if 'cover_class' in item:
91                     cover_kwargs['cover_class'] = item['cover_class']
92                 if self.logo:
93                     cover_kwargs['cover_logo'] = self.logo.path
94                 cover = lambda *args, **kwargs: make_cover(*args, **kwargs, **cover_kwargs)
95
96                 output = EpubBuilder(
97                     cover=cover,
98                     base_url=base_url,
99                     fundraising=item.get('fundraising', []),
100                 ).build(wldoc2)
101
102             elif item['type'] == 'mobi':
103                 output = MobiBuilder(
104                     cover=cover,
105                     base_url=base_url,
106                     fundraising=item.get('fundraising', []),
107                 ).build(wldoc2)
108
109             fname = f'{slug}/{slug}.'
110             if 'slug' in item:
111                 fname += item['slug'] + '.'
112             fname += ext
113
114             z.writestr(
115                 fname,
116                 output.get_bytes()
117             )