域名授权管理系统使用方法

import hashlib
import time

def generate_auth_file(domain, secret_key, duration):
    expiration_time = int(time.time()) + duration
    auth_string = f"{domain}:{expiration_time}:{secret_key}"
    auth_hash = hashlib.sha256(auth_string.encode()).hexdigest()
    with open(f"/{domain}/auth.txt", "w") as f:
        f.write(f"{expiration_time}:{auth_hash}")

def verify_auth_file(domain):
    with open(f"/{domain}/auth.txt", "r") as f:
        content = f.read().strip().split(":")
        expiration_time = int(content[0])
        auth_hash = content[1]
    current_time = int(time.time())
    if current_time > expiration_time:
        return False, "Authorization expired"
    auth_string = f"{domain}:{expiration_time}:{secret_key}"
    new_hash = hashlib.sha256(auth_string.encode()).hexdigest()
    if new_hash != auth_hash:
        return False, "Invalid authorization"
    return True, "Authorization successful"

# 示例使用
domain = "example.com"
secret_key = "your_secret_key"
duration = 86400  # 24小时
generate_auth_file(domain, secret_key, duration)
is_valid, message = verify_auth_file(domain)
print(message)