Mercurial > odfcast > odfcast
annotate odfcast/convert.py @ 60:ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
author | Björn Ricks <bjoern.ricks@intevation.de> |
---|---|
date | Wed, 12 Nov 2014 12:40:31 +0100 |
parents | 2dacbe7fb7f9 |
children | d8fc52aceb9c |
rev | line source |
---|---|
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
1 # -*- coding: utf-8 -*- |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
2 |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
3 import logging |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
4 import tempfile |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
5 |
16
f4920ab1d27c
Add basic html interface for convert
Björn Ricks <bjoern.ricks@intevation.de>
parents:
4
diff
changeset
|
6 from flask import request, Response, json, render_template |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
7 from flask.views import MethodView |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
8 |
4
46f31348fe01
Implement a TemplateConvertView to allow convert odt templates with variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
3
diff
changeset
|
9 from py3o.template import Template |
46f31348fe01
Implement a TemplateConvertView to allow convert odt templates with variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
3
diff
changeset
|
10 |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
11 from PyPDF2 import PdfFileMerger |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
12 |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
13 from werkzeug.utils import escape |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
14 |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
15 log = logging.getLogger(__name__) |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
16 |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
17 ALLOWED_FORMATS = ["pdf", "doc", "docx", "odt"] |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
18 |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
19 PDF_MIMETYPE = "application/pdf" |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
20 JSON_MIMETYPE = "application/json" |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
21 HTML_MIMETYPE = "text/html" |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
22 |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
23 MIMETYPES = { |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
24 "odt": "application/vnd.oasis.opendocument.text", |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
25 "doc": "application/msword", |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
26 "docx": "application/vnd.openxmlformats-officedocument" |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
27 ".wordprocessingml.document", |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
28 "pdf": PDF_MIMETYPE, |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
29 } |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
30 |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
31 DEFAULT_MIMETYPE = "application/octet-stream" |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
32 |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
33 |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
34 class ErrorResponse(Response): |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
35 |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
36 BAD_REQUEST_ERROR_CODE = 400 |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
37 |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
38 def __init__(self, title, error_code, details, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
39 html_error_code=BAD_REQUEST_ERROR_CODE): |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
40 data, mime_type = self.get_response_data(title, error_code, details) |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
41 super(ErrorResponse, self).__init__(response=data, mimetype=mime_type, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
42 status=html_error_code) |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
43 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
44 def json(self, title, error_code, details): |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
45 return json.dumps({ |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
46 "error": title, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
47 "error_code": error_code, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
48 "details": details, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
49 }), JSON_MIMETYPE |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
50 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
51 def html(self, title, error_code, details): |
59
2dacbe7fb7f9
Fix returning html formatted error response
Björn Ricks <bjoern.ricks@intevation.de>
parents:
58
diff
changeset
|
52 data = ( |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
53 u'<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 3.2 Final//EN">\n' |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
54 u'<title>%(code)s %(name)s</title>\n' |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
55 u'<h1>%(name)s</h1>\n' |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
56 u'%(details)s\n' |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
57 ) % { |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
58 "code": error_code, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
59 "name": escape(title), |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
60 "details": escape(details), |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
61 } |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
62 return data, HTML_MIMETYPE |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
63 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
64 def get_response_data(self, title, error_code, details): |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
65 if self.is_wants_json(): |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
66 return self.json(title, error_code, details) |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
67 return self.html(title, error_code, details) |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
68 |
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
69 def is_wants_json(self): |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
70 best = request.accept_mimetypes.best_match([JSON_MIMETYPE, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
71 HTML_MIMETYPE]) |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
72 return best == JSON_MIMETYPE and \ |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
73 request.accept_mimetypes[best] > \ |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
74 request.accept_mimetypes[HTML_MIMETYPE] |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
75 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
76 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
77 class TemplateErrorResponse(ErrorResponse): |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
78 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
79 TEMPLATE_ERROR_CODE = 100 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
80 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
81 def __init__(self, details, error_code=TEMPLATE_ERROR_CODE): |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
82 super(TemplateErrorResponse, self).__init__( |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
83 title="TemplateError", error_code=error_code, details=details, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
84 html_error_code=500) |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
85 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
86 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
87 class ConversionErrorResponse(ErrorResponse): |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
88 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
89 CONVERSION_ERROR_CODE = 200 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
90 |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
91 def __init__(self, details, error_code=CONVERSION_ERROR_CODE): |
54
64f3e3a28bd4
Fix calling super in ConversionErrorResponse class
Björn Ricks <bjoern.ricks@intevation.de>
parents:
47
diff
changeset
|
92 super(ConversionErrorResponse, self).__init__( |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
93 title="ConversionError", error_code=error_code, details=details, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
94 html_error_code=500) |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
95 |
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
96 |
60
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
97 class MergeErrorResponse(ErrorResponse): |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
98 |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
99 MERGE_ERROR_CODE = 300 |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
100 |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
101 def __init__(self, details, error_code=MERGE_ERROR_CODE): |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
102 super(MergeErrorResponse, self).__init__( |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
103 title="MergeError", error_code=error_code, details=details, |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
104 html_error_code=500) |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
105 |
ae377f7eed18
Add ErrorResponse class for resposes after an error has occured during the merge
Björn Ricks <bjoern.ricks@intevation.de>
parents:
59
diff
changeset
|
106 |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
107 class ConvertView(MethodView): |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
108 |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
109 def __init__(self, pyuno_driver_name="", hostname="localhost", port=2001): |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
110 driver_module = self._load_driver_module(pyuno_driver_name) |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
111 self.convertor = driver_module.Convertor(hostname, port) |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
112 |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
113 def _load_driver_module(self, pyuno_driver_name): |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
114 return __import__(pyuno_driver_name, globals(), locals(), |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
115 ["Convertor"]) |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
116 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
117 def is_format_supported(self, fformat): |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
118 return fformat and fformat.lower() in ALLOWED_FORMATS |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
119 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
120 def post(self): |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
121 ffile = request.files['file'] |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
122 if not ffile.filename: |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
123 return ErrorResponse( |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
124 "Upload file missing", error_code=101, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
125 details="Please upload a file for conversion", |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
126 html_error_code=400) |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
127 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
128 fformat = request.form['format'] |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
129 if not self.is_format_supported(fformat): |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
130 return ErrorResponse( |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
131 "Invalid format", error_code=102, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
132 details="Format %s not allowed" % fformat, |
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
133 html_error_code=400) |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
134 |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
135 datadict = self.get_datadict() |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
136 |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
137 mimetype = self.get_mimetype_for_format(fformat) |
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
138 |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
139 outfile = self.save_form_file(ffile) |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
140 |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
141 if datadict: |
31
83bca7dc9bfe
Return error response when a template couldn't be rendered
Björn Ricks <bjoern.ricks@intevation.de>
parents:
30
diff
changeset
|
142 try: |
83bca7dc9bfe
Return error response when a template couldn't be rendered
Björn Ricks <bjoern.ricks@intevation.de>
parents:
30
diff
changeset
|
143 tfile = tempfile.NamedTemporaryFile() |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
144 t = Template(outfile, tfile, ignore_undefined_variables=True) |
31
83bca7dc9bfe
Return error response when a template couldn't be rendered
Björn Ricks <bjoern.ricks@intevation.de>
parents:
30
diff
changeset
|
145 t.render(datadict) |
83bca7dc9bfe
Return error response when a template couldn't be rendered
Björn Ricks <bjoern.ricks@intevation.de>
parents:
30
diff
changeset
|
146 outfile.close() |
83bca7dc9bfe
Return error response when a template couldn't be rendered
Björn Ricks <bjoern.ricks@intevation.de>
parents:
30
diff
changeset
|
147 outfile = tfile |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
148 except Exception, e: |
41
a1100ec32be2
Log exception on rendering errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
34
diff
changeset
|
149 log.exception("Template error") |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
150 return TemplateErrorResponse(details=str(e)) |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
151 |
32
e6f2f1481de2
Fix we must check the fformat variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
31
diff
changeset
|
152 if fformat != "odt": |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
153 try: |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
154 outfile = self.convert(outfile, fformat) |
42
6d511e93a331
Return requested format when an error has occured
Björn Ricks <bjoern.ricks@intevation.de>
parents:
41
diff
changeset
|
155 except Exception, e: |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
156 log.exception("Conversion error") |
47
e99cbb47eafb
Improve error handling by returning flask/werkzeug responses directly on errors
Björn Ricks <bjoern.ricks@intevation.de>
parents:
42
diff
changeset
|
157 return ConversionErrorResponse(details=str(e)) |
29
1dadc59c4b9a
Make the conversion more error prone
Björn Ricks <bjoern.ricks@intevation.de>
parents:
22
diff
changeset
|
158 |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
159 return Response(outfile, mimetype=mimetype) |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
160 |
16
f4920ab1d27c
Add basic html interface for convert
Björn Ricks <bjoern.ricks@intevation.de>
parents:
4
diff
changeset
|
161 def get(self): |
f4920ab1d27c
Add basic html interface for convert
Björn Ricks <bjoern.ricks@intevation.de>
parents:
4
diff
changeset
|
162 return render_template("convert.html") |
f4920ab1d27c
Add basic html interface for convert
Björn Ricks <bjoern.ricks@intevation.de>
parents:
4
diff
changeset
|
163 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
164 def save_form_file(self, infile): |
33
00e12392f8d4
Fix saving uploaded form file to a temporary file
Björn Ricks <bjoern.ricks@intevation.de>
parents:
32
diff
changeset
|
165 outfile = tempfile.NamedTemporaryFile() |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
166 infile.save(outfile) |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
167 infile.close() |
33
00e12392f8d4
Fix saving uploaded form file to a temporary file
Björn Ricks <bjoern.ricks@intevation.de>
parents:
32
diff
changeset
|
168 outfile.seek(0) |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
169 return outfile |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
170 |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
171 def convert(self, infile, fformat): |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
172 outfile = tempfile.NamedTemporaryFile() |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
173 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
174 self.convertor.convert(infile.name, outfile.name, fformat) |
0
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
175 |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
176 infile.close() |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
177 return outfile |
4a34f72f036b
Add initial conversion service
Björn Ricks <bjoern.ricks@intevation.de>
parents:
diff
changeset
|
178 |
3
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
179 def get_mimetype_for_format(self, fformat): |
15807d87930c
Refactor ConvertView to be easier reusable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
0
diff
changeset
|
180 return MIMETYPES.get(fformat, DEFAULT_MIMETYPE) |
4
46f31348fe01
Implement a TemplateConvertView to allow convert odt templates with variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
3
diff
changeset
|
181 |
46f31348fe01
Implement a TemplateConvertView to allow convert odt templates with variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
3
diff
changeset
|
182 def get_datadict(self): |
34
89ffa12bd48d
Allow to not passing any datadict to the conversion form
Björn Ricks <bjoern.ricks@intevation.de>
parents:
33
diff
changeset
|
183 vars = request.form.get('datadict') |
30
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
184 if not vars: |
9d65de2ebe22
Merge TemplateConvertView into ConvertView
Björn Ricks <bjoern.ricks@intevation.de>
parents:
29
diff
changeset
|
185 return None |
4
46f31348fe01
Implement a TemplateConvertView to allow convert odt templates with variable
Björn Ricks <bjoern.ricks@intevation.de>
parents:
3
diff
changeset
|
186 return json.loads(vars) |
18
585904e7411a
Add html for converting a odf with a template
Björn Ricks <bjoern.ricks@intevation.de>
parents:
16
diff
changeset
|
187 |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
188 |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
189 class MergeView(MethodView): |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
190 |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
191 def get(self): |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
192 return render_template("merge.html") |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
193 |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
194 def post(self): |
57
423f99732a80
Add some debug output for merging pdfs
Björn Ricks <bjoern.ricks@intevation.de>
parents:
54
diff
changeset
|
195 log.debug("Merging PDF documents") |
423f99732a80
Add some debug output for merging pdfs
Björn Ricks <bjoern.ricks@intevation.de>
parents:
54
diff
changeset
|
196 |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
197 merger = PdfFileMerger() |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
198 ffiles = request.files.getlist('files') |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
199 |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
200 for ffile in ffiles: |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
201 merger.append(ffile) |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
202 |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
203 outfile = tempfile.NamedTemporaryFile() |
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
204 |
58
ea839127934d
Avoid creating another file descriptor for the merged output file
Björn Ricks <bjoern.ricks@intevation.de>
parents:
57
diff
changeset
|
205 merger.write(outfile) |
ea839127934d
Avoid creating another file descriptor for the merged output file
Björn Ricks <bjoern.ricks@intevation.de>
parents:
57
diff
changeset
|
206 merger.close() |
ea839127934d
Avoid creating another file descriptor for the merged output file
Björn Ricks <bjoern.ricks@intevation.de>
parents:
57
diff
changeset
|
207 outfile.seek(0) |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
208 |
57
423f99732a80
Add some debug output for merging pdfs
Björn Ricks <bjoern.ricks@intevation.de>
parents:
54
diff
changeset
|
209 log.debug("PDF documents merged") |
19
2f627039d2b4
Add a MergeView to merge pdfs based on PyPDF2
Björn Ricks <bjoern.ricks@intevation.de>
parents:
18
diff
changeset
|
210 return Response(outfile, mimetype=PDF_MIMETYPE) |
22
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
211 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
212 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
213 class TemplateView(MethodView): |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
214 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
215 template_name = "" |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
216 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
217 def __init__(self, template_name=None): |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
218 if template_name: |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
219 self.template_name = template_name |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
220 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
221 def get_template_name(self): |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
222 return self.template_name |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
223 |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
224 def get(self): |
0e40926735e1
Add a generic View to display a template for a get request
Björn Ricks <bjoern.ricks@intevation.de>
parents:
19
diff
changeset
|
225 return render_template(self.get_template_name()) |