schilder.py 10.3 KB
Newer Older
Dave Kliczbor's avatar
Dave Kliczbor committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
#!/usr/bin/python
# -*- encoding: utf8 -*-

from flask import Flask, flash, session, redirect, url_for, escape, request, Response, Markup
import sys, os, os.path, glob
from genshi.template import TemplateLoader
from genshi.template.text import NewTextTemplate
from flaskext.genshi import Genshi, render_response
from werkzeug.utils import secure_filename
from collections import defaultdict
import warnings
import shutil
import subprocess
from subprocess import CalledProcessError, STDOUT
import PythonMagick
import json
import tempfile
import config

app = Flask(__name__)
app.config['UPLOAD_FOLDER'] = config.uploaddir
app.config['PROPAGATE_EXCEPTIONS'] = True
app.secret_key = config.app_secret
genshi = Genshi(app)
genshi.extensions['html'] = 'html5'

def check_output(*popenargs, **kwargs):
    # Copied from py2.7s subprocess module
    r"""Run command with arguments and return its output as a byte string.

    If the exit code was non-zero it raises a CalledProcessError.  The
    CalledProcessError object will have the return code in the returncode
    attribute and output in the output attribute.

    The arguments are the same as for the Popen constructor.  Example:

    >>> check_output(["ls", "-l", "/dev/null"])
    'crw-rw-rw- 1 root root 1, 3 Oct 18  2007 /dev/null\n'

    The stdout argument is not allowed as it is used internally.
    To capture standard error in the result, use stderr=STDOUT.

    >>> check_output(["/bin/sh", "-c",
    ...               "ls -l non_existent_file ; exit 0"],
    ...              stderr=STDOUT)
    'ls: non_existent_file: No such file or directory\n'
    """
    if 'stdout' in kwargs:
        raise ValueError('stdout argument not allowed, it will be overridden.')
    process = subprocess.Popen(stdout=subprocess.PIPE, *popenargs, **kwargs)
    output, unused_err = process.communicate()
    retcode = process.poll()
    if retcode != 0:
        cmd = kwargs.get("args")
        if cmd is None:
            cmd = popenargs[0]
        raise CalledProcessError(retcode, cmd, output=output)
        #raise Exception(output)
    return output

def allowed_file(filename):
  return '.' in filename and filename.rsplit('.', 1)[1] in config.allowed_extensions

@app.route('/')
def index(**kwargs):
  data = defaultdict(str)
  data.update(**kwargs)
  filelist = glob.glob(config.datadir + '/*.schild')
  data['files'] = [ unicode(os.path.basename(f)) for f in sorted(filelist) ]
  return render_response('index.html', data)

@app.route('/edit')
def edit(**kwargs):
  data = defaultdict(str)
  data.update(**kwargs)
  imagelist = glob.glob(config.imagedir + '/*.png')
  data['images'] = [ os.path.basename(f) for f in imagelist ]
  templatelist = glob.glob(config.textemplatedir + '/*.tex')
  data['templates'] = [ unicode(os.path.basename(f)) for f in sorted(templatelist) ]
  return render_response('edit.html', data)

@app.route('/edit/<filename>')
def edit_one(filename):
  with open(os.path.join(config.datadir, filename), 'r') as infile:
    formdata = json.load(infile)
    return edit(form=formdata)

88
def run_pdflatex(context, outputfilename, overwrite=True):
Dave Kliczbor's avatar
Dave Kliczbor committed
89
90
91
92
  if not context.has_key('textemplate'):
    context['textemplate'] = "text-image-quer.tex"
  genshitex = TemplateLoader([config.textemplatedir])
  template = genshitex.load(context['textemplate'], cls=NewTextTemplate, encoding='utf8')
93
94
  if not overwrite and os.path.isfile(outputfilename) and os.path.getmtime(template.filepath) < os.path.getmtime(outputfilename):
      return
