use const

Author: rufoa

kg.py

@@ -7,11 +7,18 @@
 from cryptography.hazmat.primitives.asymmetric.utils import decode_dss_signature
 from cryptography.hazmat.primitives.hashes import SHA1
 
+EDITIONS = {
+    "Lite": 4,
+    "Intro": 3,
+    "Standard": 0,
+    "Suite": 2,
+}
+
 parser = argparse.ArgumentParser()
 parser.add_argument("-i", "--hwid", help="Your hardware code", required=True)
 parser.add_argument("-o", "--output", help="Authorization file", default="Authorize.auz")
 parser.add_argument("-v", "--version", help="Ableton Live version", type=int, choices=range(9, 13), default=12)
-parser.add_argument("-e", "--edition", help="Ableton Live edition", choices=["Lite", "Intro", "Standard", "Suite"], default="Suite")
+parser.add_argument("-e", "--edition", help="Ableton Live edition", type=str.capitalize, choices=EDITIONS, default="Suite")
 args = parser.parse_args()
 
 
@@ -81,8 +88,7 @@ def generate_single(k: dsa.DSAPrivateKey, id1: int, id2: int, hwid: str) -> str:
 
 
 def generate_all(k: dsa.DSAPrivateKey, edition: str, version: int, hwid: str) -> str:
-    ident = {"Lite": 4, "Intro": 3, "Standard": 0, "Suite":  2}
-    yield generate_single(k, ident[edition], version << 4, hwid)
+    yield generate_single(k, EDITIONS[edition], version << 4, hwid)
     for i in range(0x40, 0xff + 1):
         yield generate_single(k, i, 0x10, hwid)
     for i in range(0x8000, 0x80ff + 1):