diff --git a/calculate/templates/template_filters.py b/calculate/templates/template_filters.py index c253616..fdc2924 100644 --- a/calculate/templates/template_filters.py +++ b/calculate/templates/template_filters.py @@ -3,7 +3,9 @@ def cut(instr: str, field_num: int = 0, delimiter: str = "-") -> str: Разделить входную строку `instr` разделительем `delimiter` и вернуть поле `field_num`. """ - parts = instr.split(delimiter) + if instr is None: + return "" + parts = str(instr).split(delimiter) if field_num < len(parts): return parts[field_num] else: diff --git a/tests/templates/test_template_filters.py b/tests/templates/test_template_filters.py index fefece7..996b2de 100644 --- a/tests/templates/test_template_filters.py +++ b/tests/templates/test_template_filters.py @@ -1,6 +1,10 @@ import pytest import calculate.templates.template_filters as filters +class TestObj: + def __str__(self): + return "TestObj_value" + @pytest.mark.parametrize('case', [ { @@ -9,6 +13,18 @@ import calculate.templates.template_filters as filters "params": (), "result": "" }, + { + "name": "none value", + "data": None, + "params": (), + "result": "" + }, + { + "name": "obj value", + "data": TestObj(), + "params": (), + "result": "TestObj_value" + }, { "name": "default", "data": "123",