generate_code.py

from pixqrcode.model.merchant_account import MerchantAccount
from pixqrcode.model.pix import Pix
from pixqrcode.utils.crc16 import crc16
from pixqrcode.service.validate_pix import ValidatePix
import re

class GenerateCode:
    def __init__(self):
        self.merchant = MerchantAccount()

    def left_zero(self, text: str):
        return str(len(text)).zfill(2)

    def generate(self, pix: Pix):
        valPix = ValidatePix(pix)
        detect_type = valPix.detect_type_key
        code = detect_type()

        return f"00020126360014{self.merchant.globally_unique_identifier}{code}{pix.mobile}520400005303986540" \
        f"{len(pix.amount.__str__())}{pix.amount.__str__()}5802{pix.country_code}59{self.left_zero(pix.name)}" \
            f"{pix.name}60{self.left_zero(pix.city)}{pix.city}" \
            f"62{str(len(pix.reference_label) + 4).zfill(2)}05{self.left_zero(pix.reference_label)}" \
            f"{pix.reference_label}6304"

    def crc16code(self, query: str) -> str:
        return hex(crc16(query.encode())).__str__()

    def format_code(self, pix: Pix):
        hash_payment = self.generate(pix)
        return f"{hash_payment}{self.crc16code(hash_payment).upper()[2:]}".strip()