X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=portal%2Fviews.py;h=5b4399dbe44eb3884fa6bb99d608a84dda77b5b4;hb=05861f44a20edf11bacba3d0d6fad3dee9ad7957;hp=2d8d7925fd3e9919c173c2a837a1be276e320a2b;hpb=f32b4c0960cdeba72c6023b54464d1e70e0534fb;p=myslice.git diff --git a/portal/views.py b/portal/views.py index 2d8d7925..5b4399db 100644 --- a/portal/views.py +++ b/portal/views.py @@ -30,7 +30,7 @@ from django.shortcuts import render from plugins.lists.simplelist import SimpleList -from plugins.pres_view import PresView +#from plugins.pres_view import PresView from portal.event import Event import json @@ -509,9 +509,38 @@ def acc_process(request): get_user.save() return HttpResponse('Success: Password Changed!!') elif 'generate' in request.POST: - a =2 - message = 'Here will generate ssh-rsa keys :D %d' %a - return HttpResponse(message) + import os + from M2Crypto import Rand, RSA, BIO + + KEY_LENGTH = 2048 + + def blank_callback(): + "Replace the default dashes" + return + + # Random seed + Rand.rand_seed (os.urandom (KEY_LENGTH)) + # Generate key pair + key = RSA.gen_key (KEY_LENGTH, 65537, blank_callback) + # Create memory buffers + pri_mem = BIO.MemoryBuffer() + pub_mem = BIO.MemoryBuffer() + # Save keys to buffers + key.save_key_bio(pri_mem, None) + key.save_pub_key_bio(pub_mem) + + # Get keys + public_key = pub_mem.getvalue() + private_key = pri_mem.getvalue() + # Saving to DB + keypair = '{"user_public_key":"'+ public_key + '", "user_private_key":"'+ private_key + '"}' + #keypair = re.sub("\r", "", keypair) + #keypair = re.sub("\n", "\\n", keypair) + keypair = keypair.rstrip('\r\n') + get_user.keypair = keypair + get_user.save() + return HttpResponse('Success: New Keypair Generated! %s' % keypair) + elif 'upload_key' in request.POST: up_file = request.FILES['pubkey'] file_content = up_file.read() @@ -519,7 +548,7 @@ def acc_process(request): file_extension = os.path.splitext(file_name)[1] allowed_extension = ['.pub','.txt'] if file_extension in allowed_extension: - file_content = '{user_public_key:'+ file_content +'}' + file_content = '{"user_public_key":"'+ file_content +'"}' file_content = re.sub("\r", "", file_content) file_content = re.sub("\n", "\\n",file_content) get_user.keypair = file_content