Trilltechnician/hack-house
1
1
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Updated client/client.py

Browse Source
This commit is contained in:
mirai 2022-12-01 10:18:44 +03:00
parent 776ca81577
commit 420d6eb845
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
client/client.py
View File

@ -83,10 +83,10 @@ class Client:
with open('private.pem', 'rb') as f: with open('private.pem', 'rb') as f:
self.privkey = rsa.PrivateKey.load_pkcs1(f.read()) self.privkey = rsa.PrivateKey.load_pkcs1(f.read())
with open("public.pem", 'rb') as f: with open("public.pem", 'rb') as f:
with requests.get(self.key_url, data={"pubkey": f.read(), "username": self.username}, stream=True) as r: r = requests.get(self.key_url, data={"pubkey": f.read(), "username": self.username}, stream=True)
message = r.raw.read(999) message = r.raw.read(999)
self.symetric_key = rsa.decrypt(message, self.privkey) self.symetric_key = rsa.decrypt(message, self.privkey)
self.fernet = Fernet(self.symetric_key) self.fernet = Fernet(self.symetric_key)
def _remove_keys(self) -> None: def _remove_keys(self) -> None:
os.remove("private.pem") os.remove("private.pem")