o h@sjdZddlZddlZddlZddlZddlmZmZmZddl m Z ddl Z ddl mZGdddZdS)z:SSL Certificate class for handling certificate operations.N)DictAnyOptional)urlparse)Pathc @seZdZdZdeeeffddZed dede de dfd d Z ed edefd d Z d!de ede efddZ d!de ede efddZd!de ede efddZedeeeffddZedeeeffddZedefddZedefddZedefddZdS)"SSLCertificatean A class representing an SSL certificate with methods to export in various formats. Attributes: cert_info (Dict[str, Any]): The certificate information. Methods: from_url(url: str, timeout: int = 10) -> Optional['SSLCertificate']: Create SSLCertificate instance from a URL. from_file(file_path: str) -> Optional['SSLCertificate']: Create SSLCertificate instance from a file. from_binary(binary_data: bytes) -> Optional['SSLCertificate']: Create SSLCertificate instance from binary data. export_as_pem() -> str: Export the certificate as PEM format. export_as_der() -> bytes: Export the certificate as DER format. export_as_json() -> Dict[str, Any]: Export the certificate as JSON format. export_as_text() -> str: Export the certificate as text format. cert_infocCs|||_dSN)_decode_cert_data _cert_info)selfrr U/var/www/Befach/backend/venv/lib/python3.10/site-packages/crawl4ai/ssl_certificate.py__init__szSSLCertificate.__init__ urltimeoutreturnc Cszt|j}d|vr|dd}t}tj|df|d}|j||do}|jdd}t j t j j |}t |t ||t||||d|t|d }g} t|D]} || } | | t| d qj| |d <t |Wd Wd WS1swYWd Wd S1swYWd St!y} zWYd } ~ d Sd } ~ ww) a: Create SSLCertificate instance from a URL. Args: url (str): URL of the website. timeout (int): Timeout for the connection (default: 10). Returns: Optional[SSLCertificate]: SSLCertificate instance if successful, None otherwise. :ri)r)server_hostnameT) binary_formsha256) subjectissuerversion serial_number not_before not_after fingerprintsignature_algorithmraw_cert)namevalue extensionsN)"rnetlocsplitsslcreate_default_contextsocketcreate_connection wrap_socket getpeercertOpenSSLcryptoload_certificate FILETYPE_ASN1dict get_subjectget_components get_issuer get_versionhexget_serial_number get_notBefore get_notAfterdigestget_signature_algorithmbase64 b64encoderangeget_extension_count get_extensionappendget_short_namestrr Exception) rrhostnamecontextsockssock cert_binaryx509rr#iexter r rfrom_url sH       &zSSLCertificate.from_urldatacCsLt|tr |dSt|trdd|DSt|tr$dd|DS|S)z2Helper method to decode bytes in certificate data.utf-8cSs0i|]\}}t|tr|dn|t|qS)rO) isinstancebytesdecoderr ).0kvr r r Xsz4SSLCertificate._decode_cert_data..cSsg|]}t|qSr )rr )rSitemr r r ]sz4SSLCertificate._decode_cert_data..)rPrQrRr0itemslist)rNr r rr Rs    z SSLCertificate._decode_cert_dataNfilepathcCs0tj|jddd}|rt|j|dddS|S)z Export certificate as JSON. Args: filepath (Optional[str]): Path to save the JSON file (default: None). Returns: Optional[str]: JSON string if successful, None otherwise. F)indent ensure_asciirOencodingN)jsondumpsr r write_text)r r[json_strr r rto_json`s  zSSLCertificate.to_jsonc Csz,tjtjjt|jd}tjtjj| d}|r*t |j |ddWdS|WSt y?}zWYd}~dSd}~ww)z Export certificate as PEM. Args: filepath (Optional[str]): Path to save the PEM file (default: None). Returns: Optional[str]: PEM string if successful, None otherwise. r rOr_N) r,r-r.r/r; b64decoder dump_certificate FILETYPE_PEMrRrrcrC)r r[rIpem_datarLr r rto_pemps& zSSLCertificate.to_pemcCsDzt|jd}|rt||WdS|WSty!YdSw)z Export certificate as DER. Args: filepath (Optional[str]): Path to save the DER file (default: None). Returns: Optional[bytes]: DER bytes if successful, None otherwise. r N)r;rfr r write_bytesrC)r r[der_datar r rto_ders  zSSLCertificate.to_dercC|jdiS)z#Get certificate issuer information.rr getr r r rrzSSLCertificate.issuercCrn)z$Get certificate subject information.rrorqr r rrrrzSSLCertificate.subjectcC|jddS)z$Get certificate validity start date.rrorqr r r valid_fromrrzSSLCertificate.valid_fromcCrs)z"Get certificate validity end date.rrtrorqr r r valid_untilrrzSSLCertificate.valid_untilcCrs)zGet certificate fingerprint.rrtrorqr r rrrrzSSLCertificate.fingerprint)rr )__name__ __module__ __qualname____doc__rrBrr staticmethodintrrMr rerjrQrmpropertyrrrurvrr r r rr s(1 r)rzr&r(r;ratypingrrr urllib.parserOpenSSL.cryptor,pathlibrrr r r rs