U c @sbddlZddlmZmZddlmZmZmZddlmZddgZ e e dddZ Gd ddZ dS) N)datetimetimezone)DictOptionalUnion)core POEManagerdigest_for_poe)datareturncCst|SN)hashlibsha256digest)r rA/tmp/pip-unpacked-wheel-rwcmptg8/pyhanko_certvalidator/ltv/poe.pyr sc@seZdZdZdeedddZdeee j feeedddZ deeeed d d Z d d Z eee j fedddZddZddZdS)rz~ Class to manage proof-of-existence (POE) claims. :param current_dt_override: Override the current time. Ncurrent_dt_overridecCsi|_||_dSr )_poes_current_dt_override)selfrrrr__init__szPOEManager.__init__)r dtr cCs(t|tjr|}t|}|||S)a9 Register a new POE claim if no POE for an earlier time is available. :param data: Data to register a POE claim for. :param dt: The POE time to register. If ``None``, assume the current time. :return: The oldest POE datetime available. ) isinstancer Asn1Valuedumpr register_by_digest)rr rrrrrregisters zPOEManager.register)rrr cCsV|p|jpttj}z|j|}||kr0|WSWntk rFYnX||j|<|S)aQ 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. :return: The oldest POE datetime available. )rrnowrutcrKeyError)rrrZcur_poerrrr,s    zPOEManager.register_by_digestcCst|jS)z Iterate over the current earliest known POE for all items currently being managed. Returns an iterator with ``(digest, poe_dt)`` pairs. )iterritems)rrrr__iter__CszPOEManager.__iter__)itemr cCs|j|ddS)a Return the earliest available POE for an item. .. note:: This is an alias for :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)r)rr$rrr __getitem__LszPOEManager.__getitem__cCs0t|tstt|D]\}}|||qdS)ze Combine data in another POE manager with the POEs managed by this instance. N)rr TypeErrorr!r)rotherrrrrr__ior__]s zPOEManager.__ior__cCst|jd}t|j|_|S)Nr)rrdictr)rZ new_instancerrr__copy__gs  zPOEManager.__copy__)N)N)N)__name__ __module__ __qualname____doc__rrrrbytesrrrrr#r%r(r*rrrrrs$     ) r rrtypingrrrZ asn1cryptor__all__r/r rrrrrs