|
import base64 |
|
from hashlib import sha3_512 |
|
import json |
|
import random |
|
|
|
from datetime import datetime, timedelta, timezone |
|
|
|
from constants.headers import OPENAI_GET_HEADERS |
|
|
|
|
|
class ProofWorker: |
|
def __init__(self, difficulty=None, required=False, seed=None): |
|
self.difficulty = difficulty |
|
self.required = required |
|
self.seed = seed |
|
self.proof_token_prefix = "gAAAAABwQ8Lk5FbGpA2NcR9dShT6gYjU7VxZ4D" |
|
|
|
def get_parse_time(self): |
|
now = datetime.now() |
|
tz = timezone(timedelta(hours=8)) |
|
now = now.astimezone(tz) |
|
time_format = "%a %b %d %Y %H:%M:%S" |
|
return now.strftime(time_format) + " GMT+0800 (中国标准时间)" |
|
|
|
def get_config(self): |
|
cores = [8, 12, 16, 24] |
|
core = random.choice(cores) |
|
screens = [3000, 4000, 6000] |
|
screen = random.choice(screens) |
|
return [ |
|
str(core) + str(screen), |
|
self.get_parse_time(), |
|
4294705152, |
|
0, |
|
OPENAI_GET_HEADERS["User-Agent"], |
|
] |
|
|
|
def calc_proof_token(self, seed: str, difficulty: str): |
|
config = self.get_config() |
|
diff_len = len(difficulty) // 2 |
|
for i in range(100000): |
|
config[3] = i |
|
json_str = json.dumps(config) |
|
base = base64.b64encode(json_str.encode()).decode() |
|
hasher = sha3_512() |
|
hasher.update((seed + base).encode()) |
|
hash = hasher.digest().hex() |
|
if hash[:diff_len] <= difficulty: |
|
return "gAAAAAB" + base |
|
self.proof_token = ( |
|
self.proof_token_prefix + base64.b64encode(seed.encode()).decode() |
|
) |
|
return self.proof_token |
|
|
|
|
|
if __name__ == "__main__": |
|
seed, difficulty = "0.42665582693491433", "05cdf2" |
|
worker = ProofWorker() |
|
proof_token = worker.calc_proof_token(seed, difficulty) |
|
print(f"proof_token: {proof_token}") |
|
|
|
|
