diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/sec_certs/sample/cc_certificate_id.py | 8 |
1 files changed, 5 insertions, 3 deletions
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 = [ |