def setUp(self): tables = [] collections = [ 'Jails', 'Environments', 'Images', 'JailTypes', 'Keys', 'Users', 'VirtualMachines', 'YumRepositories' ] for c in collections: tables.append(getattr(models, c).table_definition()) db = ':memory:' self._connection = sqlite3.connect(db) cherrypy.thread_data.db = self._connection cherrypy.thread_data.db.row_factory = sqlite3.Row m = Migration(self._connection, tables) m.migrate()
'tools.staticdir.dir': 'static', 'tools.staticdir.root': os.getcwd(), 'tools.staticdir.index': 'index.html' } } connect(None) import libs.controller cherrypy.tools.myauth = cherrypy.Tool("before_handler", libs.controller.auth) import models, controllers if args.init: cherrypy.log("Initializing persistent storage.") from libs.model import Migration m = Migration(cherrypy.thread_data.db, []) m.init() models.Users.load() if models.Users.first(username='******'): cherrypy.log('Admin account already created, skipping.') os._exit(0) user = models.Users.new(username='******', user_group='admin', name='Administrator') user.password = models.Users.hash_password('puck') if models.Users.add(user): cherrypy.log('Admin account added.') cherrypy.log('\tUsername: admin') cherrypy.log('\tPassword: puck') os._exit(0) else: