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

Add a test calling a stored python template from a GPM template

Browse Source
This commit is contained in:
Charles Haley 2022-10-11 11:41:51 +01:00
parent 7929687138
commit 1b7cbccdaf
1 changed files with 19 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/calibre/db/tests/reading.py
View File

@ -846,4 +846,23 @@ def evaluate(book, db, **kwargs):
'''
v = formatter.safe_format(template, {}, 'TEMPLATE ERROR', mi)
self.assertEqual(set(v.split(',')), {'Tag One', 'News', 'Tag Two'})
# test calling a python stored template from a GPM template
from calibre.utils.formatter_functions import (
load_user_template_functions, unload_user_template_functions)
load_user_template_functions('aaaaa',
[['python_stored_template',
"",
0,
'''python:
def evaluate(book, db, globals, arguments):
tags = set(db.new_api.all_field_names('tags'))
tags.add(arguments[0])
return ','.join(list(tags))
'''
]], None)
v = formatter.safe_format('program: python_stored_template("one argument")', {},
'TEMPLATE ERROR', mi)
unload_user_template_functions('aaaaa')
self.assertEqual(set(v.split(',')), {'Tag One', 'News', 'Tag Two', 'one argument'})
# }}}