Added type to regex

devopsjosh · Jan 8, 2020 · 0b73160 · erebus1 · Jan 9, 2020 · 0b73160
1 parent 85d59eb
commit 0b73160
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/graphene_federation/service.py b/graphene_federation/service.py
@@ -17,7 +17,7 @@ def _mark_field(
             # todo write tests on regexp
             schema_field_name = to_camel_case(field_name) if auto_camelcase else field_name
             pattern = re.compile(
-                r"(\s%s\s[^\{]*\{[^\}]*\s%s[\s]*:[\s]*[^\s]+)(\s)" % (
+                r"(type\s*%s\s[^\{]*\{[^\}]*\s%s[\s]*:[\s]*[^\s]+)(\s)" % (
                     entity_name, schema_field_name))
             schema = pattern.sub(
                 rf'\g<1> {decorator_resolver(getattr(field, mark_attr_name))} ', schema)