mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-11 20:27:03 +03:00
99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
from singledispatch import singledispatch
|
|
from sqlalchemy import types
|
|
from sqlalchemy.orm import interfaces
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
from graphene import Enum, ID, Boolean, Float, Int, String, List, Field, Dynamic
|
|
from graphene.relay import is_node
|
|
from graphene.types.json import JSONString
|
|
from .fields import SQLAlchemyConnectionField
|
|
|
|
try:
|
|
from sqlalchemy_utils.types.choice import ChoiceType
|
|
except ImportError:
|
|
class ChoiceType(object):
|
|
pass
|
|
|
|
|
|
def convert_sqlalchemy_relationship(relationship, registry):
|
|
direction = relationship.direction
|
|
model = relationship.mapper.entity
|
|
|
|
def dynamic_type():
|
|
_type = registry.get_type_for_model(model)
|
|
if not _type:
|
|
return None
|
|
if (direction == interfaces.MANYTOONE or not relationship.uselist):
|
|
return Field(_type)
|
|
elif (direction == interfaces.ONETOMANY or
|
|
direction == interfaces.MANYTOMANY):
|
|
if is_node(_type):
|
|
return SQLAlchemyConnectionField(_type)
|
|
return Field(List(_type))
|
|
|
|
return Dynamic(dynamic_type)
|
|
|
|
|
|
def convert_sqlalchemy_column(column, registry=None):
|
|
return convert_sqlalchemy_type(getattr(column, 'type', None), column, registry)
|
|
|
|
|
|
@singledispatch
|
|
def convert_sqlalchemy_type(type, column, registry=None):
|
|
raise Exception(
|
|
"Don't know how to convert the SQLAlchemy field %s (%s)" % (column, column.__class__))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(types.Date)
|
|
@convert_sqlalchemy_type.register(types.DateTime)
|
|
@convert_sqlalchemy_type.register(types.Time)
|
|
@convert_sqlalchemy_type.register(types.String)
|
|
@convert_sqlalchemy_type.register(types.Text)
|
|
@convert_sqlalchemy_type.register(types.Unicode)
|
|
@convert_sqlalchemy_type.register(types.UnicodeText)
|
|
@convert_sqlalchemy_type.register(types.Enum)
|
|
@convert_sqlalchemy_type.register(postgresql.ENUM)
|
|
@convert_sqlalchemy_type.register(postgresql.UUID)
|
|
def convert_column_to_string(type, column, registry=None):
|
|
return String(description=column.doc, required=not(column.nullable))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(types.SmallInteger)
|
|
@convert_sqlalchemy_type.register(types.BigInteger)
|
|
@convert_sqlalchemy_type.register(types.Integer)
|
|
def convert_column_to_int_or_id(type, column, registry=None):
|
|
if column.primary_key:
|
|
return ID(description=column.doc, required=not(column.nullable))
|
|
else:
|
|
return Int(description=column.doc, required=not(column.nullable))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(types.Boolean)
|
|
def convert_column_to_boolean(type, column, registry=None):
|
|
return Boolean(description=column.doc, required=not(column.nullable))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(types.Float)
|
|
@convert_sqlalchemy_type.register(types.Numeric)
|
|
def convert_column_to_float(type, column, registry=None):
|
|
return Float(description=column.doc, required=not(column.nullable))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(ChoiceType)
|
|
def convert_column_to_enum(type, column, registry=None):
|
|
name = '{}_{}'.format(column.table.name, column.name).upper()
|
|
return Enum(name, type.choices, description=column.doc)
|
|
|
|
|
|
@convert_sqlalchemy_type.register(postgresql.ARRAY)
|
|
def convert_postgres_array_to_list(type, column, registry=None):
|
|
graphene_type = convert_sqlalchemy_type(column.type.item_type, column)
|
|
return List(graphene_type, description=column.doc, required=not(column.nullable))
|
|
|
|
|
|
@convert_sqlalchemy_type.register(postgresql.HSTORE)
|
|
@convert_sqlalchemy_type.register(postgresql.JSON)
|
|
@convert_sqlalchemy_type.register(postgresql.JSONB)
|
|
def convert_json_to_string(type, column, registry=None):
|
|
return JSONString(description=column.doc, required=not(column.nullable))
|