def edit_optical_node(request, handle_id): # Get needed data from node nh, optical_node = helpers.get_nh_node(handle_id) location = optical_node.get_location() ports = optical_node.get_ports() if request.POST: form = forms.EditOpticalNodeForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, optical_node.handle_id, form) # Optical Node specific updates _handle_location(request.user, optical_node, form.cleaned_data['relationship_location']) _handle_ports(optical_node, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditOpticalNodeForm(optical_node.data) return render(request, 'noclook/edit/edit_optical_node.html', {'node_handle': nh, 'node': optical_node, 'form': form, 'location': location, 'ports': ports})
def edit_port(request, handle_id): nh, port = helpers.get_nh_node(handle_id) parent = port.get_parent() connected_to = port.get_connected_to() parent_categories = ['external-equipment', 'firewall', 'host', 'odf', 'optical-node', 'router', 'switch'] connections_categories = Dropdown.get('cable_types').as_values(False) if request.POST: form = forms.EditPortForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, port.handle_id, form) # Port specific updates if form.cleaned_data['relationship_parent']: parent_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_parent']) helpers.set_has(request.user, parent_nh.get_node(), port.handle_id) if form.cleaned_data['relationship_connected_to']: cable_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_connected_to']) helpers.set_connected_to(request.user, cable_nh.get_node(), port.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditPortForm(port.data) return render(request, 'noclook/edit/edit_port.html', {'node_handle': nh, 'form': form, 'node': port, 'parent': parent, 'connected_to': connected_to, 'parent_categories': parent_categories, 'connections_categories': connections_categories})
def edit_optical_link(request, handle_id): # Get needed data from node nh, link = helpers.get_nh_node(handle_id) relations = link.get_relations() depends_on = link.get_dependencies() if request.POST: form = forms.EditOpticalLinkForm(request.POST) if form.is_valid(): if 'type' in form.cleaned_data: form.cleaned_data['type'] = form.cleaned_data['type'].name # Generic node update helpers.form_update_node(request.user, link.handle_id, form) # Optical Link node updates if form.cleaned_data['relationship_provider']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_provider']) helpers.set_provider(request.user, link, owner_nh.handle_id) if form.cleaned_data['relationship_end_a']: depends_on_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_end_a']) helpers.set_depends_on(request.user, link, depends_on_nh.handle_id) if form.cleaned_data['relationship_end_b']: depends_on_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_end_b']) helpers.set_depends_on(request.user, link, depends_on_nh.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditOpticalLinkForm(link.data) return render(request, 'noclook/edit/edit_optical_link.html', {'node_handle': nh, 'form': form, 'node': link, 'relations': relations, 'depends_on': depends_on})
def edit_external_equipment(request, handle_id): # Get needed data from node nh, external_equipment = helpers.get_nh_node(handle_id) relations = external_equipment.get_relations() location = external_equipment.get_location() ports = external_equipment.get_ports() if request.POST: form = forms.EditExternalEquipmentForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, external_equipment.handle_id, form) # External Equipment specific updates if form.cleaned_data['relationship_owner']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_owner']) helpers.set_owner(request.user, external_equipment, owner_nh.handle_id) _handle_location(request.user, external_equipment, form.cleaned_data['relationship_location']) _handle_ports(external_equipment, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditExternalEquipmentForm(external_equipment.data) return render(request, 'noclook/edit/edit_external_equipment.html', {'node_handle': nh, 'node': external_equipment, 'form': form, 'relations': relations, 'location': location, 'ports': ports})
def edit_rack(request, handle_id): # Get needed data from node nh, rack = helpers.get_nh_node(handle_id) parent = rack.get_parent() located_in = rack.get_located_in() located_in_categories = ['host', 'odf', 'optical-node', 'router'] if request.POST: form = forms.EditRackForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, rack.handle_id, form) # Rack specific updates if form.cleaned_data['relationship_parent']: parent_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_parent']) helpers.set_has(request.user, parent_nh.get_node(), rack.handle_id) if form.cleaned_data['relationship_located_in']: equipment = NodeHandle.objects.get(pk=form.cleaned_data['relationship_located_in']) helpers.set_location(request.user, equipment.get_node(), rack.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditRackForm(rack.data) return render(request, 'noclook/edit/edit_rack.html', {'node_handle': nh, 'form': form, 'parent': parent, 'node': rack, 'located_in': located_in, 'parent_categories': 'site', 'located_in_categories': located_in_categories})
def edit_router(request, handle_id): # Get needed data from node nh, router = helpers.get_nh_node(handle_id) location = router.get_location() if request.POST: form = forms.EditRouterForm(request.POST) ports_form = forms.BulkPortsForm(request.POST) if form.is_valid() and ports_form.is_valid(): # Generic node update helpers.form_update_node(request.user, router.handle_id, form) # Router specific updates _handle_location(request.user, router, form.cleaned_data['relationship_location']) _handle_ports(router, form.cleaned_data['relationship_ports'], request.user) if not ports_form.cleaned_data['no_ports']: data = ports_form.cleaned_data helpers.bulk_create_ports(nh.get_node(), request.user, **data) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditRouterForm(router.data) ports_form = forms.BulkPortsForm({'port_type': 'LC', 'offset': 1, 'num_ports': '0'}) return render(request, 'noclook/edit/edit_router.html', {'node_handle': nh, 'node': router, 'form': form, 'location': location, 'ports_form': ports_form})
def edit_optical_path(request, handle_id): # Get needed data from node nh, path = helpers.get_nh_node(handle_id) relations = path.get_relations() depends_on = path.get_dependencies() dependency_categories = 'odf,optical-link,optical-multiplex-section,optical-node,router,switch,optical-path' if request.POST: form = forms.EditOpticalPathForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, path.handle_id, form) # Optical Path node updates if form.cleaned_data['relationship_provider']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_provider']) helpers.set_provider(request.user, path, owner_nh.handle_id) if form.cleaned_data['relationship_depends_on']: depends_on_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_depends_on']) helpers.set_depends_on(request.user, path, depends_on_nh.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditOpticalPathForm(path.data) return render(request, 'noclook/edit/edit_optical_path.html', {'node_handle': nh, 'form': form, 'node': path, 'relations': relations, 'depends_on': depends_on, 'dependency_categories': dependency_categories})
def edit_cable(request, handle_id): # Get needed data from node nh, cable = helpers.get_nh_node(handle_id) connections = cable.get_connected_equipment() relations = cable.get_relations() if request.POST: form = forms.EditCableForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, cable.handle_id, form) if form.cleaned_data['relationship_end_a']: end_a_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_end_a']) helpers.set_connected_to(request.user, cable, end_a_nh.handle_id) if form.cleaned_data['relationship_end_b']: end_b_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_end_b']) helpers.set_connected_to(request.user, cable, end_b_nh.handle_id) if form.cleaned_data['relationship_provider']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_provider']) helpers.set_provider(request.user, cable, owner_nh.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditCableForm(cable.data) return render(request, 'noclook/edit/edit_cable.html', {'node_handle': nh, 'form': form, 'node': cable, 'connections': connections, 'relations': relations})
def update_rack_position(request, rack_handle_id, handle_id, position): """ Updates the nodes rack_position if node is placed in rack """ if request.method == 'POST': success = False nh, node = helpers.get_nh_node(handle_id) if nh.node_meta_type == 'Physical': helpers.dict_update_node(request.user, node.handle_id, {'rack_position': int(position)}) success = True return JsonResponse({'success': success, 'position': '{}'.format(position), 'handle_id': node.handle_id}) else: return Http404()
def edit_host(request, handle_id): # Get needed data from node nh, host = helpers.get_nh_node(handle_id) location = host.get_location() relations = host.get_relations() depends_on = host.get_dependencies() host_services = host.get_host_services() ports = host.get_ports() dependency_categories = 'service,host' if request.POST: form = forms.EditHostForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, host.handle_id, form) # Host specific updates if form.cleaned_data['relationship_user']: user_nh = _nh_safe_get(form.cleaned_data['relationship_user']) if user_nh: helpers.set_user(request.user, host, user_nh.handle_id) if form.cleaned_data['relationship_owner']: owner_nh = _nh_safe_get(form.cleaned_data['relationship_owner']) if owner_nh: helpers.set_owner(request.user, host, owner_nh.handle_id) # You can not set location and depends on at the same time if form.cleaned_data['relationship_depends_on']: depends_on_nh = _nh_safe_get(form.cleaned_data['relationship_depends_on']) if depends_on_nh: helpers.set_depends_on(request.user, host, depends_on_nh.handle_id) elif form.cleaned_data['relationship_location']: _handle_location(request.user, host, form.cleaned_data['relationship_location']) if form.cleaned_data['services_locked'] and form.cleaned_data['services_checked']: helpers.remove_rogue_service_marker(request.user, host.handle_id) _handle_ports(host, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditHostForm(host.data) context = { 'node_handle': nh, 'node': host, 'form': form, 'location': location, 'relations': relations, 'depends_on': depends_on, 'ports': ports, 'host_services': host_services, 'dependency_categories': dependency_categories } return render(request, 'noclook/edit/edit_host.html', context)
def edit_pdu(request, handle_id): # Get needed data from node nh, pdu = helpers.get_nh_node(handle_id) location = pdu.get_location() relations = pdu.get_relations() depends_on = pdu.get_dependencies() host_services = pdu.get_host_services() ports = pdu.get_ports() dependency_categories = 'service' ports_form = forms.BulkPortsForm(request.POST or None) if request.POST: form = forms.EditPDUForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, pdu.handle_id, form) # Host specific updates if form.cleaned_data['relationship_user']: user_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_user']) helpers.set_user(request.user, pdu, user_nh.handle_id) if form.cleaned_data['relationship_owner']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_owner']) helpers.set_owner(request.user, pdu, owner_nh.handle_id) # You can not set location and depends on at the same time if form.cleaned_data['relationship_depends_on']: depends_on_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_depends_on']) helpers.set_depends_on(request.user, pdu, depends_on_nh.handle_id) elif form.cleaned_data['relationship_location']: _handle_location(request.user, pdu, form.cleaned_data['relationship_location']) if form.cleaned_data['services_locked'] and form.cleaned_data['services_checked']: helpers.remove_rogue_service_marker(request.user, pdu.handle_id) if ports_form.is_valid() and not ports_form.cleaned_data['no_ports']: data = ports_form.cleaned_data helpers.bulk_create_ports(pdu, request.user, **data) _handle_ports(pdu, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditPDUForm(pdu.data) return render(request, 'noclook/edit/edit_pdu.html', {'node_handle': nh, 'node': pdu, 'form': form, 'location': location, 'relations': relations, 'depends_on': depends_on, 'ports': ports, 'host_services': host_services, 'dependency_categories': dependency_categories, 'ports_form': ports_form})
def edit_end_user(request, handle_id): # Get needed data from node nh, end_user = helpers.get_nh_node(handle_id) if request.POST: form = forms.EditEndUserForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, end_user.handle_id, form) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditEndUserForm(end_user.data) return render(request, 'noclook/edit/edit_end_user.html', {'node_handle': nh, 'form': form, 'node': end_user})
def delete_relationship(request, slug, handle_id, rel_id): """ Removes the relationship if the node has a relationship matching the supplied id. """ success = False if request.method == 'POST': nh, node = helpers.get_nh_node(handle_id) try: relationship = nc.get_relationship_model(nc.graphdb.manager, rel_id) if node.handle_id == relationship.start['handle_id'] or node.handle_id == relationship.end['handle_id']: activitylog.delete_relationship(request.user, relationship) relationship.delete() success = True except nc.exceptions.RelationshipNotFound: success = True return JsonResponse({'success': success, 'relationship_id': '{}'.format(rel_id)})
def edit_peering_group(request, handle_id): # Get needed data from node nh, peering_group = helpers.get_nh_node(handle_id) depends_on = peering_group.get_dependencies() if request.POST: form = forms.EditPeeringGroupForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, peering_group.handle_id, form) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditPeeringGroupForm(peering_group.data) return render(request, 'noclook/edit/edit_peering_group.html', {'node_handle': nh, 'node': peering_group, 'form': form, 'depends_on': depends_on})
def delete_node(request, slug, handle_id): """ Removes the node and all its relationships. """ redirect_url = '/{}'.format(slug) nh, node = helpers.get_nh_node(handle_id) if nh.node_type.get_label() == 'Unit': part_of = node.get_part_of() if part_of.get('Part_of'): redirect_url = helpers.get_node_url(part_of.get('Part_of')[0]['node'].handle_id) try: # Redirect to parent if deleted node was a child node parent = node.get_parent().get('Has', []) if parent: redirect_url = helpers.get_node_url(parent[0]['node'].handle_id) except AttributeError: pass helpers.delete_node(request.user, node.handle_id) return redirect(redirect_url)
def edit_service(request, handle_id): # Get needed data from node nh, service = helpers.get_nh_node(handle_id) relations = service.get_relations() depends_on = service.get_dependencies() dependency_categories = ['host', 'firewall', 'odf', 'optical-node', 'optical-path', 'optical-filter', 'router', 'service', 'switch', 'external-equipment'] user_categories = ['customer', 'end-user'] if request.POST: form = forms.EditServiceForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, service.handle_id, form) # Service node updates if form.cleaned_data['relationship_provider']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_provider']) helpers.set_provider(request.user, service, owner_nh.handle_id) if form.cleaned_data['relationship_user']: user_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_user']) helpers.set_user(request.user, service, user_nh.handle_id) if form.cleaned_data['relationship_depends_on']: depends_on_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_depends_on']) helpers.set_depends_on(request.user, service, depends_on_nh.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditServiceForm(service.data) return render(request, 'noclook/edit/edit_service.html', {'node_handle': nh, 'form': form, 'node': service, 'relations': relations, 'depends_on': depends_on, 'dependency_categories': dependency_categories, 'user_categories': user_categories})
def edit_site(request, handle_id): # Get needed data from node nh, site = helpers.get_nh_node(handle_id) relations = site.get_relations() if request.POST: form = forms.EditSiteForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, site.handle_id, form) # Set site owner if form.cleaned_data['relationship_responsible_for']: responsible_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_responsible_for']) helpers.set_responsible_for(request.user, site, responsible_nh.handle_id) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditSiteForm(site.data) return render(request, 'noclook/edit/edit_site.html', {'node_handle': nh, 'form': form, 'relations': relations, 'node': site})
def edit_optical_fillter(request, handle_id): nh, of = helpers.get_nh_node(handle_id) location = of.get_location() ports = of.get_ports() form = forms.EditOpticalFilterForm(request.POST or of.data) if request.POST and form.is_valid(): # Generic node update helpers.form_update_node(request.user, of.handle_id, form) # Optical Filter specific updates _handle_location(request.user, of, form.cleaned_data['relationship_location']) _handle_ports(of, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) return render(request, 'noclook/edit/edit_optical_filter.html', {'node_handle': nh, 'node': of, 'form': form, 'location': location, 'ports': ports})
def update_relationship(request, slug, handle_id, rel_id): """ Removes the relationship if the node has a relationship matching the supplied id. """ success = False properties = {} if request.POST: nh, node = helpers.get_nh_node(handle_id) try: for key, value in request.POST.items(): properties[key] = json.loads(value) relationship = nc.get_relationship_model(nc.graphdb.manager, int(rel_id)) if node.handle_id == relationship.start['handle_id'] or node.handle_id == relationship.end['handle_id']: success = helpers.dict_update_relationship(request.user, relationship.id, properties) except nc.exceptions.RelationshipNotFound: # If the relationship does not exist, then we cannot update success = False except ValueError: pass return JsonResponse({'success': success, 'relationship_id': '{}'.format(rel_id), 'data': properties})
def edit_firewall(request, handle_id): # Get needed data from node nh, firewall = helpers.get_nh_node(handle_id) location = firewall.get_location() relations = firewall.get_relations() host_services = firewall.get_host_services() ports = firewall.get_ports() if request.POST: form = forms.EditFirewallForm(request.POST) if form.is_valid(): # Generic node update helpers.form_update_node(request.user, firewall.handle_id, form) # Firewall specific updates if form.cleaned_data['relationship_user']: user_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_user']) helpers.set_user(request.user, firewall, user_nh.handle_id) if form.cleaned_data['relationship_owner']: owner_nh = NodeHandle.objects.get(pk=form.cleaned_data['relationship_owner']) helpers.set_owner(request.user, firewall, owner_nh.handle_id) _handle_location(request.user, firewall, form.cleaned_data['relationship_location']) if form.cleaned_data['services_locked'] and form.cleaned_data['services_checked']: helpers.remove_rogue_service_marker(request.user, firewall.handle_id) _handle_ports(firewall, form.cleaned_data['relationship_ports'], request.user) if 'saveanddone' in request.POST: return redirect(nh.get_absolute_url()) else: return redirect('%sedit' % nh.get_absolute_url()) else: form = forms.EditFirewallForm(firewall.data) return render(request, 'noclook/edit/edit_firewall.html', {'node_handle': nh, 'node': firewall, 'form': form, 'location': location, 'relations': relations, 'ports': ports, 'host_services': host_services})
def do_request(cls, request, **kwargs): form_class = kwargs.get('form_class') nimetaclass = getattr(cls, 'NIMetaClass') graphql_type = getattr(nimetaclass, 'graphql_type') nimetatype = getattr(graphql_type, 'NIMetaType') node_type = getattr(nimetatype, 'ni_type').lower() node_meta_type = getattr(nimetatype, 'ni_metatype').capitalize() id = request.POST.get('id') has_error = False # check authorization handle_id = relay.Node.from_global_id(id)[1] authorized = sriutils.authorice_write_resource(request.user, handle_id) if not authorized: raise GraphQLAuthException() # Get needed data from node nh, organization = helpers.get_nh_node(handle_id) relations = organization.get_relations() out_relations = organization.get_outgoing_relations() if request.POST: # set handle_id into POST data and remove relay id post_data = request.POST.copy() post_data.pop('id') post_data.update({'handle_id': handle_id}) # replace relay ids for handle_id in contacts if present for field, roledict in DEFAULT_ROLES.items(): if field in post_data: handle_id = post_data.get(field) handle_id = relay.Node.from_global_id(handle_id)[1] post_data.pop(field) post_data.update({field: handle_id}) relay_extra_ids = ('relationship_parent_of', 'relationship_uses_a') for field in relay_extra_ids: handle_id = post_data.get(field) if handle_id: handle_id = relay.Node.from_global_id(handle_id)[1] post_data.pop(field) post_data.update({field: handle_id}) form = form_class(post_data) form.strict_validation = True if form.is_valid(): # Generic node update # use property keys to avoid inserting contacts as a string property of the node property_keys = [ 'name', 'description', 'organization_id', 'type', 'incident_management_info', 'affiliation_customer', 'affiliation_end_customer', 'affiliation_provider', 'affiliation_partner', 'affiliation_host_user', 'affiliation_site_owner', 'website', 'organization_number' ] helpers.form_update_node(request.user, organization.handle_id, form, property_keys) # specific role setting for field, roledict in DEFAULT_ROLES.items(): if field in form.cleaned_data: contact_id = form.cleaned_data[field] role = RoleModel.objects.get(slug=field) set_contact = helpers.get_contact_for_orgrole(organization.handle_id, role) if contact_id: if set_contact: if set_contact.handle_id != contact_id: helpers.unlink_contact_with_role_from_org(request.user, organization, role) helpers.link_contact_role_for_organization(request.user, organization, contact_id, role) else: helpers.link_contact_role_for_organization(request.user, organization, contact_id, role) elif set_contact: helpers.unlink_contact_and_role_from_org(request.user, organization, set_contact.handle_id, role) # Set child organizations if form.cleaned_data['relationship_parent_of']: organization_nh = NodeHandle.objects.get(handle_id=form.cleaned_data['relationship_parent_of']) helpers.set_parent_of(request.user, organization, organization_nh.handle_id) if form.cleaned_data['relationship_uses_a']: procedure_nh = NodeHandle.objects.get(handle_id=form.cleaned_data['relationship_uses_a']) helpers.set_uses_a(request.user, organization, procedure_nh.handle_id) return has_error, { graphql_type.__name__.lower(): nh } else: # get the errors and return them has_error = True errordict = cls.format_error_array(form.errors) return has_error, errordict else: # get the errors and return them has_error = True errordict = cls.format_error_array(form.errors) return has_error, errordict
def do_request(cls, request, **kwargs): form_class = kwargs.get('form_class') nimetaclass = getattr(cls, 'NIMetaClass') graphql_type = getattr(nimetaclass, 'graphql_type') nimetatype = getattr(graphql_type, 'NIMetaType') node_type = getattr(nimetatype, 'ni_type').lower() node_meta_type = getattr(nimetatype, 'ni_metatype').capitalize() context_method = getattr(nimetatype, 'context_method') has_error = False context = context_method() # check it can write on this context authorized = sriutils.authorize_create_resource(request.user, context) if not authorized: raise GraphQLAuthException() # Get needed data from node if request.POST: # replace relay ids for handle_id in contacts if present post_data = request.POST.copy() for field, roledict in DEFAULT_ROLES.items(): if field in post_data: handle_id = post_data.get(field) handle_id = relay.Node.from_global_id(handle_id)[1] post_data.pop(field) post_data.update({field: handle_id}) relay_extra_ids = ('relationship_parent_of', 'relationship_uses_a') for field in relay_extra_ids: handle_id = post_data.get(field) if handle_id: try: handle_id = relay.Node.from_global_id(handle_id)[1] post_data.pop(field) post_data.update({field: handle_id}) except BinasciiError: pass # the id is already in handle_id format form = form_class(post_data) form.strict_validation = True if form.is_valid(): try: nh = helpers.form_to_generic_node_handle(request, form, node_type, node_meta_type, context) except UniqueNodeError: has_error = True return has_error, [ErrorType(field="_", messages=["A {} with that name already exists.".format(node_type)])] # Generic node update # use property keys to avoid inserting contacts as a string property of the node property_keys = [ 'name', 'description', 'organization_id', 'type', 'incident_management_info', 'affiliation_customer', 'affiliation_end_customer', 'affiliation_provider', 'affiliation_partner', 'affiliation_host_user', 'affiliation_site_owner', 'website', 'organization_number' ] helpers.form_update_node(request.user, nh.handle_id, form, property_keys) nh_reload, organization = helpers.get_nh_node(nh.handle_id) # add default context NodeHandleContext(nodehandle=nh, context=context).save() # specific role setting for field, roledict in DEFAULT_ROLES.items(): if field in form.cleaned_data: contact_id = form.cleaned_data[field] role = RoleModel.objects.get(slug=field) set_contact = helpers.get_contact_for_orgrole(organization.handle_id, role) if contact_id: if set_contact: if set_contact.handle_id != contact_id: helpers.unlink_contact_with_role_from_org(request.user, organization, role) helpers.link_contact_role_for_organization(request.user, organization, contact_id, role) else: helpers.link_contact_role_for_organization(request.user, organization, contact_id, role) elif set_contact: helpers.unlink_contact_and_role_from_org(request.user, organization, set_contact.handle_id, role) # Set child organizations if form.cleaned_data['relationship_parent_of']: organization_nh = NodeHandle.objects.get(handle_id=form.cleaned_data['relationship_parent_of']) helpers.set_parent_of(request.user, organization, organization_nh.handle_id) if form.cleaned_data['relationship_uses_a']: procedure_nh = NodeHandle.objects.get(handle_id=form.cleaned_data['relationship_uses_a']) helpers.set_uses_a(request.user, organization, procedure_nh.handle_id) return has_error, { graphql_type.__name__.lower(): nh } else: # get the errors and return them has_error = True errordict = cls.format_error_array(form.errors) return has_error, errordict else: # get the errors and return them has_error = True errordict = cls.format_error_array(form.errors) return has_error, errordict