246 lines
11 KiB
Python
246 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
|
|
from defusedxml.ElementTree import parse
|
|
from datetime import datetime
|
|
from os import listdir
|
|
from pathlib import Path
|
|
from uuid import uuid4
|
|
|
|
from PySide6.QtGui import QIcon
|
|
from PySide6.QtCore import QByteArray
|
|
|
|
|
|
class ResourceHandler:
|
|
|
|
def getIcon(self, iconName: str):
|
|
return self.icons[iconName]
|
|
|
|
# Load all resources needed.
|
|
def __init__(self, mainWindow, messageHandler):
|
|
self.mainWindow = mainWindow
|
|
self.messageHandler = messageHandler
|
|
self.entityCategoryList = {}
|
|
|
|
self.icons = {"uploading": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Uploading.png"),
|
|
"uploaded": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Uploaded.png"),
|
|
"upArrow": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "UpArrow.png"),
|
|
"downArrow": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "DownArrow.png"),
|
|
"isolatedNodes": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "SelectIsolated.png"),
|
|
"addCanvas": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Add_Canvas.png"),
|
|
"generateReport": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Generate_Report.png"),
|
|
"leafNodes": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "SelectLeaf.png"),
|
|
"nonIsolatedNodes": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "SelectNonIsolated.png"),
|
|
"rootNodes": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "SelectRoot.png"),
|
|
"split": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Split.png"),
|
|
"merge": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "Merge.png"),
|
|
"shortestPath": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "ShortestPath.png"),
|
|
"drawLink": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "DrawLink.png"),
|
|
"rearrange": str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) /
|
|
"Resources" / "Icons" / "RearrangeGraph.png"),
|
|
}
|
|
|
|
self.loadCoreEntities()
|
|
|
|
def getEntityCategories(self):
|
|
eList = []
|
|
for category in self.entityCategoryList:
|
|
eList.append(category)
|
|
return eList
|
|
|
|
def getAllEntityDetailsWithIconsInCategory(self, category):
|
|
eList = []
|
|
for entity in self.entityCategoryList[category]:
|
|
entityValue = self.entityCategoryList[category][entity]
|
|
eList.append((self.getBareBonesEntityJson(entity),
|
|
entityValue['Icon']
|
|
))
|
|
return eList
|
|
|
|
def getEntityAttributes(self, entityType):
|
|
aList = []
|
|
try:
|
|
for category in self.entityCategoryList:
|
|
if entityType in self.entityCategoryList[category]:
|
|
for attribute in self.entityCategoryList[category][entityType]['Attributes']:
|
|
aList.append(attribute)
|
|
break
|
|
except KeyError:
|
|
self.messageHandler.error("Attempted to get attributes for "
|
|
"nonexistent entity type: " + str(entityType), True)
|
|
return None
|
|
return aList
|
|
|
|
def getAllEntitiesInCategory(self, category):
|
|
"""
|
|
Get all Entity Types in the specified category.
|
|
"""
|
|
eList = []
|
|
for entity in self.entityCategoryList[category]:
|
|
eList.append(entity)
|
|
return eList
|
|
|
|
def getAllEntities(self):
|
|
"""
|
|
Get all recognised Entity Types.
|
|
"""
|
|
eList = []
|
|
for category in self.getEntityCategories():
|
|
for entity in self.getAllEntitiesInCategory(category):
|
|
eList.append(entity)
|
|
return eList
|
|
|
|
def addRecognisedEntityTypes(self, entityFile) -> bool:
|
|
try:
|
|
tree = parse(entityFile, forbid_dtd=True, forbid_entities=True, forbid_external=True)
|
|
except Exception as exc:
|
|
self.mainWindow.MESSAGEHANDLER.warning('Error occured when loading entities from '
|
|
+ entityFile + ', skipping.')
|
|
return False
|
|
|
|
root = tree.getroot()
|
|
|
|
category = root.tag.replace('_', ' ')
|
|
for entity in list(root):
|
|
entityName = entity.tag.replace('_', ' ')
|
|
attributes = entity.find('Attributes').text.strip().split(',')
|
|
icon = entity.find('Icon')
|
|
if icon is not None:
|
|
icon = icon.text.strip()
|
|
elif icon is None or icon == '':
|
|
icon = 'Default.svg'
|
|
if self.entityCategoryList.get(category) is None:
|
|
self.entityCategoryList[category] = {}
|
|
self.entityCategoryList[category][entityName] = {
|
|
'Attributes': attributes,
|
|
'Icon': str(Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Resources" / "Icons" / icon)}
|
|
return True
|
|
|
|
def loadCoreEntities(self):
|
|
entDir = Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Core" / "Entities"
|
|
for entFile in listdir(entDir):
|
|
if entFile.endswith('.xml'):
|
|
self.addRecognisedEntityTypes(entDir / entFile)
|
|
|
|
def loadModuleEntities(self):
|
|
entDir = Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Modules"
|
|
for module in listdir(entDir):
|
|
for entFile in listdir(entDir / module):
|
|
if entFile.endswith('.xml'):
|
|
self.addRecognisedEntityTypes(
|
|
entDir / module / entFile)
|
|
|
|
def getEntityJson(self, entityType: str, jsonData=None):
|
|
eJson = {'uid': str(uuid4())}
|
|
try:
|
|
for category in self.entityCategoryList:
|
|
if entityType in self.entityCategoryList[category]:
|
|
for attribute in self.entityCategoryList[category][entityType]['Attributes']:
|
|
eJson[attribute] = str(None)
|
|
break
|
|
except KeyError:
|
|
self.messageHandler.error("Attempted to get attributes for "
|
|
"malformed entity type: " + str(entityType), True)
|
|
return None
|
|
eJson['Entity Type'] = entityType
|
|
eJson['Date Created'] = None
|
|
eJson['Date Last Edited'] = None
|
|
eJson['Notes'] = ""
|
|
eJson['Icon'] = self.getEntityDefaultPicture(entityType)
|
|
|
|
if jsonData is not None:
|
|
for key in eJson:
|
|
value = jsonData.get(key)
|
|
if value is not None and value != '':
|
|
eJson[key] = value
|
|
|
|
utcNow = datetime.isoformat(datetime.utcnow())
|
|
if eJson['Date Created'] is None:
|
|
eJson['Date Created'] = utcNow
|
|
|
|
eJson['Date Last Edited'] = utcNow
|
|
|
|
return eJson
|
|
|
|
def getPrimaryFieldForEntityType(self, entityType: str):
|
|
try:
|
|
for category in self.entityCategoryList:
|
|
if entityType in self.entityCategoryList[category]:
|
|
return self.entityCategoryList[category][entityType]['Attributes'][0]
|
|
except KeyError:
|
|
self.messageHandler.error("Attempted to get primary attribute for "
|
|
"malformed entity type: " + str(entityType), True)
|
|
return None
|
|
|
|
def getBareBonesEntityJson(self, entityType):
|
|
eJson = {}
|
|
try:
|
|
for category in self.entityCategoryList:
|
|
if entityType in self.entityCategoryList[category]:
|
|
for attribute in self.entityCategoryList[category][entityType]['Attributes']:
|
|
eJson[attribute] = str(None)
|
|
break
|
|
except KeyError:
|
|
self.messageHandler.error("Attempted to get attributes for "
|
|
"malformed entity type: " + str(entityType), True)
|
|
return None
|
|
eJson['Entity Type'] = entityType
|
|
|
|
return eJson
|
|
|
|
def getLinkJson(self, jsonData):
|
|
linkJson = {}
|
|
try:
|
|
linkJson['uid'] = jsonData['uid']
|
|
except KeyError:
|
|
return None
|
|
|
|
utcNow = datetime.isoformat(datetime.utcnow())
|
|
linkJson['Resolution'] = str(jsonData.get('Resolution')) # This way, if it is None, it is cast to a string.
|
|
linkJson['Date Created'] = jsonData.get('Date Created')
|
|
if linkJson['Date Created'] is None:
|
|
linkJson['Date Created'] = utcNow
|
|
linkJson['Date Last Edited'] = utcNow
|
|
linkJson['Notes'] = str(jsonData.get('Notes'))
|
|
|
|
return linkJson
|
|
|
|
def getEntityDefaultPicture(self, entityType):
|
|
picture = Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Resources" / "Icons" / "Default.svg"
|
|
try:
|
|
for category in self.entityCategoryList:
|
|
if entityType in self.entityCategoryList[category]:
|
|
entityPicture = self.entityCategoryList[category][entityType]['Icon']
|
|
if Path(entityPicture).exists():
|
|
picture = entityPicture
|
|
break
|
|
except KeyError:
|
|
self.messageHandler.warning("Attempted to get icon for "
|
|
"nonexistent entity type: " + str(entityType), popUp=False)
|
|
finally:
|
|
with open(picture, 'rb') as pictureFile:
|
|
pictureContents = pictureFile.read()
|
|
pictureByteArray = QByteArray(pictureContents)
|
|
return pictureByteArray
|
|
|
|
def getLinkPicture(self):
|
|
picture = Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Resources" / "Icons" / "Resolution.png"
|
|
return QIcon(str(picture)).pixmap(40, 40)
|
|
|
|
def getLinkArrowPicture(self):
|
|
picture = Path(self.mainWindow.SETTINGS.value("Program/BaseDir")) / "Resources" / "Icons" / "Right-Arrow.svg"
|
|
return QIcon(str(picture)).pixmap(40, 40)
|