diff --git a/albert.py b/albert.py index ca447ed..a4f39f2 100644 --- a/albert.py +++ b/albert.py @@ -43,9 +43,12 @@ def _generate_csr(private_key: rsa.RSAPrivateKey) -> str: return csr.public_bytes(serialization.Encoding.PEM).decode("utf-8") -# Generates an APNs push certificate by talking to Albert -# Returns [private key PEM, certificate PEM] def generate_push_cert() -> tuple[str, str]: + """ + Generates an APNs push certificate by talking to Albert. + + Returns [private key PEM, certificate PEM] + """ private_key = rsa.generate_private_key( public_exponent=65537, key_size=2048, backend=default_backend() ) diff --git a/development/printer.py b/development/printer.py index 999cdb6..b35763f 100644 --- a/development/printer.py +++ b/development/printer.py @@ -176,11 +176,11 @@ def pretty_print_payload( ) elif id == 0x1D or id == 0x20: print( - f"{bcolors.OKGREEN}{prefix}{bcolors.ENDC}: {bcolors.WARNING}PubSub ??{bcolors.ENDC}" + f"{bcolors.OKGREEN}{prefix}{bcolors.ENDC}: {bcolors.WARNING}PubSub {id}{bcolors.ENDC}" ) elif id == 0xE: print( - f"{bcolors.OKGREEN}{prefix}{bcolors.ENDC}: {bcolors.WARNING}Token Confirmation{bcolors.ENDC}" + f"{bcolors.OKGREEN}{prefix}{bcolors.ENDC}: {bcolors.WARNING}No Storage{bcolors.ENDC}" ) elif id == 0xA: topic = ""