域名授权管理系统源码

import sqlite3

# 创建数据库连接
conn = sqlite3.connect('domain_auth.db')
cursor = conn.cursor()

# 创建授权表
cursor.execute('''
CREATE TABLE IF NOT EXISTS domain_auth (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    domain TEXT NOT NULL,
    user TEXT NOT NULL,
    permission TEXT NOT NULL
)
''')

# 授权操作
def authorize_domain(domain, user, permission):
    cursor.execute('INSERT INTO domain_auth (domain, user, permission) VALUES (?, ?, ?)', (domain, user, permission))
    conn.commit()

# 查询授权信息
def get_authorization(domain, user):
    cursor.execute('SELECT permission FROM domain_auth WHERE domain = ? AND user = ?', (domain, user))
    result = cursor.fetchone()
    return result[0] if result else None

# 示例使用
authorize_domain('example.com', 'user1', 'read')
print(get_authorization('example.com', 'user1'))  # 输出: read

# 关闭连接
conn.close()