def setUpClass(cls):
search_path = sys.argv[0].split(':')
import_modules(search_path, 'entities')
EntityType.register(199, 'TestEntity', Entity)
# empty history
Database.instance().write('DELETE FROM history')
selection = None
if query_parameters is not None:
selection = Database.instance().select(query, query_parameters)
else:
selection = Database.instance().select(query)
for row in selection:
# uniqueid, typeid, name, stateid, statevalue, lastcheckin
unique_id, row = row[0], (row[1], row[2], row[3], row[4], row[5])
yield Entity.__create_from_db_row(unique_id, row)
@classmethod
def check_database_table(cls):
''' Checks whether the related database table exists
and creates it if is does not. '''
db = Database.instance()
with db.writer():
try:
db.select(Entity.__table_exists)
except:
db.write(Entity.__table_create)
Entity.check_database_table()
# load all modules containing entity definitions
__entities_paths = ['entities']
__entities_paths.extend(sysargs.entities.search_path)
import_modules(__entities_paths, 'entities')
