ESP32使用tcp发送消息，AT指令修改IP和WIFI名字，PYTHON接收 - MKT-porter

随笔- 1525 文章- 2 评论- 66 阅读- 128万

ESP32使用tcp发送消息，AT指令修改IP和WIFI名字，PYTHON接收

python

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

import socket
import threading
import json
from flask import Flask
from concurrent.futures import ThreadPoolExecutor
import time
 
wark_bstate=0
 
def handle_device_data(device_id, data):
    if device_id == 'class1':
        temperature = data.get('temperature')
        humidity = data.get('humidity')
        print(f"Device 1 - Temperature: {temperature}, Humidity: {humidity}")
    elif device_id == 'class2':
        smoke_level = data.get('smoke')
        print(f"Device 2 - Smoke Level: {smoke_level}")
    elif device_id == 'class3':
        light_intensity = data.get('light')
        print(f"Device 3 - Light Intensity: {light_intensity}")
    else:
        print(f"Unknown device ID: {device_id}")
 
 
 
# 处理客户端连接的函数
def handle_client(client_socket):
    while True:
        try:
 
            if wark_bstate==1:
                print("TCP服务器client_socket 结束")
                break
 
 
            # 接收数据长度（4字节）
            data_length_bytes = client_socket.recv(4)
            if not data_length_bytes:
                break
 
            # 将数据长度转换为整数
            data_length = int.from_bytes(data_length_bytes, byteorder='big')
 
            # 接收JSON数据
            json_data = client_socket.recv(data_length).decode('utf-8')
            if not json_data:
                break
 
            # 解析JSON数据
            data = json.loads(json_data)
 
            # 判断数据是否存在
            if "device_ID" not in data or "sensor_class" not in data :
                print("Error: Incomplete data received")
                continue  # 或者根据需求执行其他逻辑
            device_id = data.get("device_ID")
            sensor_class =  data.get("sensor_class")
            # 打印解析结果
            print(f"Device ID: {device_id}")
            print(f"Device ID: {sensor_class}")
 
            handle_device_data(sensor_class,data)
 
        except Exception as e:
            print(f"Error: {e}")
            break
 
    # 关闭客户端连接
    client_socket.close()
 
# 启动TCP服务器
def start_server(ip, port):
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind((ip, port))
    server.listen(5)
    print(f"Server started on {ip}:{port}")
 
    # 使用线程池限制最大线程数
    with ThreadPoolExecutor(max_workers=10) as executor:
        while True:
 
            if wark_bstate==1:
                print("TCP服务器结束")
                break
            client_socket, addr = server.accept()
            print(f"Connection from {addr}")
 
            # 提交任务到线程池
            executor.submit(handle_client, client_socket)
 
# 启动Flask服务器
def start_flask_server():
    app = Flask(__name__)
 
    @app.route('/')
    def index():
        return "Hello from Flask!"
 
    app.run(host='0.0.0.0', port=5000)
 
if __name__ == "__main__":
    # 启动TCP服务器的线程
    tcp_thread = threading.Thread(target=start_server, args=("0.0.0.0", 8080))
    tcp_thread.daemon = True  # 设置为伴随线程
    tcp_thread.start()
 
    # 启动Flask服务器的线程
    flask_thread = threading.Thread(target=start_flask_server)
    flask_thread.daemon = True  # 设置为伴随线程
    flask_thread.start()
 
    # 主线程等待（实际上它们会一直运行，直到主线程退出）
    try:
        while True:
            time.sleep(2)
            print("主线程...")
            pass
    except KeyboardInterrupt:
        wark_bstate=1
        print("Main thread interrupted. Exiting...")

可以修改代码目前只在上电时候连接一次

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

#include <Preferences.h>
#include <WiFi.h>
#include <ArduinoJson.h>
#include <DHT.h>
String Device_ID = "ESP32_xxx";
String wifi_name ="love";
String wifi_pwd ="love123456";
 
String sensor_class ="class1";
 
// 定义常量和全局变量
#define DHTPIN 4          // DHT传感器连接的引脚
#define DHTTYPE DHT11     // DHT11传感器类型
 
DHT dht(DHTPIN, DHTTYPE);
Preferences preferences;
WiFiClient client;
 
char sever_ip[16] = "192.168.1.103";  // 默认IP地址
int sever_port = 8080;                // 默认端口
bool newIPPortReceived = false; // 标记是否收到新的IP和端口
String deviceID;                // 设备出厂ID
 
