chore: streamline edge creation

This commit is contained in:
Erik Wrede 2023-03-03 11:30:09 +01:00
parent d3bc70a87d
commit e3809e9aee

View File

@ -25,10 +25,10 @@ def get_edge_class(
description = f"A Relay edge containing a `{base_name}` and its cursor." description = f"A Relay edge containing a `{base_name}` and its cursor."
edge_name = f"{base_name}Edge" edge_name = f"{base_name}Edge"
if edge_class:
edge_bases = [edge_class, EdgeBase, ObjectType] edge_bases = [edge_class] if edge_class else [EdgeBase]
else: if not isinstance(edge_class, ObjectType):
edge_bases = [EdgeBase, ObjectType] edge_bases = [*edge_bases, ObjectType]
return type(edge_name, tuple(edge_bases), {"Meta": EdgeMeta}) return type(edge_name, tuple(edge_bases), {"Meta": EdgeMeta})