X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/fefdce4e24f9e397df5538fe6e7f54b5ece4d841..e870e40d5fb975ba9ec0ec327014b3d16eea51e4:/src/librarian/functions.py diff --git a/src/librarian/functions.py b/src/librarian/functions.py index e5a47d6..15e931c 100644 --- a/src/librarian/functions.py +++ b/src/librarian/functions.py @@ -69,7 +69,10 @@ def reg_ends_white(): def reg_wrap_words(): def wrap_words(context, text, wrapping): - """XPath extension function automatically wrapping words in passed text""" + """ + XPath extension function automatically wrapping words + in passed text. + """ if isinstance(text, list): text = ''.join(text) if not wrapping: @@ -108,21 +111,19 @@ def reg_texcommand(): _register_function(texcommand) -def reg_lang_code_3to2(): - def lang_code_3to2(context, text): - """Convert 3-letter language code to 2-letter code""" - result = '' - text = ''.join(text) - with open(get_resource('res/ISO-639-2_8859-1.txt'), 'rb') as f: - for line in f.read().decode('latin1').split('\n'): - list = line.strip().split('|') - if list[0] == text: - result = list[2] - if result == '': - return text - else: - return result - _register_function(lang_code_3to2) +def lang_code_3to2(text): + """Convert 3-letter language code to 2-letter code""" + result = '' + text = ''.join(text) + with open(get_resource('res/ISO-639-2_8859-1.txt'), 'rb') as f: + for line in f.read().decode('latin1').split('\n'): + codes = line.strip().split('|') + if codes[0] == text: + result = codes[2] + if result == '': + return text + else: + return result def mathml_latex(context, trees):