基于Python实现一个简单的注册机并生成卡密

pip install hashlib

# registration.py
import secrets

def generate_activation_key():
    activation_key = secrets.token_urlsafe(16)
    return activation_key

# registration.py
import hashlib

def hash_password(password):
    hashed_password = hashlib.sha256(password.encode()).hexdigest()
    return hashed_password

# registration.py
def register_user(username, password):
    hashed_password = hash_password(password)
    activation_key = generate_activation_key()

    # 存储用户信息和卡密，可以使用数据库或文件等方式
    user_data = {
        'username': username,
        'hashed_password': hashed_password,
        'activation_key': activation_key,
    }

    # 这里假设有个数据库类，用于存储用户信息
    database.save_user(user_data)

    return activation_key

# registration.py
def authenticate_user(username, password):
    user_data = database.get_user(username)

    if user_data:
        hashed_password = hash_password(password)

        if hashed_password == user_data['hashed_password']:
            return True
    return False

# registration.py
import secrets
import hashlib

class RegistrationSystem:
    def __init__(self):
        self.users = {}

    def generate_activation_key(self):
        activation_key = secrets.token_urlsafe(16)
        return activation_key

    def hash_password(self, password):
        hashed_password = hashlib.sha256(password.encode()).hexdigest()
        return hashed_password

    def register_user(self, username, password):
        hashed_password = self.hash_password(password)
        activation_key = self.generate_activation_key()

        user_data = {
            'username': username,
            'hashed_password': hashed_password,
            'activation_key': activation_key,
        }

        self.users[username] = user_data

        return activation_key

    def authenticate_user(self, username, password):
        user_data = self.users.get(username)

        if user_data:
            hashed_password = self.hash_password(password)

            if hashed_password == user_data['hashed_password']:
                return True
        return False

# 使用示例
registration_system = RegistrationSystem()
activation_key = registration_system.register_user('john_doe', 'secure_password')
print(f"Activation Key: {activation_key}")

authenticated = registration_system.authenticate_user('john_doe', 'secure_password')
print(f"Authentication Result: {authenticated}")

# registration.py
import secrets
import hashlib
import smtplib
from email.mime.text import MIMEText

class RegistrationSystem:
    def __init__(self):
        self.users = {}

    # ... 其他函数

    def send_verification_email(self, email, activation_key):
        subject = "Email Verification"
        body = f"Click the following link to verify your email: http://example.com/verify?activation_key={activation_key}"

        msg = MIMEText(body)
        msg['Subject'] = subject
        msg['From'] = 'noreply@example.com'
        msg['To'] = email

        # 这里假设有一个 SMTP 服务器，用于发送邮件
        with smtplib.SMTP('smtp.example.com') as server:
            server.sendmail('noreply@example.com', [email], msg.as_string())

    def register_user_with_email_verification(self, username, password, email):
        activation_key = self.register_user(username, password)
        self.send_verification_email(email, activation_key)
        return activation_key

# registration.py
import pyotp  # 需要安装 pyotp 库

class RegistrationSystem:
    def __init__(self):
        self.users = {}

    # ... 其他函数

    def enable_mfa(self, username):
        user_data = self.users.get(username)

        if user_data:
            totp = pyotp.TOTP(pyotp.random_base32())
            user_data['mfa_secret'] = totp.secret
            return totp.provisioning_uri(name=username, issuer_name='MyApp')

    def verify_mfa(self, username, token):
        user_data = self.users.get(username)

        if user_data and 'mfa_secret' in user_data:
            totp = pyotp.TOTP(user_data['mfa_secret'])
            return totp.verify(token)
        return False

# database.py
import sqlite3

class Database:
    def __init__(self):
        self.conn = sqlite3.connect('users.db')
        self.cursor = self.conn.cursor()
        self.create_table()

    def create_table(self):
        self.cursor.execute('''
            CREATE TABLE IF NOT EXISTS users (
                username TEXT PRIMARY KEY,
                hashed_password TEXT,
                activation_key TEXT,
                email TEXT,
                mfa_secret TEXT
            )
        ''')
        self.conn.commit()

    def save_user(self, user_data):
        self.cursor.execute('''
            INSERT INTO users (username, hashed_password, activation_key, email, mfa_secret)
            VALUES (?, ?, ?, ?, ?)
        ''', (
            user_data['username'],
            user_data['hashed_password'],
            user_data['activation_key'],
            user_data.get('email'),
            user_data.get('mfa_secret'),
        ))
        self.conn.commit()

    def get_user(self, username):
        self.cursor.execute('SELECT * FROM users WHERE username = ?', (username,))
        return dict(self.cursor.fetchone())