GenerateID.py

from cryptography.fernet import Fernet
import random
from datetime import datetime
import os
from config import key


class genID:

    def __init__(self):
        self.keysList = list()
        self.key = None
        self.id = None
        self.retrieveKeys()

    def newID(self, newSerial):
        self.key = self.randomKey()
        # print(self.key[0],":",self.key[1:])
        self.encryption_type = Fernet(self.key[1:].encode())
        # print("Total number of IDs stored already:>> ", newSerial)
        pswd = str(random.randint(1000, 9999))
        # print("random pswd", pswd)
        self.en_pswd = self.key[0] + self.enPswd(pswd).decode()
        # print(len(self.en_pswd))
        self.id = pswd + "1103" + datetime.today().strftime("%Y%m%d") + '0000'
        self.id = str(int(self.id) + int(newSerial))
        # print("New generated ID:>> ", self.id[0:4] + " " + self.id[4:8] + " "
        #       + self.id[8:12] + " " + self.id[12:14] + " " + self.id[14:16] + " " + self.id[16:])
        return self.id, self.en_pswd

    def randomKey(self):
        return random.choice(self.keysList)

    def enPswd(self, pswd):
        return self.encryption_type.encrypt(pswd.encode())

    def dePswd(self, en_pswd):

        return str(self.encryption_type.decrypt(en_pswd.encode()))

    def retrieveKeys(self):

        with open(os.path.join(os.getcwd(), "keys.key"), "r") as k:
            for i in k:
                key = i.strip()
                self.keysList.append(key)
        print("Keys retrieved")
        # print(self.keysList)


if __name__ == "__main__":
    pass