def updatePlugin(currentName: str, newName: str, desc: str):
plugin = getPlugin(currentName)
PluginData.objects(id=plugin.id).update_one(set__description=desc,
set__name=newName)
plugin = getPlugin(newName)
return plugin.name
def createPlugin(path: str, pds: str, name: str, desc: str, poi: list):
plugin = PluginData()
plugin.name = name
plugin.predefinedDataSet = pds
plugin.description = desc
plugin.path = path
plugin.pluginPOI = poi
plugin.save()
def getPluginPOITypes(pluginName: str):
plugin = PluginData.objects(name=pluginName).first()
allPOI = plugin.pluginPOI
poiNames = []
for poi in allPOI:
poiNames.append(poi.type)
return set(poiNames)
def getPluginPOI(pluginName: str):
plugin = PluginData.objects(name=pluginName).first()
allPOI = plugin.pluginPOI
pois = []
for poi in allPOI:
pois.append(poi)
return pois
def deletePOI(pluginName: str, poiName: str):
plugin = PluginData.objects(name=pluginName).first()
pois = plugin.pluginPOI
for poi in pois:
if poi.name == poiName:
plugin.pluginPOI.remove(poi)
plugin.save()
return
def updatePOI(pluginName: str, poiName: str, newPOIName: str, type: str,
return_type: str, data: dict):
plugin = PluginData.objects(name=pluginName).first()
pois = plugin.pluginPOI
for i, poi in enumerate(pois):
if poi.name == poiName:
plugin.pluginPOI[i] = createPOI(newPOIName, type, return_type,
data)
plugin.save()
return newPOIName
def getPlugin(name: str):
plugin = PluginData.objects(name=name).first()
return plugin
def addToPlugin(name: str, poi: PluginPOI):
plugin = PluginData.objects(name=name).first()
plugin.pluginPOI.append(poi)
plugin.save()