stopify/utils.py at master · jaimescose/stopify · GitHub

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
from cryptography.fernet import Fernet


def get_encryption_key(service):
    from models import SpotifyProfile, TwitterProfile

    if service == 'twitter':
        key = TwitterProfile.get_credentials()[3]
    elif service == 'spotify':
        key = SpotifyProfile.get_credentials()[3]

    return key


def encrypt_token(value, service):
    key = get_encryption_key(service)

    value = value.encode()

    f = Fernet(key)
    value = f.encrypt(value)

    return value.decode()


def decrypt_token(value, service):
    key = get_encryption_key(service)

    value = value.encode()

    f = Fernet(key)
    value = f.decrypt(value)

    return value.decode()