some checkups for manager

This commit is contained in:
Ville Rantanen
2018-03-02 12:04:48 +02:00
parent 06d42ee956
commit 01cfd02679
4 changed files with 58 additions and 25 deletions

View File

@@ -1,34 +1,34 @@
import random
import string
import base64
from Crypto.Cipher import AES
#~ from Crypto.Cipher import AES
import hashlib
class Crypto:
def __init__(self, secret):
self.secret = add_pad(secret[0:16])
self.cipher = AES.new(self.secret, AES.MODE_ECB)
#~ class Crypto:
#~ def __init__(self, secret):
#~ self.secret = add_pad(secret[0:16])
#~ self.cipher = AES.new(self.secret, AES.MODE_ECB)
def encrypt(self, msg):
#~ def encrypt(self, msg):
return base64.urlsafe_b64encode(
self.cipher.encrypt(
add_pad(
msg
)
)
).decode("utf-8")
#~ return base64.urlsafe_b64encode(
#~ self.cipher.encrypt(
#~ add_pad(
#~ msg
#~ )
#~ )
#~ ).decode("utf-8")
def decrypt(self, enc):
#~ def decrypt(self, enc):
return remove_pad(
self.cipher.decrypt(
base64.urlsafe_b64decode(
enc
)
).decode("utf-8")
)
#~ return remove_pad(
#~ self.cipher.decrypt(
#~ base64.urlsafe_b64decode(
#~ enc
#~ )
#~ ).decode("utf-8")
#~ )
def add_pad(string):