From 3eeb7b07c9885bcbb1f33a319f80ee901dd2f3d2 Mon Sep 17 00:00:00 2001
From: adamjanovsky
Date: Sun, 17 Sep 2023 14:53:00 +0200
Subject: fix anssi cert_id normalization

---
 src/sec_certs/sample/cc_certificate_id.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

(limited to 'src')

diff --git a/src/sec_certs/sample/cc_certificate_id.py b/src/sec_certs/sample/cc_certificate_id.py
index d0ebdb81..1342cab8 100644
--- a/src/sec_certs/sample/cc_certificate_id.py
+++ b/src/sec_certs/sample/cc_certificate_id.py
@@ -15,9 +15,11 @@ class CertificateId:
 
     def _canonical_fr(self) -> str:
         def pad_last_segment_with_zero(id_str: str) -> str:
-            prefix, num = id_str.split("/")
-            num = num.zfill(2)
-            return f"{prefix}/{num}"
+            splitted = id_str.split("/")
+            if len(splitted) > 1:
+                num = splitted[-1].zfill(2)
+                return f"{''.join(splitted[:-1])}/{num}"
+            return id_str
 
         new_cert_id = self.clean
         rules = [
-- 
cgit v1.3.1