• Flask初体验

    这里有一份展示Flask与Python的协同代码，Flask的web页面展示了系统的一个暴露的公共tcp port连接的所有用户ip:port列表。

    做完才发现没有什么用处，我的本意是做一个reverse的ssh或者telnet终端。看点有几个：

    1. 我原本是打算用multiprocessing.queue在python和Flask的web代码间交互，后来发现不必要。
    2. web客户处理的 app级变量最简单的方法是放在：app.config[""]这个字典里。app变量如果更新记得要回写回去。
    3. python程序最好通过post方法，与Flask交互，更新app变量，例子中包含一个完整的示例：python post, flask response 
    4. web页面为了自动更新一些服务器变量，有一个自刷新的代码。
    5. 命令行可以主动切换web port和那个tcp port.

    1.python: reverse_ssh_server.sh

    1. import socket
    2. import threading
    3. from flask import Flask, render_template, jsonify, request
    4. import requests
    5. from sys import argv
    6. import multiprocessing
    7.  
    8. TCP_PORT = 8888
    9. WEB_PORT = 5000
    10. CLIENT_MAX_LEN = 15
    11.  
    12. app= Flask(__name__, template_folder="./web") #, static_folder="./web"
    13. connected_devices = []
    14.  
    15. # TCP服务器
    16. def tcp_server():
    17.     host = '0.0.0.0'
    18.     port = TCP_PORT
    19.  
    20.     server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    21.     try:
    22.         server_socket.bind((host, port))
    23.         server_socket.listen(15)
    24.     except Exception as e:
    25.         print(f"app port {port} bind error", e)
    26.         return;
    27.  
    28.     print(f"TCP服务器已启动，监听端口 {port}...")
    29.  
    30.     while True:
    31.         try:
    32.             if(app is None): 
    33.                 continue;
    34.             client_socket, addr = server_socket.accept()
    35.             #print(f"连接来自 {addr}")
    36.             if(len(connected_devices)>CLIENT_MAX_LEN):
    37.                 client_socket.close()
    38.                 continue
    39.             json = {}
    40.             json["dev"] = f"{addr}"
    41.             dev_login(json)
    42.             print(connected_devices)
    43.         except Exception as e:
    44.             print("client accept error.", e);
    45.             server_socket.close();
    46.             return;
    47.  
    48. def dev_login(dev_addr):
    49.     url = f'http://localhost:{WEB_PORT}/api/devices/dev_login'
    50.     headers = {'Content-Type': 'application/json'}
    51.     response = requests.post(url, json=dev_addr, headers=headers)
    52.     print("dev_login", response.json())  # 打印服务器返回的 JSON 响应
    53.  
    54. # Web页面
    55. @app.route('/')
    56. def index():
    57.     connected_devices = app.config['connected_devices']
    58.     return render_template('index.html', devices=connected_devices)
    59.  
    60. @app.route('/api/devices')
    61. def get_device_list():
    62.     connected_devices = app.config['connected_devices']
    63.     # 这里应该是获取设备列表的逻辑，暂时用一个简单的列表代替
    64.     # Assuming connected_devices is a multiprocessing.Queue
    65.     ret = jsonify(connected_devices)
    66.     print(ret)
    67.     return ret
    68.  
    69. @app.route('/api/devices/dev_login', methods=['POST'])
    70. def handle_post():
    71.     # 获取 POST 请求中的 JSON 数据
    72.     json_data = request.json["dev"]
    73.     connected_devices = app.config['connected_devices']
    74.     connected_devices.append(json_data);
    75.     app.config['connected_devices'] = connected_devices
    76.     # 在这里处理 JSON 数据，这里只是简单地返回接收到的 JSON 数据
    77.     print("on_handle_post:", json_data, connected_devices)
    78.     return jsonify(json_data)
    79.  
    80. if __name__ == '__main__':
    81.     if len(argv)>=2:
    82.         WEB_PORT = int(argv[1])
    83.     if len(argv)>=3:
    84.         TCP_PORT = int(argv[2])
    85.     print(f"web port:{WEB_PORT}, tcp port:{TCP_PORT}")
    86.     # 启动TCP服务器和Web应用
    87.     app.config['connected_devices'] = connected_devices
    88.     tcp_thread = threading.Thread(target=tcp_server)
    89.     tcp_thread.start()
    90.     app.run(debug=True, port=WEB_PORT)

    2.web Flask使用的index.html

    1. html>
    2. <html>
    3. <head>
    4.     <title>Device Listtitle>
    5. head>
    6. <body>
    7.     <h1>Device Listh1>
    8.     <ul id="device-list">
    9.         
    10.         <li>ip:portli>
    11.     ul>
    12.  
    13.     <script>
    14.         function updateDeviceList() {
    15.             // 发送 AJAX 请求获取设备列表
    16.             var xhr = new XMLHttpRequest();
    17.             xhr.onreadystatechange = function() {
    18.                 if (xhr.readyState == XMLHttpRequest.DONE) {
    19.                     if (xhr.status == 200) {
    20.                         // 解析 JSON 响应
    21.                         var devices = JSzeON.parse(xhr.responseText);
    22.                         // 清空设备列表
    23.                         document.getElementById("device-list").innerHTML = "";
    24.                         // 遍历设备列表并添加到页面中
    25.                         devices.forEach(function(device) {
    26.                             var li = document.createElement("li");
    27.                             li.textContent = device; //device.name
    28.                             document.getElementById("device-list").appendChild(li);
    29.                         });
    30.                     }
    31.                 }
    32.             };
    33.             xhr.open("GET", "/api/devices", true);
    34.             xhr.send();
    35.         }
    36.         
    37.         // 页面加载时立即执行一次更新
    38.         updateDeviceList();
    39.         
    40.         // 定时刷新，1秒执行一次
    41.         setInterval(updateDeviceList, 1000);
    42.     script>
    43. body>
    44. html>

    3.执行效果：

    sudo -E python3.9 reverse_ssh_server.py 5000 3002

    [sudo] root 的密码： 
    web port:5000, tcp port:3002
    TCP服务器已启动，监听端口 3002...
     * Serving Flask app 'reverse_ssh_server'
     * Debug mode: on
    WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
     * Running on http://127.0.0.1:5000
    Press CTRL+C to quit

    4.进阶

    4.1 Post 图片及其他参数给Flask

    1. #!/usr/bin/env python3
    2. # -*- coding: utf-8 -*-
    3. # 获取当前脚本文件所在目录的父目录，并构建相对路径
    4. import os
    5. import sys
    6. current_dir = os.path.dirname(os.path.abspath(__file__))
    7. project_path = os.path.join(current_dir, '..')
    8. sys.path.append(project_path)
    9. sys.path.append(current_dir)
    10. import gpLog
    11. import paho.mqtt.client as mqtt
    12. from datetime import datetime
    13. import cv2
    14. import requests
    15. import numpy as np
    16.  
    17. def gpWebPost_PostImage(url:str, cv2Image, ai_class:int):
    18.     if not url.startswith("http://"):
    19.         # 设置 POST 请求的 URL
    20.         url = f'http://{url}'
    21.  
    22.     # 设置要发送的文件
    23.     files = {
    24.         'ai_snapshot': cv2Image  # 将文件打开为二进制模式
    25.     }
    26.     data = {
    27.     'ai_class': 1,
    28.     }
    29.  
    30.     jsonArg ={}
    31.     jsonArg["ai_class"]=ai_class
    32.     # 发送 POST 请求
    33.     response = requests.post(url, files=files, json=jsonArg, data=data)
    34.  
    35.     # 检查响应
    36.     if response.status_code == 200:
    37.         return True
    38.     else:
    39.         return False
    40.  
    41. def gpWebPost_TestPostImage():
    42.     # 读取图像
    43.     url = "http://192.168.0.1:8080/aiimage/dev_0001/ch01"
    44.     image = cv2.imread(r'/home/lubancat/图片/Rumelhart_rect.jpg')
    45.     
    46.  
    47.     # 将图像转换为字节流
    48.     _, img_encoded = cv2.imencode('.jpg', image)
    49.     image_data = img_encoded.tobytes()
    50.     gpWebPost_PostImage(url, image_data, 1)

  • 相关阅读:
    3-docker安装centos7
    Qt 关于窗口全屏显示与退出全屏的实现
    ubuntu系统使用bing wallpaper壁纸
    SQL Prompt10 安装激活教程，让你写sql 如鱼得水
    static学习
    如何读取resources目录下的文件路径（九种方式）
    【云原生】docker环境中安装mysql、redis服务
    旋转框目标检测mmrotate v0.3.1 训练DOTA数据集（二）
    idea无法添加spring MVC框架支持
    Ubuntu服务器的GitLab部署
  • 原文地址：https://blog.csdn.net/twicave/article/details/139289101