init

src/uk_plate_reader.py

@@ -0,0 +1,56 @@
+import json
+import sys
+
+with open("./data/area_codes.json", "r") as f:
+    areas = json.load(f)
+
+def get_legacy_year(plate:str):
+    base_year = 1983
+    try:
+        year_indicator = plate[0]
+    except IndexError:
+        return False
+    year_offset = ord(year_indicator.upper()) - ord("A")
+    year = base_year + year_offset
+    if year < 1983 or year >= 2001:
+        year = False
+    return year
+
+def get_year(plate:str)->int:
+    year_indicator = int(plate[2:4])
+    if year_indicator >= 50:
+        year_indicator = year_indicator - 50
+    return year_indicator + 2000
+
+def get_legacy_area(plate:str):
+    if len(plate.replace(" ","")) == 5:
+        area_code = plate[-1]
+    elif len(plate.replace(" ", "")) == 6:
+        area_code = plate[-2:]
+    return areas.get("legacy_style").get(area_code)
+
+def get_area(plate:str):
+    area_code = plate[0:2]
+    area, _ = areas.get("modern_style").get(area_code)
+    return area
+
+def is_legacy(plate):
+    try:
+        return int(plate[1])
+    except ValueError:
+        return False
+
+def read_licence_plate(plate:str):
+    if is_legacy(plate):
+        area = get_legacy_area(plate)
+        year = get_legacy_year(plate)
+    else:
+        area = get_area(plate)
+        year = get_year(plate)
+    if not area or not year:
+        raise Exception("Not a valid licence plate")
+    return [area, year]
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1:
+        print(read_licence_plate(sys.argv[1]))