diff options
Diffstat (limited to 'django-nova/src/django_nova/views/images.py')
| -rw-r--r-- | django-nova/src/django_nova/views/images.py | 229 |
1 files changed, 229 insertions, 0 deletions
diff --git a/django-nova/src/django_nova/views/images.py b/django-nova/src/django_nova/views/images.py new file mode 100644 index 00000000..1d09d2df --- /dev/null +++ b/django-nova/src/django_nova/views/images.py @@ -0,0 +1,229 @@ +# vim: tabstop=4 shiftwidth=4 softtabstop=4 + +# Copyright 2010 United States Government as represented by the +# Administrator of the National Aeronautics and Space Administration. +# All Rights Reserved. +# +# Licensed under the Apache License, Version 2.0 (the "License"); you may +# not use this file except in compliance with the License. You may obtain +# a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT +# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the +# License for the specific language governing permissions and limitations +# under the License. + +""" +Views for managing Nova images. +""" + +import boto.exception +import re + +from django import http +from django import template +from django.conf import settings +from django.contrib import messages +from django.contrib.auth.decorators import login_required +from django.shortcuts import redirect, render_to_response +from django_nova import exceptions +from django_nova import forms +from django_nova import shortcuts +from django_nova.exceptions import handle_nova_error + + +def _image_lists(images, project_id): + def image_is_project(i): + return i.ownerId == project_id + + def image_is_admin(i): + return i.ownerId in ['admin'] + + def image_is_community(i): + return (not image_is_admin(i)) and (not image_is_project(i)) + + return {'Project Images': filter(image_is_project, images), + '%s Images' % settings.SITE_BRANDING: filter(image_is_admin, + images), + 'Community Images': filter(image_is_community, images)} + + +@login_required +@handle_nova_error +def index(request, project_id): + project = shortcuts.get_project_or_404(request, project_id) + images = project.get_images() + + return render_to_response('django_nova/images/index.html', { + 'form': forms.LaunchInstanceForm(project), + 'region': project.region, + 'project': project, + 'image_lists': _image_lists(images, project_id), + }, context_instance = template.RequestContext(request)) + + +@login_required +@handle_nova_error +def launch(request, project_id, image_id): + project = shortcuts.get_project_or_404(request, project_id) + + if request.method == 'POST': + form = forms.LaunchInstanceForm(project, request.POST) + if form.is_valid(): + try: + reservation = project.run_instances( + image_id, + addressing_type='private', + key_name=form.cleaned_data['key_name'], + #security_groups=[form.cleaned_data['security_group']], + user_data=re.sub('\r\n', '\n', + form.cleaned_data['user_data']), + instance_type=form.cleaned_data['size'], + min_count=form.cleaned_data['count'], + max_count=form.cleaned_data['count'] + ) + except exceptions.NovaApiError, e: + messages.error(request, + 'Unable to launch: %s' % e.message) + else: + for instance in reservation.instances: + messages.success(request, + 'Instance %s launched.' % instance.id) + return redirect('nova_instances', project_id) + else: + form = forms.LaunchInstanceForm(project) + + ami = project.get_image(image_id) + + return render_to_response('django_nova/images/launch.html', { + 'form': form, + 'region': project.region, + 'project': project, + 'ami': ami, + }, context_instance = template.RequestContext(request)) + + +@login_required +@handle_nova_error +def detail(request, project_id, image_id): + project = shortcuts.get_project_or_404(request, project_id) + images = project.get_images() + + ami = project.get_image(image_id) + can_modify = shortcuts.get_user_image_permissions(request.user.username, + project_id) + + if not ami: + raise http.Http404() + return render_to_response('django_nova/images/index.html', { + 'form': forms.LaunchInstanceForm(project), + 'region': project.region, + 'project': project, + 'images': images, + 'image_lists': _image_lists(images, project_id), + 'ami': ami, + 'can_modify': can_modify, + }, context_instance = template.RequestContext(request)) + + +@login_required +@handle_nova_error +def remove(request, project_id, image_id): + project = shortcuts.get_project_or_404(request, project_id) + + if request.method == 'POST': + try: + project.deregister_image(image_id) + except exceptions.NovaApiError, e: + messages.error(request, + 'Unable to deregister image: %s' % e.message) + else: + messages.success(request, + 'Image %s has been successfully deregistered.' % + image_id) + + return redirect('nova_images', project_id) + else: + ami = project.get_image(image_id) + + #FIXME - is the code below used? if we reach here should we + #just redirect? (anthony) + return render_to_response('django_nova/images/detail_list.html', { + 'region': project.region, + 'project': project, + 'ami': ami, + }, context_instance = template.RequestContext(request)) + + +@login_required +@handle_nova_error +def privacy(request, project_id, image_id): + project = shortcuts.get_project_or_404(request, project_id) + + if request.method == 'POST': + ami = project.get_image(image_id) + + if ami.is_public: + try: + project.modify_image_attribute(image_id, + attribute='launchPermission', + operation='remove') + except exceptions.NovaApiError, e: + messages.error(request, + 'Unable to make image private: %s' % e.message) + else: + try: + project.modify_image_attribute(image_id, + attribute='launchPermission', + operation='add') + except exceptions.NovaApiError, e: + messages.error(request, + 'Unable to make image public: %s' % e.message) + + return redirect('nova_images_detail', project_id, image_id) + + +@login_required +@handle_nova_error +def update(request, project_id, image_id): + project = shortcuts.get_project_or_404(request, project_id) + ami = project.get_image(image_id) + + if request.method == 'POST': + form = forms.UpdateImageForm(ami, request.POST) + if form.is_valid(): + try: + project.update_image(image_id, + form.cleaned_data['nickname'], + form.cleaned_data['description']) + except exceptions.NovaApiError, e: + messages.error(request, + 'Unable to update image: %s' % e.message) + else: + messages.success(request, + 'Image %s has been updated.' % image_id) + + return redirect('nova_images_detail', project_id, image_id) + + # TODO(devcamcar): This needs to be cleaned up. Can make + # one of the render_to_response blocks go away. + else: + form = forms.UpdateImageForm(ami) + return render_to_response('django_nova/images/edit.html', { + 'form': form, + 'region': project.region, + 'project': project, + 'ami': ami, + }, context_instance = template.RequestContext(request)) + else: + form = forms.UpdateImageForm(ami) + return render_to_response('django_nova/images/edit.html', { + 'form': form, + 'region': project.region, + 'project': project, + 'ami': ami, + }, context_instance = template.RequestContext(request)) + |