From 9eb5f67a954888c4e14789b5b8acc785c789a77c Mon Sep 17 00:00:00 2001 From: juanifioren Date: Mon, 13 Jul 2015 15:49:08 -0300 Subject: [PATCH] Add a command for creating rsa key. --- .../management/commands/creatersakey.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 oidc_provider/management/commands/creatersakey.py diff --git a/oidc_provider/management/commands/creatersakey.py b/oidc_provider/management/commands/creatersakey.py new file mode 100644 index 0000000..fb406f1 --- /dev/null +++ b/oidc_provider/management/commands/creatersakey.py @@ -0,0 +1,18 @@ +from Crypto.PublicKey import RSA + +from django.conf import settings +from django.core.management.base import BaseCommand, CommandError + + +class Command(BaseCommand): + help = 'Randomly generate a new RSA key for the OpenID server' + + def handle(self, *args, **options): + try: + key = RSA.generate(1024) + file_path = settings.BASE_DIR + '/OIDC_RSA_KEY.pem' + with open(file_path, 'w') as f: + f.write(key.exportKey('PEM')) + self.stdout.write('RSA key successfully created at: ' + file_path) + except Exception as e: + self.stdout.write('Something goes wrong: ' + e.message)