graphene/graphene/contrib/sqlalchemy/utils.py
2015-11-25 20:31:49 -05:00

9 lines
272 B
Python

def get_type_for_model(schema, model):
schema = schema
types = schema.types.values()
for _type in types:
type_model = hasattr(_type, '_meta') and getattr(
_type._meta, 'model', None)
if model == type_model:
return _type