[毕设记录] python利用socket进行文件传输

前言

大四下学期,寒假不努力,现在要拼命赶进度了。毕业设计当中,我需要下位机stm32将图片发送(串口转wifi)到上位机(pc)进行图像处理。以前没有试过这样的操作,思考了一下,因为有串口转wifi的模块,所以决定上位机用python来写一个socket实现这个功能(由于下位机是stm32,所以下位机的发送就只能用c来写了)。

尝试

先用python写一个demo来测试下是否可以实现,写两个python程序。一个python程序充当下位机(客户端),一个pyhon充当上位机(服务端)。

逻辑如下:首先,服务端代码是一直在运行,监听着端口,如果有客户端发起请求,则连接之建立socket。客户端与服务端连接后,开始发送文件。期间,输出发送的状态信息。

我的目录结构是这样子的:

  • 桌面新建一个send_image_test文件夹
  • send_image_test文件夹里面分别建py_s文件夹和py_c文件夹
  • 以上两个文件夹里面分别建py_s.py和py_c.py
  • 在文件夹py_c里面放了一张bmp格式的图片,命名为image.bmp

根据上面的逻辑,写出下面的代码:

# 服务端代码,环境Mac
#!/usr/bin/env python
# -*- coding: UTF-8 -*-

# 导入库
import socket, threading, os

SIZE = 1024

# 检查当前目录下是否有等下要命名的图片,有的话删除之
def checkFile():
    list = os.listdir('.')
    for iterm in list:
        if iterm == 'image.bmp':
            os.remove(iterm)
            print 'remove'
        else:
            pass

# 接受数据线程
def tcplink(sock, addr):
    print 'Accept new connection from %s:%s...' % addr
    sock.send('Welcome from server!')
    print 'receiving, please wait for a second ...'
    while True:
        data = sock.recv(SIZE)
        if not data :
            print 'reach the end of file'
            break
        elif data == 'begin to send':
            print 'create file'
            checkFile()
            with open('./image.bmp', 'wb') as f:
                pass
        else:
            with open('./image.bmp', 'ab') as f:
                f.write(data)
    sock.close()
    print 'receive finished'
    print 'Connection from %s:%s closed.' % addr


# 创建一个socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听端口(这里的ip要在不同的情况下更改)
s.bind(('127.0.0.1', 9999))
# 每次只允许一个客户端接入
s.listen(1)
print 'Waiting for connection...'
while True:
    sock, addr = s.accept()
    # 建立一个线程用来监听收到的数据
    t = threading.Thread(target = tcplink, args = (sock, addr))
    # 线程运行
    t.start()
# 客户端代码,环境Mac
#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import socket

SIZE = 1024

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 建立连接:
s.connect(('127.0.0.1', 9999))
# 接收欢迎消息:
print s.recv(SIZE)
s.send('begin to send')
print 'sending, please wait for a second ...'
with open('./image.bmp', 'rb') as f:
    for data in f:
        s.send(data)
print 'sended !'
s.close()
print 'connection closed'

需要注意的是,以上的代码是在本地测试的。所以地址使用的是127.0.0.1(如需在不同的机器上测试,请使用对应的ip地址),端口号使用一个没有被其他应用程序使用的即可(我随便选了一个9999)。

以上的两个程序当中都有一个SIZE常量,是接收缓冲区的最大字节数,如果每次发送的数据大过SIZE个字节,则肯定会出现错误,这里需要谨慎对待。

发送数据时,我们是从文件当中读取数据到内存,然后再往外发送。那么如果读取的数据很大,超过了内存怎么办。上面的方法就是最好的方法,不要使用file.read(size)这类的方法,容易出错。而是,将整个file当做一个对象来处理,python虚拟机会自己分配缓冲区和内存,参看这里:

读取大文件时的方法(这种方法需要python2.5以上)

另外,写代码时需要注意缩进。要么使用四个空格,要么使用一个tab。不能同时使用两种,否则,需要在你的编辑器上做空格和tab的转换处理。

提供一下python 2.7的socket库相关文档链接:

官方文档(英文)
翻译的文档(中文)

通过运行上面的代码,图片成功的传输了到服务端的文件夹下:


