Commit edf9a1d7 authored by Administrator's avatar Administrator

[FIX] Database connexion fix thanks to Elias.

parent c7d3c746
from gargantext_web import settings
from django.conf import settings
from node import models
__all__ = ['literalquery', 'session', 'cache', 'Session',
'bulk_insert', 'engine', 'get_cursor']
__all__ = ['literalquery', 'session', 'cache', 'Session', 'bulk_insert', 'engine', 'get_cursor', 'User']
# initialize sqlalchemy
......@@ -15,9 +15,16 @@ from sqlalchemy import create_engine, MetaData, Table, Column, ForeignKey
from sqlalchemy.types import Integer, String, DateTime
from sqlalchemy.dialects.postgresql import JSON
engine = create_engine('postgresql+psycopg2://{USER}:{PASSWORD}@{HOST}/{NAME}'.format(
**settings.DATABASES['default']
))
# SQLAlchemy session management
def get_engine():
from sqlalchemy import create_engine
url = 'postgresql+psycopg2://{USER}:{PASSWORD}@{HOST}:{PORT}/{NAME}'.format(
**settings.DATABASES['default']
)
return create_engine(url, use_native_hstore=True)
engine = get_engine()
Base = automap_base()
Base.prepare(engine, reflect=True)
......@@ -122,17 +129,6 @@ def literalquery(statement, dialect=None):
return LiteralCompiler(dialect, statement)
# SQLAlchemy session management
def get_engine():
from sqlalchemy import create_engine
url = 'postgresql+psycopg2://{USER}:{PASSWORD}@{HOST}/{NAME}'.format(
**settings.DATABASES['default']
)
return create_engine(url, use_native_hstore=True)
engine = get_engine()
def get_sessionmaker():
from sqlalchemy.orm import sessionmaker
return sessionmaker(bind=engine)
......@@ -200,6 +196,7 @@ def get_cursor():
'user': db_settings['USER'],
'password': db_settings['PASSWORD'],
'host': db_settings['HOST'],
'port': db_settings['PORT']
})
return db, db.cursor()
......@@ -233,4 +230,3 @@ class bulk_insert:
return ''
readline = read
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment