Move session and db engine code into grumpy.database

1 files changed, 2 insertions, 8 deletions

diff --git a/utils/grumpy_sync.py b/utils/grumpy_sync.py
index 7da95c8..20c4674 100755
--- a/utils/grumpy_sync.py
+++ b/utils/grumpy_sync.py
@@ -10,7 +10,6 @@ from pkgcore.ebuild import repository
 
 from sqlalchemy import create_engine
 from sqlalchemy.orm import scoped_session, sessionmaker
-from sqlalchemy.orm.exc import NoResultFound
 
 from snakeoil.fileutils import iter_read_bash
 
@@ -18,16 +17,11 @@ path = os.path.join(os.path.dirname(__file__), os.path.pardir)
 sys.path.insert(0, path)
 del path
 
-from grumpy.models import (Base, Category, Developer, Ebuild, Herd,
+from grumpy.database import session
+from grumpy.models import (Base, Category, Developer, Ebuild, Herd, \
                            Package, Setting)
 
 def main(path):
-    engine = create_engine('postgresql://grumpy:grumpy@localhost/grumpy')
-    session = scoped_session(sessionmaker(autocommit=False,
-                                          autoflush=False,
-                                          bind=engine))
-    Base.query = session.query_property()
-
     # pkgcore part to fetch all the ebuild information
     conf = load_config()
     eclass_cache = conf.eclass_cache['eclass stack']