import serial
from xmodem import XMODEM
import os
 
def Send_File(filepath, port='COM8', baudrate=115200):
    bn = os.path.basename(filepath)
    filesize = os.stat(filepath).st_size
    strSendFileCMD = "AFF_" + str(filesize) + "_" + bn + "\n"
    # 打開串口
    ser = serial.Serial(port, baudrate)
     
    # 定義YMODEM發(fā)送函數(shù)
    def send_ymodem(filename):
        def getc(size, timeout=1):
            return ser.read(size)
        def putc(data, timeout=1):
            return ser.write(data)
        modem = XMODEM(getc, putc)
        with open(filename, "rb") as f:
            status = modem.send(f)
        return status
     
    # 發(fā)送指令
    ser.write(strSendFileCMD.encode())
    # 發(fā)送文件
    status = send_ymodem(filepath)
    if status:
        print(f"文件發(fā)送成功：{filepath}")
    else:
        print(f"文件發(fā)送失敗：{filepath}")
    # 關(guān)閉串口
    ser.close()
 
 
def Recv_File(port='COM2', baudrate=115200):
    # 打開串口
    ser = serial.Serial(port, baudrate)
     
    # 定義YMODEM接收函數(shù)
    def recv_ymodem(filename):
        def getc(size, timeout=1):
            return ser.read(size) or None
        def putc(data, timeout=1):
            return ser.write(data)
        modem = XMODEM(getc, putc)
        with open(filename, "wb") as f:
            status = modem.recv(f)
        return status
     
    # 循環(huán)監(jiān)聽指令
    while True:
        # 接收指令
        print("等待接收指令")
        strCMD = ser.read_until().strip().decode()
        print(strCMD)
        cmdlist = strCMD.split("_")
        cmd = cmdlist[0]
        filesize = cmdlist[1]
        filename = cmdlist[2]
        if cmd == "AFF":
            # 收到指令后開始接收文件
            print("開始接收文件")
            # 接收文件并保存
            status = recv_ymodem(filename)
            if status:
                print(f"文件接收成功：{filename}")
            else:
                print(f"文件接收失敗：{filename}")
            # 繼續(xù)監(jiān)聽指令
            continue
        # 其他指令
        print(f"收到指令：{cmd}")
    # 關(guān)閉串口
    ser.close()
 
if __name__=="__main__":
    Send_File("D:/users.7z") #發(fā)送文件
    #Recv_File() #接收文件