forked from floodlight/loxigen
-
Notifications
You must be signed in to change notification settings - Fork 1
/
tenjin.py
2118 lines (1807 loc) · 77.5 KB
/
tenjin.py
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
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
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
149
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
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
##
## $Release: 1.1.1 $
## $Copyright: copyright(c) 2007-2012 kuwata-lab.com all rights reserved. $
## $License: MIT License $
##
## Permission is hereby granted, free of charge, to any person obtaining
## a copy of this software and associated documentation files (the
## "Software"), to deal in the Software without restriction, including
## without limitation the rights to use, copy, modify, merge, publish,
## distribute, sublicense, and/or sell copies of the Software, and to
## permit persons to whom the Software is furnished to do so, subject to
## the following conditions:
##
## The above copyright notice and this permission notice shall be
## included in all copies or substantial portions of the Software.
##
## THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
## EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
## MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
## NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
## LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
## OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
## WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
##
"""Very fast and light-weight template engine based embedded Python.
See User's Guide and examples for details.
http://www.kuwata-lab.com/tenjin/pytenjin-users-guide.html
http://www.kuwata-lab.com/tenjin/pytenjin-examples.html
"""
__version__ = "$Release: 1.1.1 $"[10:-2]
__license__ = "$License: MIT License $"[10:-2]
__all__ = ('Template', 'Engine', )
import sys, os, re, time, marshal
from time import time as _time
from os.path import getmtime as _getmtime
from os.path import isfile as _isfile
random = pickle = unquote = None # lazy import
python3 = sys.version_info[0] == 3
python2 = sys.version_info[0] == 2
logger = None
##
## utilities
##
def _write_binary_file(filename, content):
global random
if random is None: from random import random
tmpfile = filename + str(random())[1:]
f = open(tmpfile, 'w+b') # on windows, 'w+b' is preffered than 'wb'
try:
f.write(content)
finally:
f.close()
if os.path.exists(tmpfile):
try:
os.rename(tmpfile, filename)
except:
os.remove(filename) # on windows, existing file should be removed before renaming
os.rename(tmpfile, filename)
def _read_binary_file(filename):
f = open(filename, 'rb')
try:
return f.read()
finally:
f.close()
codecs = None # lazy import
def _read_text_file(filename, encoding=None):
global codecs
if not codecs: import codecs
f = codecs.open(filename, encoding=(encoding or 'utf-8'))
try:
return f.read()
finally:
f.close()
def _read_template_file(filename, encoding=None):
s = _read_binary_file(filename) ## binary(=str)
if encoding: s = s.decode(encoding) ## binary(=str) to unicode
return s
_basestring = basestring
_unicode = unicode
_bytes = str
def _ignore_not_found_error(f, default=None):
try:
return f()
except OSError, ex:
if ex.errno == 2: # error: No such file or directory
return default
raise
def create_module(module_name, dummy_func=None, **kwargs):
"""ex. mod = create_module('tenjin.util')"""
try:
mod = type(sys)(module_name)
except:
# The module creation above does not work for Jython 2.5.2
import imp
mod = imp.new_module(module_name)
mod.__file__ = __file__
mod.__dict__.update(kwargs)
sys.modules[module_name] = mod
if dummy_func:
exec(dummy_func.func_code, mod.__dict__)
return mod
def _raise(exception_class, *args):
raise exception_class(*args)
##
## helper method's module
##
def _dummy():
global unquote
unquote = None
global to_str, escape, echo, new_cycle, generate_tostrfunc
global start_capture, stop_capture, capture_as, captured_as, CaptureContext
global _p, _P, _decode_params
def generate_tostrfunc(encode=None, decode=None):
"""Generate 'to_str' function with encode or decode encoding.
ex. generate to_str() function which encodes unicode into binary(=str).
to_str = tenjin.generate_tostrfunc(encode='utf-8')
repr(to_str(u'hoge')) #=> 'hoge' (str)
ex. generate to_str() function which decodes binary(=str) into unicode.
to_str = tenjin.generate_tostrfunc(decode='utf-8')
repr(to_str('hoge')) #=> u'hoge' (unicode)
"""
if encode:
if decode:
raise ValueError("can't specify both encode and decode encoding.")
else:
def to_str(val, _str=str, _unicode=unicode, _isa=isinstance, _encode=encode):
"""Convert val into string or return '' if None. Unicode will be encoded into binary(=str)."""
if _isa(val, _str): return val
if val is None: return ''
#if _isa(val, _unicode): return val.encode(_encode) # unicode to binary(=str)
if _isa(val, _unicode):
return val.encode(_encode) # unicode to binary(=str)
return _str(val)
else:
if decode:
def to_str(val, _str=str, _unicode=unicode, _isa=isinstance, _decode=decode):
"""Convert val into string or return '' if None. Binary(=str) will be decoded into unicode."""
#if _isa(val, _str): return val.decode(_decode) # binary(=str) to unicode
if _isa(val, _str):
return val.decode(_decode)
if val is None: return ''
if _isa(val, _unicode): return val
return _unicode(val)
else:
def to_str(val, _str=str, _unicode=unicode, _isa=isinstance):
"""Convert val into string or return '' if None. Both binary(=str) and unicode will be retruned as-is."""
if _isa(val, _str): return val
if val is None: return ''
if _isa(val, _unicode): return val
return _str(val)
return to_str
to_str = generate_tostrfunc(encode='utf-8') # or encode=None?
def echo(string):
"""add string value into _buf. this is equivarent to '#{string}'."""
lvars = sys._getframe(1).f_locals # local variables
lvars['_buf'].append(string)
def new_cycle(*values):
"""Generate cycle object.
ex.
cycle = new_cycle('odd', 'even')
print(cycle()) #=> 'odd'
print(cycle()) #=> 'even'
print(cycle()) #=> 'odd'
print(cycle()) #=> 'even'
"""
def gen(values):
i, n = 0, len(values)
while True:
yield values[i]
i = (i + 1) % n
return gen(values).next
class CaptureContext(object):
def __init__(self, name, store_to_context=True, lvars=None):
self.name = name
self.store_to_context = store_to_context
self.lvars = lvars or sys._getframe(1).f_locals
def __enter__(self):
lvars = self.lvars
self._buf_orig = lvars['_buf']
lvars['_buf'] = _buf = []
lvars['_extend'] = _buf.extend
return self
def __exit__(self, *args):
lvars = self.lvars
_buf = lvars['_buf']
lvars['_buf'] = self._buf_orig
lvars['_extend'] = self._buf_orig.extend
lvars[self.name] = self.captured = ''.join(_buf)
if self.store_to_context and '_context' in lvars:
lvars['_context'][self.name] = self.captured
def __iter__(self):
self.__enter__()
yield self
self.__exit__()
def start_capture(varname=None, _depth=1):
"""(obsolete) start capturing with name."""
lvars = sys._getframe(_depth).f_locals
capture_context = CaptureContext(varname, None, lvars)
lvars['_capture_context'] = capture_context
capture_context.__enter__()
def stop_capture(store_to_context=True, _depth=1):
"""(obsolete) stop capturing and return the result of capturing.
if store_to_context is True then the result is stored into _context[varname].
"""
lvars = sys._getframe(_depth).f_locals
capture_context = lvars.pop('_capture_context', None)
if not capture_context:
raise Exception('stop_capture(): start_capture() is not called before.')
capture_context.store_to_context = store_to_context
capture_context.__exit__()
return capture_context.captured
def capture_as(name, store_to_context=True):
"""capture partial of template."""
return CaptureContext(name, store_to_context, sys._getframe(1).f_locals)
def captured_as(name, _depth=1):
"""helper method for layout template.
if captured string is found then append it to _buf and return True,
else return False.
"""
lvars = sys._getframe(_depth).f_locals # local variables
if name in lvars:
_buf = lvars['_buf']
_buf.append(lvars[name])
return True
return False
def _p(arg):
"""ex. '/show/'+_p("item['id']") => "/show/#{item['id']}" """
return '<`#%s#`>' % arg # decoded into #{...} by preprocessor
def _P(arg):
"""ex. '<b>%s</b>' % _P("item['id']") => "<b>${item['id']}</b>" """
return '<`$%s$`>' % arg # decoded into ${...} by preprocessor
def _decode_params(s):
"""decode <`#...#`> and <`$...$`> into #{...} and ${...}"""
global unquote
if unquote is None:
from urllib import unquote
dct = { 'lt':'<', 'gt':'>', 'amp':'&', 'quot':'"', '#039':"'", }
def unescape(s):
#return s.replace('<', '<').replace('>', '>').replace('"', '"').replace(''', "'").replace('&', '&')
return re.sub(r'&(lt|gt|quot|amp|#039);', lambda m: dct[m.group(1)], s)
s = to_str(s)
s = re.sub(r'%3C%60%23(.*?)%23%60%3E', lambda m: '#{%s}' % unquote(m.group(1)), s)
s = re.sub(r'%3C%60%24(.*?)%24%60%3E', lambda m: '${%s}' % unquote(m.group(1)), s)
s = re.sub(r'<`#(.*?)#`>', lambda m: '#{%s}' % unescape(m.group(1)), s)
s = re.sub(r'<`\$(.*?)\$`>', lambda m: '${%s}' % unescape(m.group(1)), s)
s = re.sub(r'<`#(.*?)#`>', r'#{\1}', s)
s = re.sub(r'<`\$(.*?)\$`>', r'${\1}', s)
return s
helpers = create_module('tenjin.helpers', _dummy, sys=sys, re=re)
helpers.__all__ = ['to_str', 'escape', 'echo', 'new_cycle', 'generate_tostrfunc',
'start_capture', 'stop_capture', 'capture_as', 'captured_as',
'not_cached', 'echo_cached', 'cache_as',
'_p', '_P', '_decode_params',
]
generate_tostrfunc = helpers.generate_tostrfunc
##
## escaped module
##
def _dummy():
global is_escaped, as_escaped, to_escaped
global Escaped, EscapedStr, EscapedUnicode
global __all__
__all__ = ('is_escaped', 'as_escaped', 'to_escaped', ) #'Escaped', 'EscapedStr',
class Escaped(object):
"""marking class that object is already escaped."""
pass
def is_escaped(value):
"""return True if value is marked as escaped, else return False."""
return isinstance(value, Escaped)
class EscapedStr(str, Escaped):
"""string class which is marked as escaped."""
pass
class EscapedUnicode(unicode, Escaped):
"""unicode class which is marked as escaped."""
pass
def as_escaped(s):
"""mark string as escaped, without escaping."""
if isinstance(s, str): return EscapedStr(s)
if isinstance(s, unicode): return EscapedUnicode(s)
raise TypeError("as_escaped(%r): expected str or unicode." % (s, ))
def to_escaped(value):
"""convert any value into string and escape it.
if value is already marked as escaped, don't escape it."""
if hasattr(value, '__html__'):
value = value.__html__()
if is_escaped(value):
#return value # EscapedUnicode should be convered into EscapedStr
return as_escaped(_helpers.to_str(value))
#if isinstance(value, _basestring):
# return as_escaped(_helpers.escape(value))
return as_escaped(_helpers.escape(_helpers.to_str(value)))
escaped = create_module('tenjin.escaped', _dummy, _helpers=helpers)
##
## module for html
##
def _dummy():
global escape_html, escape_xml, escape, tagattr, tagattrs, _normalize_attrs
global checked, selected, disabled, nl2br, text2html, nv, js_link
#_escape_table = { '&': '&', '<': '<', '>': '>', '"': '"', "'": ''' }
#_escape_pattern = re.compile(r'[&<>"]')
##_escape_callable = lambda m: _escape_table[m.group(0)]
##_escape_callable = lambda m: _escape_table.__get__(m.group(0))
#_escape_get = _escape_table.__getitem__
#_escape_callable = lambda m: _escape_get(m.group(0))
#_escape_sub = _escape_pattern.sub
#def escape_html(s):
# return s # 3.02
#def escape_html(s):
# return _escape_pattern.sub(_escape_callable, s) # 6.31
#def escape_html(s):
# return _escape_sub(_escape_callable, s) # 6.01
#def escape_html(s, _p=_escape_pattern, _f=_escape_callable):
# return _p.sub(_f, s) # 6.27
#def escape_html(s, _sub=_escape_pattern.sub, _callable=_escape_callable):
# return _sub(_callable, s) # 6.04
#def escape_html(s):
# s = s.replace('&', '&')
# s = s.replace('<', '<')
# s = s.replace('>', '>')
# s = s.replace('"', '"')
# return s # 5.83
def escape_html(s):
"""Escape '&', '<', '>', '"' into '&', '<', '>', '"'."""
return s.replace('&', '&').replace('<', '<').replace('>', '>').replace('"', '"').replace("'", ''') # 5.72
escape_xml = escape_html # for backward compatibility
def tagattr(name, expr, value=None, escape=True):
"""(experimental) Return ' name="value"' if expr is true value, else '' (empty string).
If value is not specified, expr is used as value instead."""
if not expr and expr != 0: return _escaped.as_escaped('')
if value is None: value = expr
if escape: value = _escaped.to_escaped(value)
return _escaped.as_escaped(' %s="%s"' % (name, value))
def tagattrs(**kwargs):
"""(experimental) built html tag attribtes.
ex.
>>> tagattrs(klass='main', size=20)
' class="main" size="20"'
>>> tagattrs(klass='', size=0)
''
"""
kwargs = _normalize_attrs(kwargs)
esc = _escaped.to_escaped
s = ''.join([ ' %s="%s"' % (k, esc(v)) for k, v in kwargs.iteritems() if v or v == 0 ])
return _escaped.as_escaped(s)
def _normalize_attrs(kwargs):
if 'klass' in kwargs: kwargs['class'] = kwargs.pop('klass')
if 'checked' in kwargs: kwargs['checked'] = kwargs.pop('checked') and 'checked' or None
if 'selected' in kwargs: kwargs['selected'] = kwargs.pop('selected') and 'selected' or None
if 'disabled' in kwargs: kwargs['disabled'] = kwargs.pop('disabled') and 'disabled' or None
return kwargs
def checked(expr):
"""return ' checked="checked"' if expr is true."""
return _escaped.as_escaped(expr and ' checked="checked"' or '')
def selected(expr):
"""return ' selected="selected"' if expr is true."""
return _escaped.as_escaped(expr and ' selected="selected"' or '')
def disabled(expr):
"""return ' disabled="disabled"' if expr is true."""
return _escaped.as_escaped(expr and ' disabled="disabled"' or '')
def nl2br(text):
"""replace "\n" to "<br />\n" and return it."""
if not text:
return _escaped.as_escaped('')
return _escaped.as_escaped(text.replace('\n', '<br />\n'))
def text2html(text, use_nbsp=True):
"""(experimental) escape xml characters, replace "\n" to "<br />\n", and return it."""
if not text:
return _escaped.as_escaped('')
s = _escaped.to_escaped(text)
if use_nbsp: s = s.replace(' ', ' ')
#return nl2br(s)
s = s.replace('\n', '<br />\n')
return _escaped.as_escaped(s)
def nv(name, value, sep=None, **kwargs):
"""(experimental) Build name and value attributes.
ex.
>>> nv('rank', 'A')
'name="rank" value="A"'
>>> nv('rank', 'A', '.')
'name="rank" value="A" id="rank.A"'
>>> nv('rank', 'A', '.', checked=True)
'name="rank" value="A" id="rank.A" checked="checked"'
>>> nv('rank', 'A', '.', klass='error', style='color:red')
'name="rank" value="A" id="rank.A" class="error" style="color:red"'
"""
name = _escaped.to_escaped(name)
value = _escaped.to_escaped(value)
s = sep and 'name="%s" value="%s" id="%s"' % (name, value, name+sep+value) \
or 'name="%s" value="%s"' % (name, value)
html = kwargs and s + tagattrs(**kwargs) or s
return _escaped.as_escaped(html)
def js_link(label, onclick, **kwargs):
s = kwargs and tagattrs(**kwargs) or ''
html = '<a href="javascript:undefined" onclick="%s;return false"%s>%s</a>' % \
(_escaped.to_escaped(onclick), s, _escaped.to_escaped(label))
return _escaped.as_escaped(html)
html = create_module('tenjin.html', _dummy, helpers=helpers, _escaped=escaped)
helpers.escape = html.escape_html
helpers.html = html # for backward compatibility
sys.modules['tenjin.helpers.html'] = html
##
## utility function to set default encoding of template files
##
_template_encoding = (None, 'utf-8') # encodings for decode and encode
def set_template_encoding(decode=None, encode=None):
"""Set default encoding of template files.
This should be called before importing helper functions.
ex.
## I like template files to be unicode-base like Django.
import tenjin
tenjin.set_template_encoding('utf-8') # should be called before importing helpers
from tenjin.helpers import *
"""
global _template_encoding
if _template_encoding == (decode, encode):
return
if decode and encode:
raise ValueError("set_template_encoding(): cannot specify both decode and encode.")
if not decode and not encode:
raise ValueError("set_template_encoding(): decode or encode should be specified.")
if decode:
Template.encoding = decode # unicode base template
helpers.to_str = helpers.generate_tostrfunc(decode=decode)
else:
Template.encoding = None # binary base template
helpers.to_str = helpers.generate_tostrfunc(encode=encode)
_template_encoding = (decode, encode)
##
## Template class
##
class TemplateSyntaxError(SyntaxError):
def build_error_message(self):
ex = self
if not ex.text:
return self.args[0]
return ''.join([
"%s:%s:%s: %s\n" % (ex.filename, ex.lineno, ex.offset, ex.msg, ),
"%4d: %s\n" % (ex.lineno, ex.text.rstrip(), ),
" %s^\n" % (' ' * ex.offset, ),
])
class Template(object):
"""Convert and evaluate embedded python string.
See User's Guide and examples for details.
http://www.kuwata-lab.com/tenjin/pytenjin-users-guide.html
http://www.kuwata-lab.com/tenjin/pytenjin-examples.html
"""
## default value of attributes
filename = None
encoding = None
escapefunc = 'escape'
tostrfunc = 'to_str'
indent = 4
preamble = None # "_buf = []; _expand = _buf.expand; _to_str = to_str; _escape = escape"
postamble = None # "print ''.join(_buf)"
smarttrim = None
args = None
timestamp = None
trace = False # if True then '<!-- begin: file -->' and '<!-- end: file -->' are printed
def __init__(self, filename=None, encoding=None, input=None, escapefunc=None, tostrfunc=None,
indent=None, preamble=None, postamble=None, smarttrim=None, trace=None):
"""Initailizer of Template class.
filename:str (=None)
Filename to convert (optional). If None, no convert.
encoding:str (=None)
Encoding name. If specified, template string is converted into
unicode object internally.
Template.render() returns str object if encoding is None,
else returns unicode object if encoding name is specified.
input:str (=None)
Input string. In other words, content of template file.
Template file will not be read if this argument is specified.
escapefunc:str (='escape')
Escape function name.
tostrfunc:str (='to_str')
'to_str' function name.
indent:int (=4)
Indent width.
preamble:str or bool (=None)
Preamble string which is inserted into python code.
If true, '_buf = []; ' is used insated.
postamble:str or bool (=None)
Postamble string which is appended to python code.
If true, 'print("".join(_buf))' is used instead.
smarttrim:bool (=None)
If True then "<div>\\n#{_context}\\n</div>" is parsed as
"<div>\\n#{_context}</div>".
"""
if encoding is not None: self.encoding = encoding
if escapefunc is not None: self.escapefunc = escapefunc
if tostrfunc is not None: self.tostrfunc = tostrfunc
if indent is not None: self.indent = indent
if preamble is not None: self.preamble = preamble
if postamble is not None: self.postamble = postamble
if smarttrim is not None: self.smarttrim = smarttrim
if trace is not None: self.trace = trace
#
if preamble is True: self.preamble = "_buf = []"
if postamble is True: self.postamble = "print(''.join(_buf))"
if input:
self.convert(input, filename)
self.timestamp = False # False means 'file not exist' (= Engine should not check timestamp of file)
elif filename:
self.convert_file(filename)
else:
self._reset()
def _reset(self, input=None, filename=None):
self.script = None
self.bytecode = None
self.input = input
self.filename = filename
if input != None:
i = input.find("\n")
if i < 0:
self.newline = "\n" # or None
elif len(input) >= 2 and input[i-1] == "\r":
self.newline = "\r\n"
else:
self.newline = "\n"
self._localvars_assignments_added = False
def _localvars_assignments(self):
return "_extend=_buf.extend;_to_str=%s;_escape=%s; " % (self.tostrfunc, self.escapefunc)
def before_convert(self, buf):
if self.preamble:
eol = self.input.startswith('<?py') and "\n" or "; "
buf.append(self.preamble + eol)
def after_convert(self, buf):
if self.postamble:
if buf and not buf[-1].endswith("\n"):
buf.append("\n")
buf.append(self.postamble + "\n")
def convert_file(self, filename):
"""Convert file into python script and return it.
This is equivarent to convert(open(filename).read(), filename).
"""
input = _read_template_file(filename)
return self.convert(input, filename)
def convert(self, input, filename=None):
"""Convert string in which python code is embedded into python script and return it.
input:str
Input string to convert into python code.
filename:str (=None)
Filename of input. this is optional but recommended to report errors.
"""
if self.encoding and isinstance(input, str):
input = input.decode(self.encoding)
self._reset(input, filename)
buf = []
self.before_convert(buf)
self.parse_stmts(buf, input)
self.after_convert(buf)
script = ''.join(buf)
self.script = script
return script
STMT_PATTERN = (r'<\?py( |\t|\r?\n)(.*?) ?\?>([ \t]*\r?\n)?', re.S)
def stmt_pattern(self):
pat = self.STMT_PATTERN
if isinstance(pat, tuple):
pat = self.__class__.STMT_PATTERN = re.compile(*pat)
return pat
def parse_stmts(self, buf, input):
if not input: return
rexp = self.stmt_pattern()
is_bol = True
index = 0
for m in rexp.finditer(input):
mspace, code, rspace = m.groups()
#mspace, close, rspace = m.groups()
#code = input[m.start()+4+len(mspace):m.end()-len(close)-(rspace and len(rspace) or 0)]
text = input[index:m.start()]
index = m.end()
## detect spaces at beginning of line
lspace = None
if text == '':
if is_bol:
lspace = ''
elif text[-1] == '\n':
lspace = ''
else:
rindex = text.rfind('\n')
if rindex < 0:
if is_bol and text.isspace():
lspace, text = text, ''
else:
s = text[rindex+1:]
if s.isspace():
lspace, text = s, text[:rindex+1]
#is_bol = rspace is not None
## add text, spaces, and statement
self.parse_exprs(buf, text, is_bol)
is_bol = rspace is not None
#if mspace == "\n":
if mspace and mspace.endswith("\n"):
code = "\n" + (code or "")
#if rspace == "\n":
if rspace and rspace.endswith("\n"):
code = (code or "") + "\n"
if code:
code = self.statement_hook(code)
m = self._match_to_args_declaration(code)
if m:
self._add_args_declaration(buf, m)
else:
self.add_stmt(buf, code)
rest = input[index:]
if rest:
self.parse_exprs(buf, rest)
self._arrange_indent(buf)
def statement_hook(self, stmt):
"""expand macros and parse '#@ARGS' in a statement."""
return stmt.replace("\r\n", "\n") # Python can't handle "\r\n" in code
def _match_to_args_declaration(self, stmt):
if self.args is not None:
return None
args_pattern = r'^ *#@ARGS(?:[ \t]+(.*?))?$'
return re.match(args_pattern, stmt)
def _add_args_declaration(self, buf, m):
arr = (m.group(1) or '').split(',')
args = []; declares = []
for s in arr:
arg = s.strip()
if not s: continue
if not re.match('^[a-zA-Z_]\w*$', arg):
raise ValueError("%r: invalid template argument." % arg)
args.append(arg)
declares.append("%s = _context.get('%s'); " % (arg, arg))
self.args = args
#nl = stmt[m.end():]
#if nl: declares.append(nl)
buf.append(''.join(declares) + "\n")
s = '(?:\{.*?\}.*?)*'
EXPR_PATTERN = (r'#\{(.*?'+s+r')\}|\$\{(.*?'+s+r')\}|\{=(?:=(.*?)=|(.*?))=\}', re.S)
del s
def expr_pattern(self):
pat = self.EXPR_PATTERN
if isinstance(pat, tuple):
self.__class__.EXPR_PATTERN = pat = re.compile(*pat)
return pat
def get_expr_and_flags(self, match):
expr1, expr2, expr3, expr4 = match.groups()
if expr1 is not None: return expr1, (False, True) # not escape, call to_str
if expr2 is not None: return expr2, (True, True) # call escape, call to_str
if expr3 is not None: return expr3, (False, True) # not escape, call to_str
if expr4 is not None: return expr4, (True, True) # call escape, call to_str
def parse_exprs(self, buf, input, is_bol=False):
buf2 = []
self._parse_exprs(buf2, input, is_bol)
if buf2:
buf.append(''.join(buf2))
def _parse_exprs(self, buf, input, is_bol=False):
if not input: return
self.start_text_part(buf)
rexp = self.expr_pattern()
smarttrim = self.smarttrim
nl = self.newline
nl_len = len(nl)
pos = 0
for m in rexp.finditer(input):
start = m.start()
text = input[pos:start]
pos = m.end()
expr, flags = self.get_expr_and_flags(m)
#
if text:
self.add_text(buf, text)
self.add_expr(buf, expr, *flags)
#
if smarttrim:
flag_bol = text.endswith(nl) or not text and (start > 0 or is_bol)
if flag_bol and not flags[0] and input[pos:pos+nl_len] == nl:
pos += nl_len
buf.append("\n")
if smarttrim:
if buf and buf[-1] == "\n":
buf.pop()
rest = input[pos:]
if rest:
self.add_text(buf, rest, True)
self.stop_text_part(buf)
if input[-1] == '\n':
buf.append("\n")
def start_text_part(self, buf):
self._add_localvars_assignments_to_text(buf)
#buf.append("_buf.extend((")
buf.append("_extend((")
def _add_localvars_assignments_to_text(self, buf):
if not self._localvars_assignments_added:
self._localvars_assignments_added = True
buf.append(self._localvars_assignments())
def stop_text_part(self, buf):
buf.append("));")
def _quote_text(self, text):
text = re.sub(r"(['\\\\])", r"\\\1", text)
text = text.replace("\r\n", "\\r\n")
return text
def add_text(self, buf, text, encode_newline=False):
if not text: return
use_unicode = self.encoding and python2
buf.append(use_unicode and "u'''" or "'''")
text = self._quote_text(text)
if not encode_newline: buf.extend((text, "''', "))
elif text.endswith("\r\n"): buf.extend((text[0:-2], "\\r\\n''', "))
elif text.endswith("\n"): buf.extend((text[0:-1], "\\n''', "))
else: buf.extend((text, "''', "))
_add_text = add_text
def add_expr(self, buf, code, *flags):
if not code or code.isspace(): return
flag_escape, flag_tostr = flags
if not self.tostrfunc: flag_tostr = False
if not self.escapefunc: flag_escape = False
if flag_tostr and flag_escape: s1, s2 = "_escape(_to_str(", ")), "
elif flag_tostr: s1, s2 = "_to_str(", "), "
elif flag_escape: s1, s2 = "_escape(", "), "
else: s1, s2 = "(", "), "
buf.extend((s1, code, s2, ))
def add_stmt(self, buf, code):
if not code: return
lines = code.splitlines(True) # keep "\n"
if lines[-1][-1] != "\n":
lines[-1] = lines[-1] + "\n"
buf.extend(lines)
self._add_localvars_assignments_to_stmts(buf)
def _add_localvars_assignments_to_stmts(self, buf):
if self._localvars_assignments_added:
return
for index, stmt in enumerate(buf):
if not re.match(r'^[ \t]*(?:\#|_buf ?= ?\[\]|from __future__)', stmt):
break
else:
return
self._localvars_assignments_added = True
if re.match(r'^[ \t]*(if|for|while|def|with|class)\b', stmt):
buf.insert(index, self._localvars_assignments() + "\n")
else:
buf[index] = self._localvars_assignments() + buf[index]
_START_WORDS = dict.fromkeys(('for', 'if', 'while', 'def', 'try:', 'with', 'class'), True)
_END_WORDS = dict.fromkeys(('#end', '#endfor', '#endif', '#endwhile', '#enddef', '#endtry', '#endwith', '#endclass'), True)
_CONT_WORDS = dict.fromkeys(('elif', 'else:', 'except', 'except:', 'finally:'), True)
_WORD_REXP = re.compile(r'\S+')
depth = -1
##
## ex.
## input = r"""
## if items:
## _buf.extend(('<ul>\n', ))
## i = 0
## for item in items:
## i += 1
## _buf.extend(('<li>', to_str(item), '</li>\n', ))
## #endfor
## _buf.extend(('</ul>\n', ))
## #endif
## """[1:]
## lines = input.splitlines(True)
## block = self.parse_lines(lines)
## #=> [ "if items:\n",
## [ "_buf.extend(('<ul>\n', ))\n",
## "i = 0\n",
## "for item in items:\n",
## [ "i += 1\n",
## "_buf.extend(('<li>', to_str(item), '</li>\n', ))\n",
## ],
## "#endfor\n",
## "_buf.extend(('</ul>\n', ))\n",
## ],
## "#endif\n",
## ]
def parse_lines(self, lines):
block = []
try:
self._parse_lines(lines.__iter__(), False, block, 0)
except StopIteration:
if self.depth > 0:
fname, linenum, colnum, linetext = self.filename, len(lines), None, None
raise TemplateSyntaxError("unexpected EOF.", (fname, linenum, colnum, linetext))
else:
pass
return block
def _parse_lines(self, lines_iter, end_block, block, linenum):
if block is None: block = []
_START_WORDS = self._START_WORDS
_END_WORDS = self._END_WORDS
_CONT_WORDS = self._CONT_WORDS
_WORD_REXP = self._WORD_REXP
get_line = lines_iter.next
while True:
line = get_line()
linenum += line.count("\n")
m = _WORD_REXP.search(line)
if not m:
block.append(line)
continue
word = m.group(0)
if word in _END_WORDS:
if word != end_block and word != '#end':
if end_block is False:
msg = "'%s' found but corresponding statement is missing." % (word, )
else:
msg = "'%s' expected but got '%s'." % (end_block, word)
colnum = m.start() + 1
raise TemplateSyntaxError(msg, (self.filename, linenum, colnum, line))
return block, line, None, linenum
elif line.endswith(':\n') or line.endswith(':\r\n'):
if word in _CONT_WORDS:
return block, line, word, linenum
elif word in _START_WORDS:
block.append(line)
self.depth += 1
cont_word = None
try:
child_block, line, cont_word, linenum = \
self._parse_lines(lines_iter, '#end'+word, [], linenum)
block.extend((child_block, line, ))
while cont_word: # 'elif' or 'else:'
child_block, line, cont_word, linenum = \
self._parse_lines(lines_iter, '#end'+word, [], linenum)
block.extend((child_block, line, ))
except StopIteration:
msg = "'%s' is not closed." % (cont_word or word)
colnum = m.start() + 1
raise TemplateSyntaxError(msg, (self.filename, linenum, colnum, line))
self.depth -= 1
else:
block.append(line)
else:
block.append(line)
assert "unreachable"
def _join_block(self, block, buf, depth):
indent = ' ' * (self.indent * depth)
for line in block:
if isinstance(line, list):
self._join_block(line, buf, depth+1)
elif line.isspace():
buf.append(line)
else:
buf.append(indent + line.lstrip())
def _arrange_indent(self, buf):
"""arrange indentation of statements in buf"""
block = self.parse_lines(buf)
buf[:] = []
self._join_block(block, buf, 0)
def render(self, context=None, globals=None, _buf=None):
"""Evaluate python code with context dictionary.
If _buf is None then return the result of evaluation as str,
else return None.
context:dict (=None)
Context object to evaluate. If None then new dict is created.
globals:dict (=None)
Global object. If None then globals() is used.
_buf:list (=None)
If None then new list is created.
"""
if context is None:
locals = context = {}
elif self.args is None:
locals = context.copy()
else:
locals = {}
if '_engine' in context:
context.get('_engine').hook_context(locals)
locals['_context'] = context
if globals is None:
globals = sys._getframe(1).f_globals
bufarg = _buf
if _buf is None:
_buf = []
locals['_buf'] = _buf
if not self.bytecode:
self.compile()
if self.trace:
_buf.append("<!-- ***** begin: %s ***** -->\n" % self.filename)
exec(self.bytecode, globals, locals)
_buf.append("<!-- ***** end: %s ***** -->\n" % self.filename)
else:
exec(self.bytecode, globals, locals)
if bufarg is not None:
return bufarg
elif not logger:
return ''.join(_buf)
else:
try:
return ''.join(_buf)
except UnicodeDecodeError, ex:
logger.error("[tenjin.Template] " + str(ex))