def getSiteKeyData():
keyData = {}
keyData['pubKey'] = ""
keyData['pubKeyHash'] = ""
keyData['priKey'] = ""
keyData['priKeyHash'] = ""
_priKeyPath = ""
_pubKeyPath = ""
srv_dict = return_data_for_root_key('server')
if all(k in srv_dict for k in ("priKey", "pubKey")):
_priKeyPath = srv_dict['priKey']
_pubKeyPath = srv_dict['pubKey']
if os.path.exists(_priKeyPath) and os.path.exists(_pubKeyPath):
keyData['priKeyHash'] = md5ForFile(_priKeyPath)
keyData['pubKeyHash'] = md5ForFile(_pubKeyPath)
keyData['pubKey'] = open(_pubKeyPath, 'r').read()
keyData['priKey'] = open(_priKeyPath, 'r').read()
return keyData
def addDefaultSWGroup():
# Check for config
if hasDefaultSWGroup():
return False
adm_dict = return_data_for_root_key('users')
adm_user = adm_dict['admin']['name']
# Create UUID
_uuid = str(uuid.uuid4())
# Add Agent Config
dts = datetime.now()
db.session.add(MpSoftwareGroupPrivs(gid=_uuid, uid=adm_user, isowner='1'))
db.session.commit()
db.session.add(
MpSoftwareGroup(gid=_uuid,
gName="Default",
gDescription="Default",
gType="0",
gHash='0',
state='1',
cdate=dts,
mdate=dts))
db.session.commit()
def getSiteKeyData():
keyData = {}
keyData['pubKey'] = ""
keyData['pubKeyHash'] = ""
keyData['priKey'] = ""
keyData['priKeyHash'] = ""
_priKeyPath = ""
_pubKeyPath = ""
srv_dict = return_data_for_root_key('server')
if all(k in srv_dict for k in ("priKey", "pubKey")):
_priKeyPath = srv_dict['priKey']
_pubKeyPath = srv_dict['pubKey']
if os.path.exists(_priKeyPath) and os.path.exists(_pubKeyPath):
keyData['priKeyHash'] = md5ForFile(_priKeyPath)
keyData['pubKeyHash'] = md5ForFile(_pubKeyPath)
keyData['pubKey'] = open(_pubKeyPath, 'r').read()
keyData['priKey'] = open(_priKeyPath, 'r').read()
return keyData
def addDefaultAdminAccount():
usr_dict = return_data_for_root_key('users')
if 'admin' in usr_dict:
adm_account = usr_dict['admin']
if adm_account['enabled']:
_pass = generate_password_hash(adm_account['pass'])
db.session.add(AdmUsers(user_id=adm_account['name'], user_RealName="MPAdmin", user_pass=_pass, enabled='1'))
db.session.commit()
return True
return False
def addDefaultAdminAccount():
usr_dict = return_data_for_root_key('users')
if 'admin' in usr_dict:
adm_account = usr_dict['admin']
if adm_account['enabled']:
_pass = generate_password_hash(adm_account['pass'])
db.session.add(
AdmUsers(user_id=adm_account['name'],
user_RealName="MPAdmin",
user_pass=_pass,
enabled='1'))
db.session.commit()
return True
return False
def addDefaultSWGroup():
# Check for config
if hasDefaultSWGroup():
return False
adm_dict = return_data_for_root_key('users')
adm_user = adm_dict['admin']['name']
# Create UUID
_uuid = str(uuid.uuid4())
# Add Agent Config
dts = datetime.now()
db.session.add(MpSoftwareGroupPrivs(gid=_uuid, uid=adm_user, isowner='1'))
db.session.commit()
db.session.add(MpSoftwareGroup(gid=_uuid, gName="Default", gDescription="Default", gType="0", gHash='0', state='1', cdate=dts, mdate=dts))
db.session.commit()
def addDefaultPatchGroup():
# Check for config
if hasDefaultPatchGroup():
return False
adm_dict = return_data_for_root_key('users')
adm_user = adm_dict['admin']['name']
# Create UUID
_uuid = str(uuid.uuid4())
# Add Agent Config
db.session.add(MpPatchGroup(name="Default", id=_uuid, type="0"))
db.session.commit()
# Add Agent Config
db.session.add(PatchGroupMembers(user_id=adm_user, patch_group_id=_uuid, is_owner="1"))
db.session.commit()
def addDefaultPatchGroup():
# Check for config
if hasDefaultPatchGroup():
return False
adm_dict = return_data_for_root_key('users')
adm_user = adm_dict['admin']['name']
# Create UUID
_uuid = str(uuid.uuid4())
# Add Agent Config
db.session.add(MpPatchGroup(name="Default", id=_uuid, type="0"))
db.session.commit()
# Add Agent Config
db.session.add(
PatchGroupMembers(user_id=adm_user, patch_group_id=_uuid,
is_owner="1"))
db.session.commit()