Dave Kliczbor's avatar
Dave Kliczbor committed
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
  tmpdir = tempfile.mkdtemp(dir=config.tmpdir)
  if context.has_key('img') and context['img'] and context['img'] != '__none':
    try:
      shutil.copy(os.path.join(config.imagedir, context['img']), os.path.join(tmpdir, context['img']))
    except:
      raise IOError("COULD NOT COPY")
  else:
    #print "MEH No image"
    pass
  tmptexfile = os.path.join(tmpdir, 'output.tex')
  tmppdffile = os.path.join(tmpdir, 'output.pdf')
  with open(tmptexfile, 'w') as texfile:
    texfile.write(template.generate(form = context).render(encoding='utf8'))
  cwd = os.getcwd()
  os.chdir(tmpdir)
  os.symlink(config.texsupportdir, os.path.join(tmpdir, 'support'))
  try:
    texlog = check_output(['pdflatex', '--halt-on-error', tmptexfile], stderr=STDOUT)
  except CalledProcessError as e:
114
115
    if overwrite:
        flash(Markup("<p>PDFLaTeX Output:</p><pre>%s</pre>" % e.output), 'log')
Dave Kliczbor's avatar
Dave Kliczbor committed
116
117
118
    raise SyntaxWarning("PDFLaTeX bailed out")
  finally:
    os.chdir(cwd)
119
120
  if overwrite:
    flash(Markup("<p>PDFLaTeX Output:</p><pre>%s</pre>" % texlog), 'log')
Dave Kliczbor's avatar
Dave Kliczbor committed
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
  shutil.copy(tmppdffile, outputfilename)
  shutil.rmtree(tmpdir)

def save_and_convert_image_upload(inputname):
  file = request.files[inputname]
  if file:
    if not allowed_file(file.filename):
      raise UserWarning("Uploaded image is not in the list of allowed file types.")
    filename = os.path.join(config.uploaddir, secure_filename(file.filename))
    file.save(filename)
    img = PythonMagick.Image(filename)
    imgname = os.path.splitext(secure_filename(file.filename))[0].replace('.', '_') + '.png'
    savedfilename = os.path.join(config.imagedir, imgname)
    img.write(savedfilename)
    os.remove(filename)
    return imgname
  return None

@app.route('/create', methods=['POST'])
def create():
  if request.method == 'POST':
    formdata = request.form.to_dict(flat=True)
    for a in ('headline', 'text'):
      formdata[a] = unicode(formdata[a])
    try:
      imgpath = save_and_convert_image_upload('imgupload')
      if imgpath is not None:
        formdata['img'] = imgpath
Dave Kliczbor's avatar
Dave Kliczbor committed
149
      outfilename = secure_filename(formdata['headline'][:16])+str(hash(formdata['headline']+formdata['text']+os.path.splitext(formdata['textemplate'])[0]+os.path.splitext(formdata['img'])[0]))+'.schild'
Dave Kliczbor's avatar
Dave Kliczbor committed
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
      outpdfname = outfilename + '.pdf'
      formdata['pdfname'] = outpdfname
      run_pdflatex(formdata, os.path.join(config.pdfdir, outpdfname))
      with open(os.path.join(config.datadir, outfilename), 'w') as outfile:
        json.dump(formdata, outfile)
      flash(Markup(u"""PDF created and data saved. You might create another one. Here's a preview. Click to print.<br/>
          <a href="%s"><img src="%s"/></a>""" %
          (url_for('schild', filename=outfilename), url_for('pdfthumbnail', pdfname=outpdfname, maxgeometry=200))
        ))
    except Exception as e:
      flash(u"Could not create pdf or save data: %s" % str(e), 'error')

    data = {'form': formdata }
    imagelist = glob.glob(config.imagedir + '/*.png')
    data['images'] = [ os.path.basename(f) for f in imagelist ]
    templatelist = glob.glob(config.textemplatedir + '/*.tex')
    data['templates'] = [ os.path.basename(f) for f in sorted(templatelist) ]
    return render_response('edit.html', data)
  flash("No POST data. You've been redirected to the edit page.", 'warning')
  return redirect(url_for('edit'))

@app.route('/schild/<filename>')
def schild(filename):
  return render_response('schild.html', {'filename':filename, 'printer':[ unicode(f) for f in sorted(config.printers.keys()) ]})

@app.route('/printout', methods=['POST'])
def printout():
  filename = os.path.join(config.pdfdir, secure_filename(request.form['filename']))
  printer = config.printers[request.form['printer']]
  copies = int(request.form['copies']) or 0
  if copies > 0 and copies <= 6:
    try:
      lprout = check_output(['lpr', '-H', str(config.printserver), '-P', str(printer), '-#', str(copies)] + config.lproptions + [filename], stderr=STDOUT)
      flash(u'Schild wurde zum Drucker geschickt!')
    except CalledProcessError as e:
      flash(Markup("<p>Could not print:</p><pre>%s</pre>" % e.output), 'error')
  else:
    flash(u'Ungültige Anzahl Kopien!')
  return redirect(url_for('index'))

@app.route('/delete', methods=['POST'])
def delete():
  filename = secure_filename(request.form['filename'])
  try:
    os.unlink(os.path.join(config.datadir, filename))
    for f in glob.glob(os.path.join(config.pdfdir, filename + '.pdf*')):
      os.unlink(f)
    flash(u"Schild %s wurde gelöscht" % filename)
    return redirect(url_for('index'))
  except:
    flash(u"Schild %s konnte nicht gelöscht werden." % filename, 'error')
    return redirect(url_for('schild', filename=filename))

@app.route('/image/<imgname>')
def image(imgname):
  imgpath = os.path.join(config.imagedir, secure_filename(imgname))
  #print(imgpath)
  if os.path.exists(imgpath):
    with open(imgpath, 'r') as imgfile:
      return Response(imgfile.read(), mimetype="image/png")
  else:
    return "Meh" #redirect(url_for('index'))

def make_thumb(filename, maxgeometry):
  thumbpath = filename + '.' + str(maxgeometry)
  if not os.path.exists(thumbpath) or os.path.getmtime(filename) > os.path.getmtime(thumbpath):
    img = PythonMagick.Image(str(filename))
    img.transform("%sx%s" % (maxgeometry,maxgeometry))
    img.quality(90)
    img.write(str("png:%s" % thumbpath))
  return thumbpath

@app.route('/thumbnail/<imgname>/<int:maxgeometry>')
def thumbnail(imgname, maxgeometry):
  imgpath = os.path.join(config.imagedir, secure_filename(imgname))
  thumbpath = make_thumb(imgpath, maxgeometry)
  with open(thumbpath, 'r') as imgfile:
    return Response(imgfile.read(), mimetype="image/png")

@app.route('/pdfthumb/<pdfname>/<int:maxgeometry>')
def pdfthumbnail(pdfname, maxgeometry):
  pdfpath = os.path.join(config.pdfdir, secure_filename(pdfname))
  thumbpath = make_thumb(pdfpath, maxgeometry)
  with open(thumbpath, 'r') as imgfile:
    return Response(imgfile.read(), mimetype="image/png")

236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
@app.route('/tplthumb/<tplname>/<int:maxgeometry>')
def tplthumbnail(tplname, maxgeometry):
  pdfpath = os.path.join(config.cachedir, secure_filename(tplname)+'.pdf')
  try:
    run_pdflatex(
      { 'textemplate' : secure_filename(tplname),
        'img'         : 'pictograms-nps-misc-camera.png',
        'headline'    : u'Überschrift',
        'text'        : u'Dies ist der Text, der in der UI als Text bezeichnet ist.',
      }, pdfpath, overwrite=False
    )
  except Exception as e:
    return str(e)
  else:
    thumbpath = make_thumb(pdfpath, maxgeometry)
    with open(thumbpath, 'r') as imgfile:
      return Response(imgfile.read(), mimetype="image/png")

Dave Kliczbor's avatar
Dave Kliczbor committed
254
255
256
257
258
259
260
261
262
@app.route('/pdfdownload/<pdfname>')
def pdfdownload(pdfname):
  pdfpath = os.path.join(config.pdfdir, secure_filename(pdfname))
  with open(pdfpath, 'r') as pdffile:
    return Response(pdffile.read(), mimetype="application/pdf")

if __name__ == '__main__':
  app.debug = True
  app.run(host=config.listen, port=config.port)