def __init__(self, filename, initialize=False): import sqlite3 self._filename = filename conn = self.get_conn() try: if initialize: users = users_mgmt(conn) roles = userroles_mgmt(conn) whitelist = whitelist_mgmt(conn) if (users.find_user('admin') is None): users.add_user('admin', self._encode_password('admin') , self.AUTH_ADMIN, 'init admin') #if (users.find_user('guest') is None): # users.add_user('guest', self._encode_password('guest'), self.AUTH_USER, 'init user') if (whitelist.find_ipaddr('127.0.0.1') is None): whitelist.add_ipaddr('127.0.0.1', 'loopback ipaddress') if (whitelist.find_ipaddr('localhost') is None): whitelist.add_ipaddr('localhost', 'localhost ipaddress') if (roles.find_role('special') is None): roles.add_role('special', self.AUTH_SPECIAL, 'Super user privilege') if (roles.find_role('admin') is None): roles.add_role('admin', self.AUTH_ADMIN, 'Management user privilege') if (roles.find_role('user') is None): roles.add_role('user', self.AUTH_USER, 'Common user privilege') except sqlite3.IntegrityError, e: conn.rollback()
def show_role(self, conn=None): if conn == None : return None, self.LOADDB_FAILED roles = userroles_mgmt(conn) return roles.show_role()