From 279a9ef2a21861c0bbf2a5c62dbb11d6e88bd636 Mon Sep 17 00:00:00 2001
From: JJTech0130 <jjtech@jjtech.dev>
Date: Thu, 17 Aug 2023 14:53:01 -0400
Subject: [PATCH] start changes

---
 albert.py              | 7 +++++--
 development/printer.py | 4 ++--
 2 files changed, 7 insertions(+), 4 deletions(-)

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 = ""