def __init__(self, session=None, dbref=None, initdb=False): if not session: assert dbref, ( dbref, initdb ) # set for SA, get engine to use as DBAPI-2.0 compatible connection self.session = util.get_session(dbref, True) self.connection = util.SqlBase.metadata.bind.raw_connection() else: self.session = session # XXX can I get raw-connection from self.session? #self.connection = SqlBase.metadata.bind.raw_connection() #logger.info("Connected to %s", self.connection) logger.info("Extractor store to %s", self.session)
def __init__(self, session=None, dbref=None): if not session: assert dbref, ("Missing SQL-alchemy DB ref", self) # set for SA, then get engine to use as DBAPI-2.0 compatible connection self.session = get_session(dbref, True) else: self.session = session # XXX can I get raw-connection from self.session? #self.connection = SqlBase.metadata.bind.raw_connection() #logger.info("Connected to %s", self.connection) logger.info("Extractor store to %s", self.session) class Title(SqlBase): __table__ = SqlBase.metadata.tables['titles'] self.Title = Title
def reset_schema(self, argv): self.prepare_initial_components() self.process_command_line(argv=argv) # XXX self.prepare(None, **self.store_params) session = get_session(self.settings.dbref, True) conn = SqlBase.metadata.bind.raw_connection() self.init_extractors() for extractor, storage in self.extractors: store = storage(session=session) store.connection = conn try: store.reset_schema() except sqlite3.OperationalError, e: exc_type, exc_value, exc_traceback = sys.exc_info() e.info = sys.exc_info() logger.error( "Error in extractor SQL %r: %s" % (storage, e) ) raise e
def __init__(self, dbref=None, initdb=False): assert dbref, self #print 'InventoryStorage', 'init', dbref, initdb self.sa = get_session(dbref, initdb)