WebODM/coreplugins/lightning/plugin.py at master · leNeo/WebODM · GitHub

Repository navigation

History

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

import json

from django.dispatch import receiver

from django.http import HttpResponse

from guardian.shortcuts import get_objects_for_user, assign_perm

from rest_framework.renderers import JSONRenderer

from django.utils.translation import gettext as _

from app.plugins import GlobalDataStore, logger

from app.plugins import PluginBase, Menu, MountPoint, UserDataStore, signals

from django.shortcuts import render

from django.contrib.auth.decorators import login_required

from django.views.decorators.http import require_POST

from nodeodm.models import ProcessingNode

from app.api.processingnodes import ProcessingNodeSerializer

ds = GlobalDataStore('lightning')

def JsonResponse(dict):

return HttpResponse(json.dumps(dict), content_type='application/json')

class Plugin(PluginBase):

def main_menu(self):

return [Menu(_("Lightning"), self.public_url(""), "fa fa-bolt fa-fw")]

def include_js_files(self):

return ['add_cost_estimate.js']

def build_jsx_components(self):

return ['app.jsx', 'CostEstimateItem.jsx']

def app_mount_points(self):

@login_required

def main(request):

uds = UserDataStore('lightning', request.user)

return render(request, self.template_path("index.html"), {

'title': _('Lightning'),

'api_key': uds.get_string("api_key")

})

@login_required

@require_POST

def save_api_key(request):

api_key = request.POST.get('api_key')

if api_key is None:

return JsonResponse({'error': 'api_key is required'})

uds = UserDataStore('lightning', request.user)

uds.set_string('api_key', api_key)

return JsonResponse({'success': True})

@login_required

@require_POST

def sync_processing_node(request):

hostname = request.POST.get('hostname')

port = int(request.POST.get('port'))

token = request.POST.get('token')

if hostname is not None and port is not None and token is not None:

nodes = get_objects_for_user(request.user, 'view_processingnode', ProcessingNode,

accept_global_perms=False)

matches = [n for n in nodes if n.hostname == hostname and n.port == port and n.token == token]

if len(matches) == 0:

# Add

node = ProcessingNode.objects.create(hostname=hostname, port=port, token=token, label="Lightning")

assign_perm('view_processingnode', request.user, node)

assign_perm('change_processingnode', request.user, node)

assign_perm('delete_processingnode', request.user, node)

# Keep track of lightning node IDs

lightning_nodes = ds.get_json('nodes', [])

lightning_nodes.append(node.id)

ds.set_json('nodes', lightning_nodes)

return get_processing_nodes(request)

else:

# Already added

return get_processing_nodes(request)

else:

return JsonResponse({'error': 'Invalid call (params missing)'})

@login_required

def get_processing_nodes(request):

nodes = get_objects_for_user(request.user, 'view_processingnode', ProcessingNode,

accept_global_perms=False)

lightning_node_ids = ds.get_json("nodes", [])

nodes = [n for n in nodes if n.id in lightning_node_ids]

serializer = ProcessingNodeSerializer(nodes, many=True)

return JsonResponse(serializer.data)

@login_required

def is_lightning_node(request):

lightning_node_ids = ds.get_json("nodes", [])

return JsonResponse({'result': int(request.GET.get('id')) in lightning_node_ids})

return [

MountPoint('$', main),

MountPoint('save_api_key$', save_api_key),

MountPoint('sync_processing_node$', sync_processing_node),

MountPoint('get_processing_nodes$', get_processing_nodes),

MountPoint('is_lightning_node$', is_lightning_node),

]

@receiver(signals.processing_node_removed, dispatch_uid="lightning_on_processing_node_removed")

def lightning_on_processing_node_removed(sender, processing_node_id, **kwargs):

node_ids = ds.get_json('nodes', [])

try:

node_ids.remove(processing_node_id)

logger.info("Removing lightning node {}".format(str(processing_node_id)))

ds.set_json('nodes', node_ids)

except ValueError:

pass

WebODM/coreplugins/lightning/plugin.py at master · leNeo/WebODM · GitHub

Repository navigation

Expand file tree

Breadcrumbs

plugin.py

Latest commit

History

Breadcrumbs

plugin.py

File metadata and controls