o gi,@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddl m Z ddl m Z ddlmZeeZGdd d Zed krYejd ed d ddSdS)a A license (or licence) is an official permission or permit to do, use, or own something (as well as the document of that permission or permit). A license is granted by a party (licensor) to another party (licensee) as an element of an agreement between those parties. In the case of a license issued by a government, the license is obtained by applying for it. In the case of a private party, it is by a specific agreement, usually in writing (such as a lease or other contract). The simplest definition is "A license is a promise not to sue," because a license usually either permits the licensed party to engage in an activity which is illegal, and subject to prosecution, without the license (e.g. fishing, driving an automobile, or operating a broadcast radio or television station), or it permits the licensed party to do something that would violate the rights of the licensing party (e.g. make copies of a copyrighted work), which, without the license, the licensed party could be sued, civilly, criminally, or both. N)hashes) serialization)padding)rsa)UUIDc@seZdZUdZdZejeed<dZ ejeed<dZ eje ed<dZ eje ed<e eejdeejdfd ejd ejfd d Ze d eejdedee ejde ejddfdedejd ejdede de ddfddZe dejedeje fddZe dejefddZe dejefdd Ze deje fd!d"Ze deje fd#d$Ze d%ejdefd&d'ZdS)(Licensea A license (or licence) is an official permission or permit to do, use, or own something (as well as the document of that permission or permit). A license is granted by a party (licensor) to another party (licensee) as an element of an agreement between those parties. In the case of a license issued by a government, the license is obtained by applying for it. In the case of a private party, it is by a specific agreement, usually in writing (such as a lease or other contract). N_LICENSE_COMPANY_LICENSE_USER_ID_LICENSE_VALID_FROM_IN_MS_LICENSE_VALID_UNTIL_IN_MSzprivate_key.pemzpublic_key.pemprivate_key_filepublic_key_filecCstjddd}t|d}||jtjjtjj t dWdn1s(wY| }t|d}||j tjjtj jdWddS1sPwYdS)Nii)public_exponentkey_sizewb)encodingformatencryption_algorithm)rr)rgenerate_private_keyopenwrite private_bytesrEncodingPEM PrivateFormatPKCS8 NoEncryption public_key public_bytes PublicFormatSubjectPublicKeyInfo)r r private_keyfrr#H/home/ubuntu/.local/lib/python3.10/site-packages/borb/license/license.py_create_key_pair;s,  "zLicense._create_key_pairzlicense_key.jsonzJ/home/joris/Code/borb-license-key-private/borb-license-key-private-001.pemi $company output_fileuser_idvalid_from_in_msvalid_until_in_msreturncCs||tj|ddtj|ddd}tj|dd}t|}t |d} t j | dd} Wdn1sDwY| |tt} t| } | |d<t |d } | tj|ddWddS1szwYdS) Nr'%Y-%m-%d %H:%M:%S)anonymous_user_idr(valid_from_as_strvalid_until_as_strindentrb)password license_keyw)datetime fromtimestampstrftimejsondumpsencodehashlibsha256 hexdigestrrload_pem_private_keyreadsignrPKCS1v15rSHA256base64 b64encodedecoder)r(r)r r*r+r, license_dict license_byteslicense_bytes_unsigned_hashr"r!license_bytes_signed_hash base64_license_bytes_signed_hashfhr#r#r$_create_license\s&     "zLicense._create_licensescCsH|durdSzttj|ddWStdt|YdS)Nr.r'z"Unable to convert %s to a datetime)intr9strptime timestamploggerwarningstr)rQr#r#r$_datetime_str_to_msszLicense._datetime_str_to_mscCtjS)z This function returns the company to which this License was granted :return: the company to which this License was granted )rrr#r#r#r$ get_companyzLicense.get_companycCrY)z This function returns the user_id to which this License was granted :return: the user_id to which this License was granted )rr r#r#r#r$ get_user_idr[zLicense.get_user_idcCrY)z This function returns the start of the validity period of this License :return: the start of the validity period of this License )rr r#r#r#r$get_valid_from_in_msr[zLicense.get_valid_from_in_mscCrY)z This function returns the end of the validity period of this License :return: the end of the validity period of this License )rr r#r#r#r$get_valid_until_in_msr[zLicense.get_valid_until_in_mspath_to_license_filec Csv|stdt|dSd}zt|d}t|}Wdn1s)wYWntdt|YdS|durEdS|dd }| dtj |dd  }t | }ttjd }|srJ|sxJ|d fD]}t|d }t|} Wdn1swYt|} z| | |tt|d dt_|ddt_t |ddt_!t |ddt_"t#t$j$%&d} tj!durtj!| krtdt||ddfdt_dt_dt_!dt_"WdStj"dur0tj"| kr0tdt||ddfdt_dt_dt_!dt_"WdSWdSYq}dS)z This function (attempts to) register a license (specified as a Path) :param path_to_license_file: the license to register :return: True if the license is valid, False otherwise zFile %s does not existFNrz(Unable to convert contents of %s to JSONr7r&r2r3 public_keyszpublic_key_001.pemr5r(r/r0r1r'zLicense %s is valid FROM %szN.A.zLicense %s is valid UNTIL %sT)'existsrUrVrWrr<loadsrCgetr>popr=r?r@rApathlibPath__file__parentis_dirrload_pem_public_keyrG b64decodeverifyrrErrFrrr rXr r rRr9nowrT) r_ json_dictrOlicense_hash_0license_bytes1license_hash_1key_dirr r"rencrypted_message now_in_msr#r#r$registerst           zLicense.register) __name__ __module__ __qualname____doc__rtypingOptionalrW__annotations__r r rRr staticmethodrfrgrhrir%rrdr9rnrTrPrXrZr\r]r^boolrvr#r#r#r$r#sp       . r__main__z borb (EZ)zT/home/joris/Code/borb-dev/tests/license/artifacts_test_register_license/license.jsonzJoris Schellekens)r(r)r*)rzrGr9r<loggingr{r?rfcryptography.hazmat.primitivesrr)cryptography.hazmat.primitives.asymmetricrrborb.license.uuidr getLoggerrwrUrrPrgr#r#r#r$s.      }