smartpi 0.1.1__py3-none-any.whl → 0.1.2__py3-none-any.whl

smartpi/__init__.py

@@ -1,4 +1,4 @@
-from smartpi import module
+__all__ = ["base_driver"]
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"
 

smartpi/base_driver.py

@@ -0,0 +1,501 @@
+# coding=utf-8
+import serial,time,struct
+from typing import List, Optional
+from collections import deque
+
+
+# 命令常量
+BOOT_UPDATE_H = 0XFF
+BOOT_UPDATE_L = 0XFF
+READ_MODEL_H = 0XFF
+READ_MODEL_L = 0X01
+READ_VERSION_H = 0XFF
+READ_VERSION_L = 0X02
+READ_FACTORY_H = 0XFF
+READ_FACTORY_L = 0X03
+READ_HW_ID_H = 0XFF
+READ_HW_ID_L = 0X04
+READ_NAME_H = 0XFF
+READ_NAME_L = 0X05
+WRITE_NAME_H = 0XFF
+WRITE_NAME_L = 0X06
+READ_CONNECT_H = 0XFF
+READ_CONNECT_L = 0X07
+READ_BAT_H = 0XFF
+READ_BAT_L = 0X0C
+
+UPDATE_REQUEST_H = 0XFF
+UPDATE_REQUEST_L = 0X10
+MAX_COM_LEN_H = 0XFF
+MAX_COM_LEN_L = 0X11
+DL_MESSAGE_H = 0XFF
+DL_MESSAGE_L = 0X12
+READ_STATUS_H = 0XFF
+READ_STATUS_L = 0X13
+PAGE_CHECK_H = 0XFF
+PAGE_CHECK_L = 0X14
+PAGE_SEND_H = 0XFF
+PAGE_SEND_L = 0X15
+
+READ_PERIPH_H = 0X01
+READ_PERIPH_L = 0X01
+SINGLE_OP_H = 0X01
+SINGLE_OP_L = 0X02
+MODE_CHANGE_H = 0X01
+MODE_CHANGE_L = 0X03
+SEND_CYCLE_H = 0X01
+SEND_CYCLE_L = 0X04
+
+frames_queue = deque()
+buffer = bytearray()
+HEADER = bytes.fromhex('86 AB')  # 帧头
+FOOTER = bytes.fromhex('CF')     # 帧尾
+MIN_FRAME_LEN = 9                # 最小帧长度
+
+# 串口配置参数
+SERIAL_PORT = "/dev/ttyS3"  # 串口设备路径
+BAUD_RATE = 115200          # 波特率
+TIMEOUT = 0.1                # 读取超时时间（秒）
+
+ser = serial.Serial(
+            port=SERIAL_PORT,
+            baudrate=BAUD_RATE,
+            bytesize=serial.EIGHTBITS,  # 8位数据位
+            parity=serial.PARITY_NONE,  # 无校验位
+            stopbits=serial.STOPBITS_ONE,  # 1位停止位
+            timeout=TIMEOUT,
+            xonxoff=False,  # 关闭软件流控
+            rtscts=False,   # 关闭硬件流控
+        
+        )
+
+
+def uart3_init() -> Optional[serial.Serial]:
+    """初始化串口"""
+    try:        
+        if ser.is_open:
+            print("UART3初始化成功")
+            return ser
+        else:
+            print("Error opening UART3")
+            return None
+    except Exception as e:
+        print(f"Error opening UART3: {e}")
+        return None
+
+def calculate_pro_check(command_h: int, command_l: int, data: List[bytes] = None) -> int:    
+    if data:
+        len_data = len(data)
+        base_sum = 0x86 + 0xAB + 0x00 + 0x09 + len_data + command_h + command_l + 0x01 + 0xCF
+        base_sum += sum(data)
+    else:
+        base_sum = 0x86 + 0xAB + 0x00 + 0x09 + command_h + command_l + 0x01 + 0xCF
+        
+    return base_sum % 256  # 确保结果为单字节（原C代码未取模，需根据实际协议调整）
+
+def write_data(command_h: int, command_l: int, send_data: bytes= None) -> Optional[bytes]:                   
+    buffer = bytearray()
+    HEADER = bytes.fromhex('86 AB')  # 帧头
+    FOOTER = bytes.fromhex('CF')     # 帧尾
+    MIN_FRAME_LEN = 9                # 最小帧长度           
+    if send_data:            
+        pro_check = calculate_pro_check(command_h, command_l, list(send_data))
+        send_packet = [0x86, 0xAB, (0x09+len(send_data))//256, (0x09+len(send_data))%256, command_h, command_l, *send_data, 0x01, pro_check, 0xCF]
+    else:
+        pro_check = calculate_pro_check(command_h, command_l)
+        send_packet = [0x86, 0xAB, 0x00, 0x09, command_h, command_l, 0x01, pro_check, 0xCF]
+    send_bytes = bytes(send_packet)
+    
+    ser.write(send_bytes)
+                 
+        
+def check_frame(frame: bytes) -> bool:
+    # 简化版校验，实际应根据协议实现
+    if len(frame) < 9:
+        return False
+        
+    # 计算校验和
+    calculated_checksum = (sum(frame[:-2])+frame[-1])%256
+    frame_checksum = frame[-2]  # 倒数第二个字节是校验和
+    
+    return calculated_checksum == frame_checksum
+
+def process_received_data():
+    global buffer
+    # 读取所有可用数据
+    data = ser.read(ser.in_waiting or 1)
+    if data:
+        buffer.extend(data)
+    while len(buffer)>=2:
+#            for x in buffer:
+#                print(f"{x:02X}", end=' ')
+#            print("\n")
+        # 1. 查找帧头
+        start_idx = buffer.find(HEADER)
+        if start_idx == -1:
+            # 没有找到帧头，清空无效数据（保留最后可能的部分帧头）
+            #print("Header no found")
+            if len(buffer) > len(HEADER) - 1:
+                buffer = buffer[-len(HEADER) + 1:]
+            return                 
+        # 2. 检查帧头后的长度字段是否足够
+        if start_idx + 4 > len(buffer):
+            # 长度字段不完整，等待更多数据
+            return                
+        # 3. 解析帧长度
+        frame_length = (buffer[start_idx + 2] << 8) + buffer[start_idx + 3]          
+        # 4. 检查完整帧是否已到达
+        end_idx = start_idx + frame_length - 1
+        if end_idx >= len(buffer):
+            # 完整帧尚未完全到达
+            return                
+        # 5. 检查帧尾
+        if buffer[end_idx] != ord(FOOTER):
+            # 跳过当前帧头，继续查找
+            print("End no found")
+            buffer = buffer[start_idx + 1:]
+            continue               
+        # 6. 提取完整帧
+        frame = buffer[start_idx:end_idx + 1]
+        frames_queue.append(frame)            
+        # 7. 从缓冲区移除已处理帧
+        buffer = buffer[end_idx + 1:]        
+    # 处理所有完整帧
+    if len(frames_queue) > 0:
+        frame = frames_queue.popleft()
+        if check_frame(frame):
+            return frame
+        else:
+            print("Check byte error")
+                
+# 功能函数类
+##############################################################################读取设备信息
+"""固件升级模式"""
+def boot_update() -> None:
+    write_data(BOOT_UPDATE_H, BOOT_UPDATE_L)       
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+            if display_data[0]==0X03:
+                print("固件更新启动\n")
+            elif display_data[0]==0X01:
+                print("擦除中......",f"{10*display_data[1]}","%\n")
+            elif display_data[0]==0X02:
+                print("升级中......",f"{10*display_data[1]}","%\n")              
+            elif display_data[0]==0X05:
+                print("更新成功！\n")
+            elif display_data[0]==0X06:
+                print("更新失败！\n")
+            else:
+                for x in display_data:
+                    print(f"{x:02X}", end=' ')
+                print("\n")
+            
+
+"""读取设备型号"""
+def read_device_model() -> Optional[bytes]:
+    write_data(READ_MODEL_H, READ_MODEL_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"设备型号: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""读取版本号"""
+def read_version() -> Optional[bytes]:
+    write_data(READ_VERSION_H, READ_VERSION_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"版本号: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""读取工厂信息"""
+def read_factory_data() -> Optional[bytes]:
+    write_data(READ_FACTORY_H, READ_FACTORY_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"厂家信息: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""读取硬件ID"""
+def read_hardware_ID() -> Optional[bytes]:
+    write_data(READ_HW_ID_H, READ_HW_ID_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"硬件ID: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+"""读取设备名称"""
+def read_device_name() -> Optional[bytes]:        
+    write_data(READ_NAME_H, READ_NAME_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"设备名称: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+"""设置设备名称"""
+def write_device_name(send_data: str) -> Optional[bytes]:
+    data_bytes = send_data.encode('utf-8')
+    write_data(WRITE_NAME_H, WRITE_NAME_L, data_bytes)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"设置状态: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+"""读取连接方式"""#长度位少了1
+def read_connected() -> Optional[bytes]: 
+    write_data(READ_CONNECT_H, READ_CONNECT_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"连接方式: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+    
+"""读取电池电量"""        
+def read_battery() -> Optional[bytes]: 
+    write_data(READ_BAT_H, READ_BAT_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+#                for x in display_data:
+#                    print(f"{x:02X}", end=' ')
+            print(f"电池电量: {((display_data[0]*256+display_data[1])/1000):.2f}V")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+###############################################################################固件升级
+
+"""下载更新请求"""
+def update_request() -> Optional[bytes]:
+    write_data(UPDATE_REQUEST_H, UPDATE_REQUEST_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""查询最大通讯长度"""
+def read_max_com_len() -> Optional[bytes]:
+    write_data(MAX_COM_LEN_H, MAX_COM_LEN_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""下载文件的信息"""
+def download_massage() -> Optional[bytes]:
+    write_data(DL_MESSAGE_H, DL_MESSAGE_L, file_data)#文件信息来源从哪获取？
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+"""查询设备状态"""
+def read_device_status() -> Optional[bytes]:
+    write_data(READ_STATUS_H, READ_STATUS_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+"""发送页校验码"""
+def write_page_check() -> Optional[bytes]:
+    write_data(PAGE_CHECK_H, PAGE_CHECK_L, page_check_data)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1          
+        
+"""发送页数据"""
+def write_page_check() -> Optional[bytes]:
+    write_data(PAGE_SEND_H, PAGE_SEND_L, page_send_data)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3].decode(errors="ignore")
+            print(f"从机响应: {display_data}")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+     
+###############################################################################读取传感器信息
+
+"""读取外设连接情况"""
+def read_peripheral() -> Optional[bytes]:   
+    write_data(READ_PERIPH_H, READ_PERIPH_L)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+            for x in display_data:
+                print(f"{x:02X}", end=' ')
+            print("\n")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1 
+
+"""单次操作外设"""         
+def single_operate_sensor(op_struct: bytes) -> Optional[bytes]:       
+    write_data(SINGLE_OP_H, SINGLE_OP_L, op_struct)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+            for x in display_data:
+                print(f"{x:02X}", end=' ')
+            print("\n")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1    
+              
+"""从机模式转换"""         
+def mode_change(send_data: str) -> Optional[bytes]:     
+    write_data(MODE_CHANGE_H, MODE_CHANGE_L, send_data)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+            for x in display_data:
+                print(f"{x:02X}", end=' ')
+            print("\n")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+        
+"""智能模式发送周期"""         
+def mode_change(send_data: str) -> Optional[bytes]:       
+    write_data(SEND_CYCLE_H, SEND_CYCLE_L, send_data)
+    start_time = time.time()
+    while True:
+        response =process_received_data()
+        if response:
+            display_data = response[6:-3]
+            for x in display_data:
+                print(f"{x:02X}", end=' ')
+            print("\n")
+            return display_data
+        else:
+            if time.time() - start_time > 3:
+                print("读取超时")
+                buffer.clear() 
+                return -1
+
+            
+         
+            

smartpi/color_sensor.py

@@ -0,0 +1,6 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+

smartpi/humidity.py

@@ -0,0 +1,20 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+             
+#湿度读取 port:连接P端口；
+def get_value(port:bytes) -> Optional[bytes]:
+    humiture_str=[0xA0, 0x0C, 0x01, 0x71, 0x00, 0xBE]
+    humiture_str[0]=0XA0+port
+    humiture_str[2]=0 
+    response = base_driver.single_operate_sensor(humiture_str)
+    if response:
+        return 0
+    else:
+        return -1
+
+
+
+

smartpi/led.py

@@ -0,0 +1,17 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+
+#�ʵƿ��� port:����P�˿ڣ�command:0:�صƣ�1:�죻2:�̣�3:����4:�ƣ�5:�ϣ�6:�ࣻ7:��
+def set_color(port:bytes,command:bytes) -> Optional[bytes]:
+    color_lamp_str=[0xA0, 0x05, 0x00, 0xBE]
+    color_lamp_str[0]=0XA0+port
+    color_lamp_str[2]=command
+    response = base_driver.single_operate_sensor(color_lamp_str)
+    if response:
+        return 0
+    else:
+        return -1
+        

smartpi/light_sensor.py

@@ -0,0 +1,27 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+
+
+#����ȡ port:����P�˿ڣ�command:1:��ȡ��2:���ƣ�3:�صƣ�
+#def get_value(port:bytes,command:bytes) -> Optional[bytes]:
+#    light_str=[0xA0, 0x02, 0x00, 0xBE]
+#    light_str[0]=0XA0+port
+#    light_str[2]=command 
+#    response = base_driver.single_operate_sensor(light_str)
+#    if response:
+#        return 0
+#    else:
+#        return -1
+        
+def get_value(port:bytes) -> Optional[bytes]:
+    light_str=[0xA0, 0x02, 0x00, 0xBE]
+    light_str[0]=0XA0+port
+    light_str[2]=1 
+    response = base_driver.single_operate_sensor(light_str)
+    if response:
+        return 0
+    else:
+        return -1

smartpi/motor.py

@@ -0,0 +1,60 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+
+##############################################################
+#            motor.write_motor_dir(1,0)
+#            motor.write_motor_speed(1,100)
+#            time.sleep(0.5)
+#            motor.write_motor_dir(1,1)
+#            motor.write_motor_speed(1,50)
+#            time.sleep(0.5)
+##############################################################
+
+#��������ȡ port:����M�˿ڣ�
+def read_motor_code(port:bytes) -> Optional[bytes]:
+    motor_str=[0xA0, 0x01, 0x01, 0xBE]           
+    motor_str[0]=0XA0+port       
+    response = base_driver.single_operate_sensor(motor_str)
+    if response:
+        return 0
+    else:
+        return -1
+        
+#�����ٶȿ��� port:����M�˿ڣ�speed:0~100
+def write_motor_speed(port:bytes,speed:bytes) -> Optional[bytes]:
+    motor_str=[0xA0, 0x01, 0x02, 0x71, 0x00, 0xBE]           
+    motor_str[0]=0XA0+port
+    motor_str[4]=speed
+    response = base_driver.single_operate_sensor(motor_str)
+    if response:
+        return 0
+    else:
+        return -1
+        
+#�����ٶȱ������ port:����M�˿ڣ�speed:0~100��code:0~65535
+def motor_servoctl(port:bytes,speed:bytes,code:int) -> Optional[bytes]:
+    motor_str=[0xA0, 0x01, 0x04, 0x81, 0x00, 0x81, 0x00, 0x00, 0xBE]           
+    motor_str[0]=0XA0+port
+    motor_str[4]=speed
+    motor_str[6]=code//256
+    motor_str[7]=code%256
+    response = base_driver.single_operate_sensor(motor_str)
+    if response:
+        return 0
+    else:
+        return -1
+        
+#���﷽����� port:����M�˿ڣ�dir:
+def write_motor_dir(port:bytes,direc:bytes) -> Optional[bytes]:
+    motor_str=[0xA0, 0x01, 0x06, 0x71, 0x00, 0xBE]           
+    motor_str[0]=0XA0+port
+    motor_str[4]=direc
+    response = base_driver.single_operate_sensor(motor_str)
+    if response:
+        return 0
+    else:
+        return -1
+        

smartpi/servo.py

@@ -0,0 +1,25 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+
+###############################################
+#                  servo.servo_operate(1,1,0)                  
+#                  time.sleep(0.5)
+#                  servo.servo_operate(1,1,180)
+#                  time.sleep(0.5)
+###############################################
+
+#������� port:����P�˿ڣ�command:1:����  angle:�Ƕ�
+def servo_operate(port:bytes,command:bytes,angle:bytes) -> Optional[bytes]:
+    servo_str=[0xA0, 0x0E, 0x01, 0x71, 0x00, 0xBE]
+    servo_str[0]=0XA0+port
+    servo_str[2]=command
+    servo_str[4]=angle
+    response = base_driver.single_operate_sensor(servo_str)
+    if response:
+        return 0
+    else:
+        return -1
+        

smartpi/temperature.py

@@ -0,0 +1,18 @@
+# coding=utf-8
+import time
+from typing import List, Optional
+from smartpi import base_driver
+
+
+
+#�¶ȶ�ȡ port:����P�˿ڣ�
+def get_value(port:bytes) -> Optional[bytes]:
+    humiture_str=[0xA0, 0x0C, 0x01, 0x71, 0x00, 0xBE]
+    humiture_str[0]=0XA0+port
+    humiture_str[2]=1 
+    response = base_driver.single_operate_sensor(humiture_str)
+    if response:
+        return 0
+    else:
+        return -1
+