Answers SSO python example

CryptographyUtil.py

import base64
import hashlib
import json
import time
from Crypto.Cipher import AES
from django.http import HttpResponseRedirect

def _pad(bs, s):
    return s + (bs - len(s) % bs) * chr(bs - len(s) % bs)

def _encryptUserData(data, key):
    key = hashlib.sha1(key.encode()).digest()[0:16]
    message = _pad(AES.block_size, data)
    cipher = AES.new(key, AES.MODE_ECB)
    return base64.urlsafe_b64encode(cipher.encrypt(message)).decode('utf-8')

def _omitPadding(token):
    return token.replace('=', '')

def _getRedirectUrl():
    WIX_SECRET = '<your secret>'
    WIX_KEY_ID = '<your key>'
    user_data = {
        'id': '<the user id>',
        'email': '<the user email>',
        'firstName': '<the user first name>',
        'lastName': '<the user last name>',
        'timestamp': int(round(time.time() * 1000))  # milliseconds required
    }
    answersRedirectUrl = 'redirectUrl'
    token = _omitPadding(_encryptUserData(json.dumps(user_data), WIX_SECRET))
    success_url = answersRedirectUrl + '&token=' + token + '&key=' + WIX_KEY_ID
    return HttpResponseRedirect(success_url)

_getRedirectUrl()