mirror of
https://github.com/retspen/webvirtcloud
synced 2024-12-25 23:55:24 +00:00
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
|
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]
|