|
|
|
""" |
|
Created on Wed Jun 8 22:21:45 2022 |
|
|
|
@author: Abinash.m |
|
""" |
|
|
|
import sys, time |
|
import base64 |
|
import streamlit as st |
|
def printError(msg=''): |
|
if msg != "": |
|
print("Error: " + msg) |
|
print('Usage: ') |
|
print(sys.argv[0] + " [private key file] [consumer ID] [Key version]") |
|
print('') |
|
print('Example: ') |
|
print(sys.argv[0] + " private_key.pem 44444444-23f9-3333-2222-111111111111 1") |
|
print('') |
|
print('') |
|
sys.exit() |
|
|
|
def sign_data(private_key, data): |
|
from Crypto.PublicKey import RSA |
|
from Crypto.Signature import PKCS1_v1_5 |
|
from Crypto.Hash import SHA256 |
|
from base64 import b64encode, b64decode |
|
key = open(private_key, "r") |
|
rsakey = RSA.importKey(key.read()) |
|
signer = PKCS1_v1_5.new(rsakey) |
|
digest = SHA256.new() |
|
digest.update(data.encode('utf-8')) |
|
sign = signer.sign(digest) |
|
return b64encode(sign) |
|
|
|
def main(argv): |
|
global logData |
|
global hostname |
|
try: |
|
|
|
|
|
privateKey ='untitled1.txt' |
|
|
|
|
|
except: |
|
printError() |
|
epoch_time = int(time.time()) * 1000 |
|
data = 'd23d5ad5-b016-4a76-b6bb-c6bae0372374' + '\n' + str(epoch_time) + '\n' + '1' + '\n' |
|
|
|
|
|
sig = sign_data(privateKey, data).decode() |
|
my_dict = {'Authorization Token':sig,'Timestamp':epoch_time} |
|
st.json(my_dict) |
|
|
|
if __name__ == "__main__": |
|
main(sys.argv) |
|
|