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)
