o vh@sddlZddlZddlmZddlmZmZddlmZmZm Z m Z m Z ddl m Z mZddlmZmZgdZejGdd d ejZe eeeejfZd ed e efd d ZeddGdddZejGdddejZeddGdddZded efddZGdddZdS)N) dataclass)datetimetimezone)AnyDictIteratorOptionalUnion)corex509) CRLContainer OCSPContainer)ValidationObjectTypeValidationObjectPOETypeKnownPOE POEManagerdigest_for_poec@s8eZdZdZdZdZdZdZdZdZ dZ d Z d d Z d S) rzF Types of validation objects recognised by ETSI TS 119 102-2. certificateCRL OCSPResponse timestampevidencerecord publicKey signedDataothercC d|jS)Nz!urn:etsi:019102:validationObject:valueselfr![/var/www/html/hyperkenya/venv/lib/python3.10/site-packages/pyhanko_certvalidator/ltv/poe.pyurn#s zValidationObjectType.urnN) __name__ __module__ __qualname____doc__ CERTIFICATEr OCSP_RESPONSE TIMESTAMPEVIDENCE_RECORD PUBLIC_KEY SIGNED_DATAOTHERr#r!r!r!r"rs rthingreturncCs6t|trtjSt|trtjSt|tjrtjSdSN) isinstancer rrr r)r Certificater()r/r!r!r"guess_validation_object_type*s   r4T)frozenc@s$eZdZUdZeed< eed<dS)rz A validation object used in the course of a validation operation for which proofs of existence can potentially be gathered. object_typerN)r$r%r&r'r__annotations__rr!r!r!r"r6s rc@s*eZdZdZdZdZedefddZdS)rprovided validationpolicyr0cCr)Nzurn:etsi:019102:poetype:rrr!r!r"r#Us z POEType.urnN) r$r%r&PROVIDED VALIDATIONPOLICYpropertystrr#r!r!r!r"rOs rc@s6eZdZUeed<eed<eed<dZee ed<dS)rpoe_typedigestpoe_timeNvalidation_object) r$r%r&rr7bytesrrCrrr!r!r!r"rZs rdatacCst|Sr1)hashlibsha256rA)rEr!r!r"rbsrc @seZdZdZddeefddZ ddededeed e fd d Z dd e dedeed e fd dZ de d e fddZ d ee fddZded efddZddZddZdS)rz~ Class to manage proof-of-existence (POE) claims. :param current_dt_override: Override the current time. Ncurrent_dt_overridecCsi|_||_dSr1)_poes_current_dt_override)r rHr!r!r"__init__ns zPOEManager.__init__rEr@dtr0cCst|tr|}n#t|tjr|}nt|tr|j}n t|tr)|j}nt t |}|p9|j p9t tj}t|}d}|rHt||d}|t||||dS)ao Register a new POE claim if no POE for an earlier time is available. :param data: Data to register a POE claim for. :param poe_type: The type of POE. :param dt: The POE time to register. If ``None``, assume the current time. :return: The oldest POE datetime available. N)r6rr@rArBrC)r2rDr Asn1Valuedumpr crl_datar ocsp_response_dataNotImplementedErrorrrJrnowrutcr4rregister_known_poer)r rEr@rLb_datarAvo_typevor!r!r"registerrs.        zPOEManager.registerrAcCs,|p |jp ttj}|t|||ddS)a Register a new POE claim if no POE for an earlier time is available. :param digest: SHA-256 digest of the data to register a POE claim for. :param dt: The POE time to register. If ``None``, assume the current time. :param poe_type: The type of POE. :return: The oldest POE datetime available. NrM)rJrrSrrTrUr)r rAr@rLr!r!r"register_by_digestszPOEManager.register_by_digest known_poecCsL|j}|j}z|j|}|j|kr|WSWn tyYnw||j|<|S)z Register a new POE claim if no POE for an earlier time is available. :param known_poe: The POE object to register. :return: The oldest POE for the given digest. )rBrArIKeyError)r r[rLrAcur_poer!r!r"rUs     zPOEManager.register_known_poecCst|jS)z Iterate over the current earliest known POE for all items currently being managed. Returns an iterator with :class:`KnownPOE` objects. )iterrIvaluesrr!r!r"__iter__szPOEManager.__iter__itemcCs|j|tjddjS)a Return the earliest available POE for an item. .. note:: This is a wrapper around :meth:`register` with `dt=None`, and hence will register the current time as the POE time for the given item. This side effect is intentional. :param item: Item to get the current POE time for. :return: A datetime object representing the earliest available POE for the item. N)r@rL)rYrr<rB)r rar!r!r" __getitem__s zPOEManager.__getitem__cCs*t|tstt|D]}||q dS)ze Combine data in another POE manager with the POEs managed by this instance. N)r2r TypeErrorr^rU)r rpoer!r!r"__ior__s   zPOEManager.__ior__cCst|jd}t|j|_|S)N)rH)rrJdictrI)r new_instancer!r!r"__copy__s  zPOEManager.__copy__r1)r$r%r&r'rrrKKnownObjectTyperrrYrDrZrUrr`rbrerhr!r!r!r"rfs8 0   r) enumrF dataclassesrrrtypingrrrrr asn1cryptor r &pyhanko_certvalidator.revinfo.archivalr r __all__uniqueEnumrrDr3riobjectr4rrrrrr!r!r!r"s0