Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Extend functionality of template function 'lookup'

Browse Source
This commit is contained in:
Charles Haley 2010-10-17 05:30:09 +01:00
parent 0acd342fef
commit 9c0ac34187
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
src/calibre/utils/formatter.py
View File

@ -22,11 +22,21 @@ class TemplateFormatter(string.Formatter):
self.book = None self.book = None
self.kwargs = None self.kwargs = None
def _lookup(self, val, field_if_set, field_not_set): def _lookup(self, val, *args):
if len(args) == 2:
if val: if val:
return self.vformat('{'+field_if_set.strip()+'}', [], self.kwargs) return self.vformat('{'+args[0].strip()+'}', [], self.kwargs)
else: else:
return self.vformat('{'+field_not_set.strip()+'}', [], self.kwargs) return self.vformat('{'+args[1].strip()+'}', [], self.kwargs)
if (len(args) % 2) != 1:
raise ValueError(_('lookup requires either 2 or an odd number of arguments'))
i = 0
while i < len(args):
if i + 1 >= len(args):
return self.vformat('{' + args[i].strip() + '}', [], self.kwargs)
if re.search(args[i], val):
return self.vformat('{'+args[i+1].strip() + '}', [], self.kwargs)
i += 2
def _test(self, val, value_if_set, value_not_set): def _test(self, val, value_if_set, value_not_set):
if val: if val:
@ -41,6 +51,8 @@ class TemplateFormatter(string.Formatter):
return value_if_not return value_if_not
def _switch(self, val, *args): def _switch(self, val, *args):
if (len(args) % 2) != 1:
raise ValueError(_('switch requires an odd number of arguments'))
i = 0 i = 0
while i < len(args): while i < len(args):
if i + 1 >= len(args): if i + 1 >= len(args):
@ -73,7 +85,7 @@ class TemplateFormatter(string.Formatter):
'capitalize' : (0, lambda s,x: x.capitalize()), 'capitalize' : (0, lambda s,x: x.capitalize()),
'contains' : (3, _contains), 'contains' : (3, _contains),
'ifempty' : (1, _ifempty), 'ifempty' : (1, _ifempty),
'lookup' : (2, _lookup), 'lookup' : (-1, _lookup),
're' : (2, _re), 're' : (2, _re),
'shorten' : (3, _shorten), 'shorten' : (3, _shorten),
'switch' : (-1, _switch), 'switch' : (-1, _switch),
@ -129,9 +141,9 @@ class TemplateFormatter(string.Formatter):
(func[0] > 0 and func[0] != len(args)): (func[0] > 0 and func[0] != len(args)):
raise ValueError('Incorrect number of arguments for function '+ fmt[0:p]) raise ValueError('Incorrect number of arguments for function '+ fmt[0:p])
if func[0] == 0: if func[0] == 0:
val = func[1](self, val) val = func[1](self, val).strip()
else: else:
val = func[1](self, val, *args) val = func[1](self, val, *args).strip()
if val: if val:
val = string.Formatter.format_field(self, val, dispfmt) val = string.Formatter.format_field(self, val, dispfmt)
if not val: if not val: