diff --git a/dictionary/management/commands/get_pred_prepnp_jako_str2check.py b/dictionary/management/commands/get_pred_prepnp_jako_str2check.py index 8e9d0d4..c130531 100644 --- a/dictionary/management/commands/get_pred_prepnp_jako_str2check.py +++ b/dictionary/management/commands/get_pred_prepnp_jako_str2check.py @@ -17,6 +17,8 @@ class Command(BaseCommand): for phrase_type in pos.arguments.all(): if self.phrase_type_is_prepnp_jako_str(phrase_type): pred_prepnp_jako_str_exists = True + elif self.phrase_type_is_prepadjp_jako_str(phrase_type): + pred_prepnp_jako_str_exists = True else: other_pred_exists = True if pred_prepnp_jako_str_exists and other_pred_exists: @@ -30,3 +32,11 @@ class Command(BaseCommand): phrase_type.atributes.get(type='TYP FRAZY').values.all()[0].argument.text_rep == 'prepnp(jako,str)'): return True return False + + def phrase_type_is_prepadjp_jako_str(self, phrase_type): + if (phrase_type.text_rep == 'prepadjp(jako,str)'): + return True + if (phrase_type.type == 'lex' and + phrase_type.atributes.get(type='TYP FRAZY').values.all()[0].argument.text_rep == 'prepadjp(jako,str)'): + return True + return False