From 893d6f681e4d98440baf2d1dc4d0518470112658 Mon Sep 17 00:00:00 2001
From: Syrus Akbary <me@syrusakbary.com>
Date: Fri, 9 Sep 2016 19:04:21 -0700
Subject: [PATCH] Added get_query into sqlalchemy objecttype

---
 graphene-sqlalchemy/graphene_sqlalchemy/__init__.py | 8 +++++++-
 graphene-sqlalchemy/graphene_sqlalchemy/types.py    | 9 ++++++---
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/graphene-sqlalchemy/graphene_sqlalchemy/__init__.py b/graphene-sqlalchemy/graphene_sqlalchemy/__init__.py
index bdb21103..3635827d 100644
--- a/graphene-sqlalchemy/graphene_sqlalchemy/__init__.py
+++ b/graphene-sqlalchemy/graphene_sqlalchemy/__init__.py
@@ -4,6 +4,12 @@ from .types import (
 from .fields import (
     SQLAlchemyConnectionField
 )
+from .utils import (
+	get_query,
+	get_session
+)
 
 __all__ = ['SQLAlchemyObjectType',
-           'SQLAlchemyConnectionField']
+           'SQLAlchemyConnectionField',
+           'get_query',
+           'get_session']
diff --git a/graphene-sqlalchemy/graphene_sqlalchemy/types.py b/graphene-sqlalchemy/graphene_sqlalchemy/types.py
index 9237f840..bade191f 100644
--- a/graphene-sqlalchemy/graphene_sqlalchemy/types.py
+++ b/graphene-sqlalchemy/graphene_sqlalchemy/types.py
@@ -126,12 +126,15 @@ class SQLAlchemyObjectType(six.with_metaclass(SQLAlchemyObjectTypeMeta, ObjectTy
             ).format(root))
         return type(root) == cls._meta.model
 
+    @classmethod
+    def get_query(cls, context):
+        model = cls._meta.model
+        return get_query(model, context)
+
     @classmethod
     def get_node(cls, id, context, info):
         try:
-            model = cls._meta.model
-            query = get_query(model, context)
-            return query.get(id)
+            return cls.get_query(context).get(id)
         except NoResultFound:
             return None