ungleich-public/dynamicweb
6
0
Fork 0
Code Issues 1 Pull requests 2 Projects Releases Wiki Activity

Add UserHosting public key to opennebula user

Browse source
This commit is contained in:
modulos 2017-05-25 11:27:49 +02:00
parent fb8dac7d00
commit 9753dfd2fc
2 changed files with 110 additions and 88 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

51
hosting/views.py
View file

@ -40,6 +40,7 @@ from oca.pool import WrongNameError
CONNECTION_ERROR = "Your VMs cannot be displayed at the moment due to a backend \ CONNECTION_ERROR = "Your VMs cannot be displayed at the moment due to a backend \
connection error. please try again in a few minutes." connection error. please try again in a few minutes."
class DjangoHostingView(ProcessVMSelectionMixin, View): class DjangoHostingView(ProcessVMSelectionMixin, View):
template_name = "hosting/django.html" template_name = "hosting/django.html"
@ -189,7 +190,8 @@ class SignupView(CreateView):
model = CustomUser model = CustomUser
def get_success_url(self): def get_success_url(self):
next_url = self.request.session.get('next', reverse_lazy('hosting:virtual_machines')) next_url = self.request.session.get(
'next', reverse_lazy('hosting:virtual_machines'))
return next_url return next_url
def form_valid(self, form): def form_valid(self, form):
@ -243,12 +245,14 @@ class PasswordResetConfirmView(PasswordResetConfirmViewMixin):
return self.form_valid(form) return self.form_valid(form)
else: else:
messages.error(request, 'Password reset has not been successful.') messages.error(
request, 'Password reset has not been successful.')
form.add_error(None, 'Password reset has not been successful.') form.add_error(None, 'Password reset has not been successful.')
return self.form_invalid(form) return self.form_invalid(form)
else: else:
messages.error(request, 'The reset password link is no longer valid.') messages.error(
request, 'The reset password link is no longer valid.')
form.add_error(None, 'The reset password link is no longer valid.') form.add_error(None, 'The reset password link is no longer valid.')
return self.form_invalid(form) return self.form_invalid(form)
@ -337,6 +341,20 @@ class GenerateVMSSHKeysView(LoginRequiredMixin, FormView):
'form': UserHostingKeyForm(request=self.request), 'form': UserHostingKeyForm(request=self.request),
}) })
owner = self.request.user
# Create OpenNebulaManager
manager = OpenNebulaManager(email=owner.email,
password=owner.password)
# Get OpenNebula user id
user_pool = manager._get_user_pool()
opennebula_user = user_pool.get_by_name(owner.email)
# Get user ssh key
user_key = UserHostingKey.objects.get(user=owner)
# Add ssh key to user
manager.oneadmin_client.call('user.update', opennebula_user.id,
'<CONTEXT><SSH_PUBLIC_KEY>{ssh_key}</SSH_PUBLIC_KEY></CONTEXT>'.format(ssh_key=user_key.public_key))
# return HttpResponseRedirect(reverse('hosting:key_pair')) # return HttpResponseRedirect(reverse('hosting:key_pair'))
return render(self.request, self.template_name, context) return render(self.request, self.template_name, context)
@ -385,9 +403,11 @@ class PaymentVMView(LoginRequiredMixin, FormView):
user = self.request.user user = self.request.user
# Get user last order # Get user last order
last_hosting_order = HostingOrder.objects.filter(customer__user=user).last() last_hosting_order = HostingOrder.objects.filter(
customer__user=user).last()
# If user has already an hosting order, get the credit card data from it # If user has already an hosting order, get the credit card data from
# it
if last_hosting_order: if last_hosting_order:
credit_card_data = last_hosting_order.get_cc_data() credit_card_data = last_hosting_order.get_cc_data()
context.update({ context.update({
@ -476,11 +496,10 @@ class PaymentVMView(LoginRequiredMixin, FormView):
except UserHostingKey.DoesNotExist: except UserHostingKey.DoesNotExist:
pass pass
# Create a vm using logged user # Create a vm using logged user
vm_id = manager.create_vm( vm_id = manager.create_vm(
template_id=vm_template_id, template_id=vm_template_id,
#XXX: Confi # XXX: Confi
specs=specs, specs=specs,
ssh_key=user_key.public_key, ssh_key=user_key.public_key,
) )
@ -494,14 +513,16 @@ class PaymentVMView(LoginRequiredMixin, FormView):
) )
# Create a Hosting Bill # Create a Hosting Bill
bill = HostingBill.create(customer=customer, billing_address=billing_address) bill = HostingBill.create(
customer=customer, billing_address=billing_address)
# Create Billing Address for User if he does not have one # Create Billing Address for User if he does not have one
if not customer.user.billing_addresses.count(): if not customer.user.billing_addresses.count():
billing_address_data.update({ billing_address_data.update({
'user': customer.user.id 'user': customer.user.id
}) })
billing_address_user_form = UserBillingAddressForm(billing_address_data) billing_address_user_form = UserBillingAddressForm(
billing_address_data)
billing_address_user_form.is_valid() billing_address_user_form.is_valid()
billing_address_user_form.save() billing_address_user_form.save()
@ -553,7 +574,7 @@ class OrdersHostingDetailView(PermissionRequiredMixin, LoginRequiredMixin, Detai
vm = manager.get_vm(obj.vm_id) vm = manager.get_vm(obj.vm_id)
context['vm'] = VirtualMachineSerializer(vm).data context['vm'] = VirtualMachineSerializer(vm).data
except ConnectionRefusedError: except ConnectionRefusedError:
messages.error( request, messages.error(request,
'In order to create a VM, you need to create/upload your SSH KEY first.' 'In order to create a VM, you need to create/upload your SSH KEY first.'
) )
return context return context
@ -595,7 +616,7 @@ class VirtualMachinesPlanListView(LoginRequiredMixin, ListView):
serializer = VirtualMachineSerializer(queryset, many=True) serializer = VirtualMachineSerializer(queryset, many=True)
return serializer.data return serializer.data
except ConnectionRefusedError: except ConnectionRefusedError:
messages.error( self.request, messages.error(self.request,
'We could not load your VMs due to a backend connection \ 'We could not load your VMs due to a backend connection \
error. Please try again in a few minutes' error. Please try again in a few minutes'
) )
@ -603,12 +624,11 @@ class VirtualMachinesPlanListView(LoginRequiredMixin, ListView):
self.kwargs['error'] = 'connection' self.kwargs['error'] = 'connection'
return [] return []
def get_context_data(self, **kwargs): def get_context_data(self, **kwargs):
error = self.kwargs.get('error') error = self.kwargs.get('error')
if error is not None: if error is not None:
print(error) print(error)
context = { 'error' : 'connection' } context = {'error': 'connection'}
else: else:
context = super(ListView, self).get_context_data(**kwargs) context = super(ListView, self).get_context_data(**kwargs)
return context return context
@ -658,7 +678,8 @@ class CreateVirtualMachinesView(LoginRequiredMixin, View):
template = manager.get_template(template_id) template = manager.get_template(template_id)
configuration_id = int(request.POST.get('configuration')) configuration_id = int(request.POST.get('configuration'))
configuration = HostingPlan.objects.get(id=configuration_id) configuration = HostingPlan.objects.get(id=configuration_id)
request.session['template'] = VirtualMachineTemplateSerializer(template).data request.session['template'] = VirtualMachineTemplateSerializer(
template).data
request.session['specs'] = configuration.serialize() request.session['specs'] = configuration.serialize()
return redirect(reverse('hosting:payment')) return redirect(reverse('hosting:payment'))
@ -680,7 +701,7 @@ class VirtualMachineView(LoginRequiredMixin, View):
vm = manager.get_vm(vm_id) vm = manager.get_vm(vm_id)
return vm return vm
except ConnectionRefusedError: except ConnectionRefusedError:
messages.error( self.request, messages.error(self.request,
'We could not load your VM due to a backend connection \ 'We could not load your VM due to a backend connection \
error. Please try again in a few minutes' error. Please try again in a few minutes'
) )

31
opennebula_api/models.py
View file

@ -10,6 +10,7 @@ from oca.pool import WrongNameError
from oca.exceptions import OpenNebulaException from oca.exceptions import OpenNebulaException
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class OpenNebulaManager(): class OpenNebulaManager():
"""This class represents an opennebula manager.""" """This class represents an opennebula manager."""
@ -66,6 +67,7 @@ class OpenNebulaManager():
protocol=settings.OPENNEBULA_PROTOCOL) protocol=settings.OPENNEBULA_PROTOCOL)
) )
raise ConnectionRefusedError raise ConnectionRefusedError
def _get_user_pool(self): def _get_user_pool(self):
try: try:
user_pool = oca.UserPool(self.oneadmin_client) user_pool = oca.UserPool(self.oneadmin_client)
@ -108,7 +110,6 @@ class OpenNebulaManager():
except ConnectionRefusedError: except ConnectionRefusedError:
raise ConnectionRefusedError raise ConnectionRefusedError
def get_vm(self, vm_id): def get_vm(self, vm_id):
vm_id = int(vm_id) vm_id = int(vm_id)
try: try:
@ -118,7 +119,7 @@ class OpenNebulaManager():
raise ConnectionRefusedError raise ConnectionRefusedError
def create_template(self, name, cores, memory, disk_size, core_price, memory_price, def create_template(self, name, cores, memory, disk_size, core_price, memory_price,
disk_size_price, ssh='' ): disk_size_price, ssh=''):
"""Create and add a new template to opennebula. """Create and add a new template to opennebula.
:param name: A string representation describing the template. :param name: A string representation describing the template.
Used as label in view. Used as label in view.
@ -136,11 +137,11 @@ class OpenNebulaManager():
template_string_formatter.format( template_string_formatter.format(
name=name, name=name,
vcpu=cores, vcpu=cores,
cpu=0.1*cores, cpu=0.1 * cores,
size=1024 * disk_size, size=1024 * disk_size,
memory=1024 * memory, memory=1024 * memory,
# * 10 because we set cpu to *0.1 # * 10 because we set cpu to *0.1
cpu_cost=10*core_price, cpu_cost=10 * core_price,
memory_cost=memory_price, memory_cost=memory_price,
disk_cost=disk_size_price, disk_cost=disk_size_price,
ssh=ssh ssh=ssh
@ -160,7 +161,7 @@ class OpenNebulaManager():
image_id = disk.image_id image_id = disk.image_id
vm_specs = vm_specs_formatter.format( vm_specs = vm_specs_formatter.format(
vcpu=int(specs['cpu']), vcpu=int(specs['cpu']),
cpu=0.1* int(specs['cpu']), cpu=0.1 * int(specs['cpu']),
memory=1024 * int(specs['memory']), memory=1024 * int(specs['memory']),
) )
@ -181,7 +182,7 @@ class OpenNebulaManager():
vm_specs = vm_specs_formatter.format( vm_specs = vm_specs_formatter.format(
vcpu=int(specs['cpu']), vcpu=int(specs['cpu']),
cpu=0.1* int(specs['cpu']), cpu=0.1 * int(specs['cpu']),
memory=1024 * int(specs['memory']), memory=1024 * int(specs['memory']),
) )
@ -219,7 +220,8 @@ class OpenNebulaManager():
self.opennebula_user.group_ids[0] self.opennebula_user.group_ids[0]
) )
except AttributeError: except AttributeError:
logger.info('Could not change owner for vm with id: {}.'.format(vm_id)) logger.info(
'Could not change owner for vm with id: {}.'.format(vm_id))
self.oneadmin_client.call( self.oneadmin_client.call(
oca.VirtualMachine.METHODS['action'], oca.VirtualMachine.METHODS['action'],
@ -241,7 +243,8 @@ class OpenNebulaManager():
except socket.timeout as socket_err: except socket.timeout as socket_err:
logger.info("Socket timeout error: {0}".format(socket_err)) logger.info("Socket timeout error: {0}".format(socket_err))
except OpenNebulaException as opennebula_err: except OpenNebulaException as opennebula_err:
logger.info("OpenNebulaException error: {0}".format(opennebula_err)) logger.info(
"OpenNebulaException error: {0}".format(opennebula_err))
except OSError as os_err: except OSError as os_err:
logger.info("OSError : {0}".format(os_err)) logger.info("OSError : {0}".format(os_err))
except ValueError as value_err: except ValueError as value_err:
@ -263,7 +266,6 @@ class OpenNebulaManager():
except: except:
raise ConnectionRefusedError raise ConnectionRefusedError
def get_templates(self): def get_templates(self):
try: try:
public_templates = [ public_templates = [
@ -291,10 +293,8 @@ class OpenNebulaManager():
except: except:
raise ConnectionRefusedError raise ConnectionRefusedError
def create_template(self, name, cores, memory, disk_size, core_price, memory_price, def create_template(self, name, cores, memory, disk_size, core_price, memory_price,
disk_size_price, ssh='' ): disk_size_price, ssh=''):
"""Create and add a new template to opennebula. """Create and add a new template to opennebula.
:param name: A string representation describing the template. :param name: A string representation describing the template.
Used as label in view. Used as label in view.
@ -327,11 +327,11 @@ class OpenNebulaManager():
template_string_formatter.format( template_string_formatter.format(
name=name, name=name,
vcpu=cores, vcpu=cores,
cpu=0.1*cores, cpu=0.1 * cores,
size=1024 * disk_size, size=1024 * disk_size,
memory=1024 * memory, memory=1024 * memory,
# * 10 because we set cpu to *0.1 # * 10 because we set cpu to *0.1
cpu_cost=10*core_price, cpu_cost=10 * core_price,
memory_cost=memory_price, memory_cost=memory_price,
disk_cost=disk_size_price, disk_cost=disk_size_price,
ssh=ssh ssh=ssh
@ -341,7 +341,8 @@ class OpenNebulaManager():
return template_id return template_id
def delete_template(self, template_id): def delete_template(self, template_id):
self.oneadmin_client.call(oca.VmTemplate.METHODS['delete'], template_id, False) self.oneadmin_client.call(oca.VmTemplate.METHODS[
'delete'], template_id, False)
def change_user_password(self, new_password): def change_user_password(self, new_password):
self.oneadmin_client.call( self.oneadmin_client.call(