Minor fix to get_pred_prepnp_jako_str2check command.

Bartłomiej Nitoń
1 parent 80b919c4
Showing 1 changed file with 10 additions and 0 deletions
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