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 delete_whitelist(self, ipaddress, conn=None):
if conn == None :
return None, self.LOADDB_FAILED
whitelist = whitelist_mgmt(conn)
return whitelist.delete_ipaddr(ipaddress)
def show_whitelist(self, conn=None):
if conn == None :
return None, self.LOADDB_FAILED
whitelist = whitelist_mgmt(conn)
return whitelist.show_ipaddr()
def modify_whitelist(self, ipaddress, desc, conn=None):
if conn == None :
return None, self.LOADDB_FAILED
whitelist = whitelist_mgmt(conn)
return whitelist.modify_ipaddr(ipaddress, desc)
def checkipaddress(self, remoteip, conn=None):
if conn == None :
return None, self.LOADDB_FAILED
whitelist = whitelist_mgmt(conn)
ipaddress = whitelist.find_ipaddr(remoteip)
return (ipaddress != None)
