|
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}")
|
|
|
|
|
