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