Hansimov commited on
Commit
0616d17
1 Parent(s): c5f3581

:zap: [Enhance] Replace Crypto with hashlib

Browse files
Files changed (1) hide show
  1. networks/proof_worker.py +8 -6
networks/proof_worker.py CHANGED
@@ -1,8 +1,10 @@
1
- import json
2
  import base64
 
 
3
  import random
 
4
  from datetime import datetime, timedelta, timezone
5
- from Crypto.Hash import SHA3_512
6
  from constants.headers import OPENAI_GET_HEADERS
7
 
8
 
@@ -40,10 +42,10 @@ class ProofWorker:
40
  config[3] = i
41
  json_str = json.dumps(config)
42
  base = base64.b64encode(json_str.encode()).decode()
43
- hasher = SHA3_512.new()
44
  hasher.update((seed + base).encode())
45
- hash = hasher.digest()
46
- if hash.hex()[:diff_len] <= difficulty:
47
  return "gAAAAAB" + base
48
  self.proof_token = (
49
  self.proof_token_prefix + base64.b64encode(seed.encode()).decode()
@@ -56,4 +58,4 @@ if __name__ == "__main__":
56
  worker = ProofWorker()
57
  proof_token = worker.calc_proof_token(seed, difficulty)
58
  print(f"proof_token: {proof_token}")
59
- # python -m networks.proof_of_work
 
 
1
  import base64
2
+ from hashlib import sha3_512
3
+ import json
4
  import random
5
+
6
  from datetime import datetime, timedelta, timezone
7
+
8
  from constants.headers import OPENAI_GET_HEADERS
9
 
10
 
 
42
  config[3] = i
43
  json_str = json.dumps(config)
44
  base = base64.b64encode(json_str.encode()).decode()
45
+ hasher = sha3_512()
46
  hasher.update((seed + base).encode())
47
+ hash = hasher.digest().hex()
48
+ if hash[:diff_len] <= difficulty:
49
  return "gAAAAAB" + base
50
  self.proof_token = (
51
  self.proof_token_prefix + base64.b64encode(seed.encode()).decode()
 
58
  worker = ProofWorker()
59
  proof_token = worker.calc_proof_token(seed, difficulty)
60
  print(f"proof_token: {proof_token}")
61
+ # python -m networks.proof_worker