统一信息平台与AI助手在登录系统中的应用

from flask import Flask, request, jsonify

import jwt

import datetime

app = Flask(__name__)

# 模拟统一信息平台的用户数据库

users = {

"user1": {"password": "pass123", "role": "user"},

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

}

SECRET_KEY = 'your-secret-key'

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

def login():

data = request.json

username = data.get('username')

password = data.get('password')

if not username or not password:

return jsonify({"error": "Missing username or password"}), 400

user = users.get(username)

if not user or user['password'] != password:

return jsonify({"error": "Invalid username or password"}), 401

# 生成JWT token

payload = {

'username': username,

'exp': datetime.datetime.utcnow() + datetime.timedelta(minutes=30)

}

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

return jsonify({

"message": "Login successful",

"token": token,

"role": user['role']

})

if __name__ == '__main__':

app.run(debug=True)

import random

def check_anomaly(login_data):

# 简单的异常检测逻辑

# 假设用户在凌晨3点登录，视为异常

hour = login_data.get('hour', 0)

ip = login_data.get('ip', '')

user_agent = login_data.get('user_agent', '')

if hour < 6 or hour > 22:

return True, "Unusual login time detected."

if 'bot' in user_agent.lower():

return True, "Possible bot activity detected."

if '192.168.' in ip:

return False, "Local IP address, no anomaly."

# 随机返回结果作为模拟

if random.random() < 0.1:

return True, "Suspicious activity detected."

return False, "No anomalies found."

# 示例调用

login_data = {

'hour': 3,

'ip': '192.168.1.1',

'user_agent': 'Mozilla/5.0'

}

is_anomaly, message = check_anomaly(login_data)

print(f"Anomaly: {is_anomaly}, Message: {message}")