api architecture

djadhere/decorators.py

@@ -0,0 +1,13 @@
+from django.core.exceptions import PermissionDenied
+from django.conf import settings
+
+from functools import wraps
+
+
+def api_key_required(view_func):
+    def wrapped_view(request, **kwargs):
+        key = request.GET.get('key')
+        if key != settings.API_KEY:
+            raise PermissionDenied
+        return view_func(request, **kwargs)
+    return wraps(view_func)(wrapped_view)

services/management/commands/architecture.py

@@ -1,6 +1,6 @@
 from django.core.management.base import BaseCommand, CommandParser
 
-from services.models import IPResource, IPPrefix, Service, ServiceType, ServiceAllocation, Switch
+from services.utils.architecture import export_switch, export_ip
 from djadhere.utils import get_active_filter
 
 
@@ -16,50 +16,7 @@ class Command(BaseCommand):
             self.export_ip()
     
     def export_switch(self):
-        for switch in Switch.objects.all():
-            print()
-            print("=== Switch %s ===" % switch.name)
-            print(switch.notes)
-            for port in switch.ports.all():
-                line = "Port %d" % port.port
-                if port.service:
-                    line += ' - (%s %s) %s' % (port.service.adhesion, port.service.adhesion.adherent, port.service)
-                if port.notes:
-                    line += ' - %s' % port.notes
-                print(line)
+        print(export_switch())
 
     def export_ip(self):
-        public_ips = IPResource.objects.filter(category=IPResource.CATEGORY_PUBLIC).values_list('pk', flat=True)
-        for prefix in IPPrefix.objects.filter(ipresource__in=public_ips).distinct():
-            print("\n=== %s ===\n" % prefix)
-            active_ips = prefix.ipresource_set.filter(in_use=True)
-            if not active_ips.exists():
-                print("empty")
-                continue
-            elif active_ips.count() < 20:
-                ip_set = active_ips
-            else:
-                ip_set = prefix.ipresource_set
-            for ip in ip_set.all():
-                line = str(ip)
-                if ip.reserved:
-                    line += ' - réservé'
-                else:
-                    try:
-                        allocation = ip.allocations.get(active=True)
-                    except ServiceAllocation.DoesNotExist:
-                        line += ' - disponible'
-                        allocation = ip.allocations.order_by('-end').first()
-                        if allocation:
-                            service = allocation.service
-                            end = allocation.end.strftime('%Y%m%d')
-                            adhesion = service.adhesion
-                            adherent = adhesion.adherent
-                            line += ' depuis %s (dernière utilisation : %s %s %s)' % (end, adhesion, adherent, service)
-                    else:
-                        service = allocation.service
-                        start = allocation.start.strftime('%Y%m%d')
-                        adhesion = service.adhesion
-                        adherent = adhesion.adherent
-                        line += ' - %s %s %s (depuis %s)' % (adhesion, adherent, service, start)
-                print(line)
+        print(export_ip())

services/urls.py

@@ -7,4 +7,6 @@ urlpatterns = [
     url(r'^services/(?P<pk>[0-9]+)/$', views.ServiceDetail.as_view(), name='service-detail'),
     url(r'^api/fastpinger/$', views.fastpinger, name='fastpinger'),
     url(r'^api/routes/(?:(?P<route>[\w-]+)/)?$', views.routes, name='routes'),
+    url(r'^api/architecture/switch/$', views.architecture_switch, name='archi-switch'),
+    url(r'^api/architecture/ip/$', views.architecture_ip, name='archi-ip'),
 ]

services/utils/architecture.py

@@ -0,0 +1,85 @@
+from django.db.models.functions import Concat
+from django.db import models
+
+from services.models import IPResource, IPPrefix, Service, ServiceType, ServiceAllocation, Switch
+from djadhere.utils import get_active_filter
+
+from textwrap import indent
+
+
+def export_switch():
+    output = []
+    for switch in Switch.objects.prefetch_related(
+                            'ports', 'ports__service', 'ports__service__service_type', 'ports__service__adhesion',
+                            'ports__service__adhesion__user', 'ports__service__adhesion__corporation').all():
+        sw_output = "=== Switch %s ===\n" % switch.name
+        sw_output += indent(switch.notes, "# ") + "\n"
+        ports_output = []
+        for port in switch.ports.all():
+            line = "Port %d" % port.port
+            if port.service:
+                line += ' - (%s %s) %s' % (port.service.adhesion, port.service.adhesion.adherent, port.service)
+            if port.notes:
+                line += ' - %s' % port.notes
+            ports_output += [line]
+        sw_output += "\n".join(ports_output)
+        output += [sw_output]
+    return "\n\n".join(output)
+
+
+def export_ip():
+    output = []
+    for prefix in IPPrefix.objects.filter(ipresource__category=IPResource.CATEGORY_PUBLIC).order_by('prefix').distinct():
+        prefix_output = ["=== %s ===" % prefix]
+        active_ips = prefix.ipresource_set.filter(in_use=True)
+        if not active_ips.exists():
+            prefix_output += ["empty"]
+            output += ["\n".join(prefix_output)]
+            continue
+        elif active_ips.count() < 20:
+            ip_set = active_ips
+        else:
+            ip_set = prefix.ipresource_set
+        ip_set = ip_set.annotate(
+            allocation=models.Subquery(
+                ServiceAllocation.objects.filter(
+                    models.Q(resource=models.OuterRef('pk')) & get_active_filter(),
+                ).values('pk')[:1],
+            ),
+            service=models.Subquery(
+                Service.objects.filter(
+                    models.Q(allocation__resource=models.OuterRef('pk')) & get_active_filter('allocation'),
+                ).annotate(
+                    adherent=models.Case(
+                        models.When(adhesion__user__isnull=False, then=Concat('adhesion__user__first_name', models.Value(' '), 'adhesion__user__last_name')),
+                        models.When(adhesion__corporation__isnull=False, then='adhesion__corporation__social_reason'),
+                    ),
+                    fullname=Concat(
+                        models.Value('ADT'),
+                        'adhesion_id',
+                        models.Value(' '),
+                        'adherent',
+                        models.Value(' #'),
+                        'pk',
+                        models.Value(' '),
+                        'service_type__name',
+                        models.Value(' '),
+                        'label',
+                        output_field=models.CharField(),
+                    ),
+                ).values('fullname')[:1],
+                output_field=models.CharField(),
+            ),
+            status=models.Case(
+                models.When(reserved=True, then=models.Value('réservé')),
+                models.When(allocation__isnull=True, then=models.Value('disponible')),
+                default='service',
+                output_field=models.CharField(),
+            ),
+        )
+        for ip in ip_set.all():
+            line = str(ip)
+            line += ' - ' + ip.status
+            prefix_output += [line]
+        output += ["\n".join(prefix_output)]
+    return "\n\n".join(output)

services/views.py

@@ -9,8 +9,10 @@ from django.db import models
 from django.db.models.functions import Concat
 
 from djadhere.utils import get_active_filter
+from djadhere.decorators import api_key_required
 from .models import Service, Route, IPResource, ServiceAllocation
 from .utils.fastpinger import fastpinger_update
+from .utils.architecture import export_switch, export_ip
 
 
 class ServiceDetail(LoginRequiredMixin, DetailView):
@@ -31,8 +33,9 @@ def fastpinger(request):
 
 
 @require_GET
+@api_key_required
 def routes(request, route=None):
-    resources = IPResource.objects.filter(category=0) # IP publics
+    resources = IPResource.objects.filter(category=IPResource.CATEGORY_PUBLIC)
     if route:
         route = get_object_or_404(Route, name=route)
         resources = resources.filter(pk__in=route.get_ip())
@@ -63,4 +66,16 @@ def routes(request, route=None):
     )
     resources = resources.order_by('order', 'ip')
     content = '\n'.join(resources.values_list('route', flat=True)) + '\n'
-    return HttpResponse(content, content_type='text/plain')
+    return HttpResponse(content, content_type='text/plain; charset=utf-8')
+
+
+@require_GET
+@api_key_required
+def architecture_switch(request):
+    return HttpResponse(export_switch(), content_type='text/plain; charset=utf-8')
+
+
+@require_GET
+@api_key_required
+def architecture_ip(request):
+    return HttpResponse(export_ip(), content_type='text/plain; charset=utf-8')