Total Complexity | 5 |
Total Lines | 24 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | from asyncua.crypto import uacrypto |
||
2 | import sys |
||
3 | import logging |
||
4 | sys.path.append('..') |
||
5 | |||
6 | |||
7 | class CertificateHandler: |
||
8 | def __init__(self): |
||
9 | self._trusted_certificates = {} |
||
10 | |||
11 | async def trust_certificate(self, certificate_path: str, format: str = None, label: str = None): |
||
12 | certificate = await uacrypto.load_certificate(certificate_path, format) |
||
13 | if label is None: |
||
14 | label = certificate_path |
||
15 | if label in self._trusted_certificates: |
||
16 | logging.warning(f"certificate with label {label} " |
||
17 | f"attempted to be added multiple times, only the last version will be kept.") |
||
18 | self._trusted_certificates[label] = uacrypto.der_from_x509(certificate) |
||
19 | |||
20 | def __contains__(self, certificate): |
||
21 | return any(certificate == prospective_cert |
||
22 | for prospective_cert |
||
23 | in self._trusted_certificates.values()) |
||
24 |