def initTask(self): super(AtcdDBQueueTask, self).initTask() try: self.sqlite_manager = SQLiteManager(self.sqlite_file, self.logger) except OperationalError: self.logger.exception( 'Unable to initialize DB from file "{0}"'.format( self.sqlite_file)) raise
def initTask(self): super(AtcdDBQueueTask, self).initTask() try: self.sqlite_manager = SQLiteManager(self.sqlite_file, self.logger) except OperationalError: self.logger.exception( 'Unable to initialize DB from file "{0}"'.format( self.sqlite_file ) ) raise
class AtcdDBQueueTask(QueueTask): OPT_PREFIX = 'sqlite' workers = 1 DEFAULT_SQLITE_FILE = '/var/lib/atcd.db' sqlite_file = option( default=DEFAULT_SQLITE_FILE, metavar='SQLITE_FILE', help='Location to store the sqlite3 db [%(default)s]', name='file', ) def initTask(self): super(AtcdDBQueueTask, self).initTask() try: self.sqlite_manager = SQLiteManager(self.sqlite_file, self.logger) except OperationalError: self.logger.exception( 'Unable to initialize DB from file "{0}"'.format( self.sqlite_file ) ) raise def execute(self, item, context): try: obj, action = item except ValueError: self.logger.exception('Error executing on item: {0}'.format(item)) return try: func = getattr(self.sqlite_manager, action) except AttributeError: self.logger.exception( 'unable to run action, {0}, no such method'.format(action) ) raise try: if isinstance(obj, tuple): func(*obj) else: func(obj) except OperationalError: self.logger.exception("Unsupported operation") return def get_saved_shapings(self): return self.sqlite_manager.get_saved_shapings()
class AtcdDBQueueTask(QueueTask): OPT_PREFIX = 'sqlite' workers = 1 DEFAULT_SQLITE_FILE = '/var/lib/atcd.db' sqlite_file = option( default=DEFAULT_SQLITE_FILE, metavar='SQLITE_FILE', help='Location to store the sqlite3 db [%(default)s]', name='file', ) def initTask(self): super(AtcdDBQueueTask, self).initTask() try: self.sqlite_manager = SQLiteManager(self.sqlite_file, self.logger) except OperationalError: self.logger.exception( 'Unable to initialize DB from file "{0}"'.format( self.sqlite_file)) raise def execute(self, item, context): try: obj, action = item except ValueError: self.logger.exception('Error executing on item: {0}'.format(item)) return try: func = getattr(self.sqlite_manager, action) except AttributeError: self.logger.exception( 'unable to run action, {0}, no such method'.format(action)) raise try: if isinstance(obj, tuple): func(*obj) else: func(obj) except OperationalError: self.logger.exception("Unsupported operation") return def get_saved_shapings(self): return self.sqlite_manager.get_saved_shapings()
def test_sqlite_file_not_found(self): with pytest.raises(sqlite3.OperationalError): SQLiteManager('/this/path/should/not/exist')
def dbm(atc_db_file): return SQLiteManager(atc_db_file.name)