// 生成设备ID
String generateDeviceID() {
  // 生成一个唯一的ID，例如基于MAC地址
  String mac = WiFi.macAddress();
  mac.replace(":", "");
  Device_ID= "ESP32_" + mac;
  return Device_ID;
}
 
 
// 处理AT指令
void processATCommand(String command) {
  if (command.startsWith("AT+IP=")) {
    String newIP = command.substring(6);
    newIP.toCharArray(sever_ip, 16);
    preferences.putString("ip", newIP);
    newIPPortReceived = true;
    Serial.println("New IP set: " + newIP);
  } else if (command.startsWith("AT+PORT=")) {
    int newPort = command.substring(8).toInt();
    sever_port = newPort;
    preferences.putInt("port", newPort);
    newIPPortReceived = true;
    Serial.println("New Port set: " + String(newPort));
  }
  else if (command.startsWith("AT+WIFINAME=")) {
    String newWIFINAME  = command.substring(12);
    wifi_name = newWIFINAME;
    preferences.putString("wifi_name", newWIFINAME);
    newIPPortReceived = true;
    Serial.println("New WIFINAME set: " + String(newWIFINAME));
  }
  else if (command.startsWith("AT+WIFIPWD=")) {
    String newWIFIPWD= command.substring(11);
    wifi_pwd = newWIFIPWD;
    preferences.putString("wifi_pwd", newWIFIPWD);
    newIPPortReceived = true;
    Serial.println("New WIFIPWDrt set: " + String(newWIFIPWD));
  }
  else {
    Serial.println("Unknown command");
  }
}
 
// 连接TCP服务器
void connectToServer() {
  if (client.connect(sever_ip, sever_port)) {
    Serial.println("Connected to server");
  } else {
    Serial.println("Connection failed");
  }
}
 
// 发送温湿度数据
void sendSensorData() {
  float humidity = dht.readHumidity();
  float temperature = dht.readTemperature();
   
  if (isnan(humidity) || isnan(temperature)) {
    Serial.println("Failed to read from DHT sensor!");
    return;
  }
   
  // 创建JSON对象
  StaticJsonDocument<200> jsonDoc;
  jsonDoc["device_ID"] = deviceID; // 添加设备ID
  jsonDoc["sensor_class"] = sensor_class;
  jsonDoc["temperature"] = temperature;
  jsonDoc["humidity"] = humidity;
   
 
  String msg = "temperature:"+ String(temperature)+"   humidity: "+ String(humidity);
  Serial.println(msg);
   
   
  // 序列化JSON
  char jsonBuffer[200];
  serializeJson(jsonDoc, jsonBuffer);
   
    // 计算JSON数据长度
  int dataLength = strlen(jsonBuffer);
 
  // 发送数据长度（4字节）
  if (client.connected()) {
    client.write((uint8_t*)&dataLength, sizeof(dataLength));
    // 发送JSON数据
    client.print(jsonBuffer);
    Serial.println("Data sent: " + String(jsonBuffer));
  } else {
    Serial.println("Client not connected");
  }
   //client.stop();
}
 
// 初始化设置
void setup() {
  Serial.begin(115200);
  dht.begin();
 
  // 初始化Pr
  preferences.begin("wifi-config", false);
   
  // 从Preferences中读取保存的IP和端口
  String savedIP = preferences.getString("ip", sever_ip);// 如果没有该元素则返回默认值"555"
  int savedPort = preferences.getInt("port", 8080);
 
  String wifi_name_ = preferences.getString("wifi_name", String(wifi_name));// 如果没有该元素则返回默认值"555"
  String wifi_pwd_ = preferences.getString("wifi_pwd", String(wifi_pwd));
   
  savedIP.toCharArray(sever_ip, 16);
  sever_port = savedPort;
   
  // 读取或生成设备ID
  deviceID = preferences.getString("deviceID", "");
  if (deviceID == "") {
    deviceID = generateDeviceID(); // 生成唯一ID
    preferences.putString("deviceID", deviceID); // 保存ID
  }
   
  Serial.println("Device ID: " + deviceID);
  Serial.println("Saved IP: " + String(sever_ip));
  Serial.println("Saved Port: " + String(sever_port));
  Serial.println("Saved wifi_name_: " + String(wifi_name_));
  Serial.println("Saved wifi_pwd_: " + String(wifi_pwd_));
   
  // 连接WiFi
  WiFi.begin(wifi_name_.c_str(), wifi_pwd_.c_str());
  while (WiFi.status() != WL_CONNECTED) {
    delay(500);
    Serial.print(".");
  }
  Serial.println("WiFi connected");
   
  // 连接TCP服务器
  connectToServer();
}
 
// 主循环
void loop() {
  // 检查是否有新的AT指令
  if (Serial.available()) {
    String command = Serial.readStringUntil('\n');
    processATCommand(command);
  }
   
  // 如果收到新的IP和端口，重新连接服务器
  if (newIPPortReceived) {
    connectToServer();
    newIPPortReceived = false;
 
    // 软件重启
    Serial.println("即将重启...");
    ESP.restart();
  }
   
  // 每2秒发送一次温湿度数据
  static unsigned long lastSendTime = 0;
  if (millis() - lastSendTime >= 2000) {
    sendSensorData();
    lastSendTime = millis();
  }
}