测试结果

进阶修改

上述的代码只是直接将一张图片,从客户端发送到了服务端。在实际的使用当中,应该是需要发送命令和文件两种形式。所以,我们需要对上面的代码进行一些修改。

我们先定义一个通信协议:

建立连接后开启一个线程,用于处理本次的通信(一个while循环),如果遇到接收文件的情况,则开启一个子线程用于接收。

发送数据有两种情况,发送命令,发送文件
接收数据有两种情况,一种是接收命令,一种是接收文件

发送数据:
发送命令时,第一个字符为c(command)表示后面接的是命令,第二个字符是命令代号。
发送文件时,第一个字符为f(file)表示后面接的是文件,后面就是文件的二进制形式。
接收数据:
接收数据分成两段,首先查看第一段报文是命令还是文件标识,第二段则是接收具体的数据
命令标识为”c“,文件标识为”f“

修改后的代码粘贴如下:

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

# 导入socket库
import socket, threading, time, os

SIZE = 1024

# 检查当前目录下是否有等下要命名的图片,有的话删除之
def checkExsit():
    list = os.listdir('.')
    for iterm in list:
        if iterm == 'image.bmp':
            os.remove(iterm)
            print 'Exsit file has been removed'
    print 'Create file ...'
    with open('./image.bmp', 'wb') as f: pass

def recvImage(sock):
    while True:
        data = sock.recv(SIZE)
        if not data:
            break
        else:
            with open('./image.bmp', 'ab') as f:
                f.write(data)
    print 'data received'

def saveImage(sock):
    print 'Begin to save image ...'
    checkExsit()
    t = threading.Thread(target = recvImage, args = (sock,))
    t.setDaemon(True)
    t.start()
    t.join()
    print 'Finished saving image ...'

def tcplink(sock, addr):
    # 打印连接信息
    print 'Accept new connection from %s:%s...' % addr
    # 发送问候信息(客户端接收到后返回一个'hello server')
    sock.send('hello client')
    print sock.recv(SIZE)
    print 'Communication test success'
    # 接受数据循环(一直等待接收数据并进行处理 *****注意这是在一个线程里面******)
    while True:
        recv = sock.recv(SIZE)
        # 接收命令
        if recv == 'c':
            print 'receive command'
            cmd = sock.recv(SIZE)
            print 'recv: %s' %cmd
            # 判断命令并执行相应的程序
            recv = None
        # 接收文件(这里主要是图片)
        elif recv == 'f':
            print 'file command'
            saveImage(sock)
            recv = None

# 创建一个socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听端口(这里的ip要在不同的情况下更改)
s.bind(('127.0.0.1', 9999))
# 每次只允许一个客户端接入
s.listen(1)
print 'Waiting for connection...'
while True:
    sock, addr = s.accept()
    # 建立一个线程用来监听收到的数据
    t = threading.Thread(target = tcplink, args = (sock, addr))
    # 线程运行
    t.start()
#!/usr/bin/env python
# -*- coding: UTF-8 -*-

import socket,time

SIZE = 1024

s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 建立连接:
s.connect(('127.0.0.1', 9999))
# 接收欢迎消息:
print s.recv(SIZE)

s.send('hello server')
time.sleep(0.5)
print 'command test begins ...'
s.send('c')
s.send('weeding')
print 'command test ended'
time.sleep(0.5)

print 'image test begins ...'
s.send('f')
time.sleep(0.2)
with open('./image.bmp', 'rb') as f:
    for data in f:
        s.send(data)
print 'image test ended'

s.close()
print 'connection closed'

上面的代码,运用了一个子线程来执行接收图片。使用多线程,最好可以了解下下面的内容:

守护线程
线程锁
线程的中文文档

修改后运行结果
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 156,907评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,546评论 1 289
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 106,705评论 0 238
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,624评论 0 203
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 51,940评论 3 285
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,371评论 1 210
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,672评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,396评论 0 195
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,069评论 1 238
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,350评论 2 242
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,876评论 1 256
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,243评论 2 251
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,847评论 3 231
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,004评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,755评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,378评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,266评论 2 259

推荐阅读更多精彩内容