Onepad_Cipher.py

from __future__ import print_function

import random


class Onepad:
    def encrypt(self, text):
        '''Function to encrypt text using psedo-random numbers'''
        plain = [ord(i) for i in text]
        key = []
        cipher = []
        for i in plain:
            k = random.randint(1, 300)
            c = (i+k)*k
            cipher.append(c)
            key.append(k)
        return cipher, key
    
    def decrypt(self, cipher, key):
        '''Function to decrypt text using psedo-random numbers.'''
        plain = []
        for i in range(len(key)):
            p = (cipher[i]-(key[i])**2)/key[i]
            plain.append(chr(p))
        plain = ''.join([i for i in plain])
        return plain


if __name__ == '__main__':
    c, k = Onepad().encrypt('Hello')
    print(c, k)
    print(Onepad().decrypt(c, k))