1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
|
# 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 keypairs.
"""
from django import http
from django import template
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
@login_required
@handle_nova_error
def index(request, project_id, download_key=None):
project = shortcuts.get_project_or_404(request, project_id)
keypairs = project.get_key_pairs()
return render_to_response('django_nova/keypairs/index.html', {
'create_form': forms.CreateKeyPairForm(project),
'region': project.region,
'project': project,
'keypairs': keypairs,
'download_key': download_key
}, context_instance = template.RequestContext(request))
@login_required
@handle_nova_error
def add(request, project_id):
project = shortcuts.get_project_or_404(request, project_id)
if request.method == 'POST':
form = forms.CreateKeyPairForm(project, request.POST)
if form.is_valid():
try:
keypair = project.create_key_pair(form.cleaned_data['name'])
except exceptions.NovaApiError, e:
messages.error(request,
'Unable to create key: %s' % e.message)
else:
if request.POST['js'] == '1':
request.session['key.%s' % keypair.name] = keypair.material
return index(request,
project_id,
download_key=keypair.name)
else:
response = http.HttpResponse(mimetype='application/binary')
response['Content-Disposition'] = \
'attachment; filename=%s.pem' % \
form.cleaned_data['name']
response.write(keypair.material)
return response
else:
keypairs = project.get_key_pairs()
return render_to_response('django_nova/keypairs/index.html', {
'create_form': form,
'region': project.region,
'project': project,
'keypairs': keypairs,
}, context_instance = template.RequestContext(request))
return redirect('nova_keypairs', project_id)
@login_required
@handle_nova_error
def delete(request, project_id):
project = shortcuts.get_project_or_404(request, project_id)
if request.method == 'POST':
key_name = request.POST['key_name']
try:
project.delete_key_pair(key_name)
except exceptions.NovaApiError, e:
messages.error(request,
'Unable to delete key: %s' % e.message)
else:
messages.success(request,
'Key %s has been successfully deleted.' % \
key_name)
return redirect('nova_keypairs', project_id)
@login_required
@handle_nova_error
def download(request, project_id, key_name):
# Ensure the project exists.
shortcuts.get_project_or_404(request, project_id)
try:
material = request.session.pop('key.%s' % key_name)
except KeyError:
return redirect('nova_keypairs', project_id)
response = http.HttpResponse(mimetype='application/binary')
response['Content-Disposition'] = 'attachment; filename=%s.pem' % key_name
response.write(material)
return response
|
