锦中人工智能助手

import base64

from cryptography.fernet import Fernet

# 生成密钥

key = Fernet.generate_key()

cipher = Fernet(key)

# 加密数据

data = b"Student ID: 123456"

encrypted_data = cipher.encrypt(data)

print("Encrypted:", encrypted_data)

# 解密数据

decrypted_data = cipher.decrypt(encrypted_data)

print("Decrypted:", decrypted_data.decode())

from flask import Flask, request, jsonify

import jwt

import datetime

app = Flask(__name__)

# 模拟数据库中的用户

users = {

"admin": {"password": "123456", "role": "admin"},

"user": {"password": "654321", "role": "user"}

}

SECRET_KEY = 'your-secret-key'

def generate_token(username):

payload = {

'username': username,

'exp': datetime.datetime.utcnow() + datetime.timedelta(hours=1)

}

token = jwt.encode(payload, SECRET_KEY, algorithm='HS256')

return token

def verify_token(token):

try:

payload = jwt.decode(token, SECRET_KEY, algorithms=['HS256'])

return payload['username']

except:

return None

@app.route('/login', methods=['POST'])

def login():

data = request.get_json()

username = data.get('username')

password = data.get('password')

if username in users and users[username]['password'] == password:

token = generate_token(username)

return jsonify({"token": token})

else:

return jsonify({"error": "Invalid credentials"}), 401

@app.route('/protected', methods=['GET'])

def protected():

token = request.headers.get('Authorization')

if not token:

return jsonify({"error": "Missing token"}), 401

user = verify_token(token)

if not user:

return jsonify({"error": "Invalid token"}), 401

return jsonify({"message": f"Welcome, {user}!"})

if __name__ == '__main__':

app.run(debug=True)

import logging

# 配置日志

logging.basicConfig(filename='app.log', level=logging.INFO,

format='%(asctime)s - %(levelname)s - %(message)s')

def log_login(username):

logging.info(f"User {username} logged in successfully.")

def log_failure(username):

logging.warning(f"Failed login attempt for user {username}.")