Ejemplo n.º 1
0
 def init_db(self):
     self.engine = db.Engine()
     table = db.Tables()[self.table_name]
     self.columns = [
         getattr(table, column_name) for column_name in self.column_names
     ]
     self.session = db.sessionmaker(bind=self.engine)()
Ejemplo n.º 2
0
import datetime
import itertools
import logging
from contextlib import contextmanager

from nems import db

log = logging.getLogger(__name__)

tQueue = db.Tables()['tQueue']


@contextmanager
def db_session():
    """Context manager to handle database connections."""
    session = db.Session()

    try:
        yield session
        session.commit()
    except:
        session.rollback()
    finally:
        session.close()


def enqueue_exacloud_models(cellist,
                            batch,
                            modellist,
                            user,
                            linux_user,
Ejemplo n.º 3
0
 def init_db(self):
     self.engine = db.Engine()
     table = db.Tables()[self.table_name]
     self.column = getattr(table, self.column_name)
     self.session = db.sessionmaker(bind=self.engine)()
Ejemplo n.º 4
0
 def init_db(self):
     self.engine = db.Engine()
     self.table = db.Tables()[self.table_name]
     self.session = db.sessionmaker(bind=self.engine)()