add new application datasource. provides basic interface for cloud-init tool (hostname, root ssh authorized keys)

datasource/views.py

@@ -0,0 +1,64 @@
+from django.shortcuts import render
+from django.http import HttpResponse, Http404
+from accounts.models import UserInstance, UserSSHKey
+import json
+import socket
+
+OS_VERSIONS = [ 'latest', '' ]
+OS_UUID = "iid-dswebvirtcloud"
+
+def os_index(request):
+    response = '\n'.join(OS_VERSIONS)
+    return HttpResponse(response)
+
+def os_metadata_json(request, version):
+    """
+    :param request:
+    :param version:
+    :return:
+    """
+
+    if version == 'latest':
+        ip = get_client_ip(request)
+        hostname = get_hostname_by_ip(ip)
+        response = { 'uuid': OS_UUID, 'hostname': hostname }
+        return HttpResponse(json.dumps(response))
+    else:
+        err = 'Invalid version: %s' % version
+        raise Http404(err)
+
+def os_userdata(request, version):
+    """
+    :param request:
+    :param version:
+    :return:
+    """
+    if version == 'latest':
+        ip = get_client_ip(request)
+        hostname = get_hostname_by_ip(ip)
+        vname = hostname.split('.')[0]
+        
+        instance_keys = []
+        userinstances = UserInstance.objects.filter(instance__name=vname)
+        
+        for ui in userinstances:
+            keys = UserSSHKey.objects.filter(user=ui.user)
+            for k in keys:
+                instance_keys.append(k.keypublic)
+
+        return render(request, 'user_data', locals())
+    else:
+        err = 'Invalid version: %s' % version
+        raise Http404(err)
+
+def get_client_ip(request):
+    x_forwarded_for = request.META.get('HTTP_X_FORWARDED_FOR')
+    if x_forwarded_for:
+        ip = x_forwarded_for.split(',')[-1].strip()
+    else:
+        ip = request.META.get('REMOTE_ADDR')
+    return ip
+
+def get_hostname_by_ip(ip):
+    addrs = socket.gethostbyaddr(ip)
+    return addrs[0]