-
Notifications
You must be signed in to change notification settings - Fork 8
/
OneTimePadCipher.py
31 lines (25 loc) · 885 Bytes
/
OneTimePadCipher.py
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
from ciphers import VignereCipher
import numpy as np
class OneTimePadCipher:
def __init__(self):
self.key = ''
self.pad_used = False
self.cipher = None
def encrypt(self, plaintext: str):
if not self.pad_used:
self.pad_used = True
self.generate_random_key(plaintext)
self.cipher = VignereCipher(self.key)
return self.cipher.encrypt(plaintext)
else:
return None
def decrypt(self, ciphertext: str):
if not self.pad_used:
return None
else:
return self.cipher.decrypt(ciphertext)
def generate_random_key(self, plaintext: str) -> None:
self.key = ''.join(list(map(self.num_2_char, np.random.randint(0, 25, len(plaintext)))))
@staticmethod
def num_2_char(number: int) -> str:
return chr(number + ord('a'))