Python socket服务常用操作代码实例


Posted in Python onJune 22, 2020

套接字(socket)是一个抽象层,应用程序可以通过它发送或接收数据,可对其进行像对文件一样的打开、读写和关闭等操作。

1. 实现客户端发送字符,服务器返回大写的字符:

服务器:

import socketserver

class MyTCPHandler(socketserver.BaseRequestHandler):     # 通过类的继承,实现
  def handle(self):          # 重写父类的handle方法,所有的操作都在此方法中
    while True:           # 循环,不停的接收从客户端来的数据
      try:
        self.data = self.request.recv(1024).strip()     # 从客户端接收数据,每次收1024字节
        print("{} send:".format(self.client_address), self.data)
        self.request.send(self.data.upper())         # 从服务器发送给客户端数据
      except ConnectionResetError as e:
        print('Error: ',e)
        break

if __name__ == '__main__':
  host,port = 'localhost',9999
  server = socketserver.ThreadingTCPServer((host,port),MyTCPHandler)   # 通过多线程实现多个客户端连接,每个客户端连接都是一个线程
  server.serve_forever()                        # 一直运行服务

客户端:

import socket

client = socket.socket()         # socket对象
client.connect(('localhost',9999))   # 连接服务器地址和端口

while True:               # 循环,不停的输入发送数据
  con = input('>>>:').strip()
  if len(con) ==0: continue      # 不能发送空数据,否则会阻塞
  client.send(con.encode('utf-8'))  # 发送数据,必须是二进制的
  data = client.recv(1024)      # 接收服务器返回的数据
  print(data.decode())        # 打印 解码后的数据

client.close()           # 关闭

2. 通过socket执行服务器命令:

用法:直接在客户端输入处输入命令如:ipconfig

服务器:

import socket
import os
import threading

def tcplink(sock, addr):
  print('Accept new connection from %s:%s...' % addr)
  while True:     # 和每个接入的客户端,进行多次数据通信
    data = sock.recv(1024) # 接收客户端数据
    if not data or data.decode('utf-8') == 'exit': # 如果客户端不发送数据或者发送了exit
      print('client disconnected.')
      break
    content = os.popen(data.decode('utf-8')).read() # 对发送来的数据执行cmd命令,获取结果
    if len(content) == 0:      #如果执行的命令结果为空的,就手动造一个结果。因为如果为空数据,会挂起,无法正常发送。
      content = 'cmd not exists.'
    sock.send(str(len(content.encode('utf-8'))).encode('utf-8')) # 发送数据的长度
    print('send length:', (len(content.encode('utf-8'))))
    # print('content,', content.encode('utf-8'))
    recv = sock.recv(1024) # 因为上下都有一个send连在一起,可能发生粘包现象,为了防止这种情况,可以让客户端重新应答一下
    print('Answer:',recv.decode('utf-8'))
    sock.send(content.encode('utf-8'))    # 发送数据
    print('send finished.')
  sock.close()
  print('Connection from %s:%s closed.' % addr)


s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听端口:
s.bind(('127.0.0.1', 9999))
s.listen(3)
print('Waiting for connection...')

while True:
  # 接受一个新连接:
  sock, addr = s.accept()
  # 创建新线程来处理TCP连接:
  t = threading.Thread(target=tcplink, args=(sock, addr))
  t.start()

客户端:

import socket

# AF_INET 代表ipv4,SOCK_STREAM 代表TCP
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)   # 确定网络协议,生成对象
s.connect(('127.0.0.1',9999)) # 连接服务器的地址和端口,元组的形式。
while True:
  msg = input('>>:').strip()
  if len(msg) != 0:         # 如果消息为空,会一直挂起,所以不能为空
    if msg =='exit':
      s.close()           # 关闭连接
      print('Connection closed.')
      break
    s.send(msg.encode('utf-8'))    # 给服务器发送数据,必须是二进制的
    length = s.recv(1024)      # 首先接收服务器返回的将要接收的数据的长度信息。
    s.send(b'Ready to receive...')  # 发送接收命令
    length = int(length.decode('utf-8'))
    print('receive len:', length)
    data_len = 0
    data_recv = b''
    while data_len < length:  # 已经接收的信息的长度,如果小于总长度
      data = s.recv(1024)   # 从服务器接收数据
      data_recv += data
      data_len += len(data)
    print(data_recv.decode('utf-8')) # 打印返回的数据。

3. 通过socket传输文件:

用法:get 文件名

服务器:

import socket
import os
import hashlib
import threading

def tcplink(sock, addr):
  print('Accept new connection from %s:%s...' % addr)
  while True:     # 和每个接入的客户端,进行多次数据通信
    data = sock.recv(1024) # 接收客户端数据
    if not data or data.decode('utf-8') == 'exit': # 如果客户端不发送数据或者发送了exit
      print('client disconnected.')
      break
    oper,filename = data.decode('utf-8').split() # 对接收的数据按照空格分割
    if oper == 'get':
      m = hashlib.md5()
      if os.path.isfile(filename):
        size = os.stat(filename).st_size  # 获取文件大小
        print('Send size:',size)
        sock.send(str(size).encode('utf-8')) # 发送文件大小
        recv = sock.recv(1024)       # 接收客户端确认信息(因为上下文两个send是连着的,所以为了防止粘包,接收一次信息)
        f = open(filename,'rb')
        for line in f:
          sock.send(line)  #读取文件,发送给客户端
          m.update(line)
        # print('Send finished.',m.hexdigest())  # 打印md5的值
        sock.send(m.hexdigest().encode('utf-8')) # 把md5的值发送给客户端
  sock.close()
  print('Connection from %s:%s closed.' % addr)
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
# 监听端口:
s.bind(('127.0.0.1', 9999))
s.listen(3)
print('Waiting for connection...')

while True:
  # 接受一个新连接:
  sock, addr = s.accept()
  # 创建新线程来处理TCP连接:
  t = threading.Thread(target=tcplink, args=(sock, addr))
  t.start()

客户端:

import socket
import hashlib

# AF_INET 代表ipv4,SOCK_STREAM 代表TCP
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)   # 确定网络协议,生成对象
s.connect(('127.0.0.1',9999)) # 连接服务器的地址和端口,元组的形式。
while True:
  msg = input('>>:').strip()
  if len(msg) != 0:         # 如果消息为空,会一直挂起,所以不能为空
    if msg =='exit':
      s.close()           # 关闭连接
      print('Connection closed.')
      break
    s.send(msg.encode('utf-8'))    # 给服务器发送数据,必须是二进制的
    length = s.recv(1024)      # 首先接收服务器返回的将要接收的数据的长度信息。
    s.send(b'Ready to receive...')  # 发送接收确认命令
    length = int(length.decode('utf-8'))
    print('Recv size:', length)
    data_len = 0
    data_recv = b''
    # 新文件名
    fileName = msg.split()[-1].split('.')[0]
    fileExt = msg.split()[-1].split('.')[-1]
    newFile = fileName+'-1.'+fileExt
    f = open(newFile,'wb') # 打开文件,准备写入服务器发过来的文件
    m = hashlib.md5()
    while data_len < length:  # 已经接收的信息的长度,如果小于总长度
      size = length - data_len 
      if size > 1024:   # 如果剩下的信息长度大于1024,即不能一次性发完。
        size = 1024
      else:    # 如果能一次性发完,就只收剩下的信息。目的是准确的接收文件的大小,把可能粘连的send的数据留给下一次recv
        size = length-data_len
      data = s.recv(size)   # 从服务器接收数据
      f.write(data)
      m.update(data)
      data_len += len(data)
    f.close()
    print('recv_md5:',m.hexdigest()) # 打印返回的数据。
    recv = s.recv(1024)   # 接收下一次send的数据,即md5的值。
    print('orig_md5:',recv.decode())

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python使用pil生成图片验证码的方法
May 08 Python
基于Django框架利用Ajax实现点赞功能实例代码
Aug 19 Python
python3中替换python2中cmp函数的实现
Aug 20 Python
Python如何应用cx_Oracle获取oracle中的clob字段问题
Aug 27 Python
Python队列、进程间通信、线程案例
Oct 25 Python
python加载自定义词典实例
Dec 06 Python
Python hashlib模块实例使用详解
Dec 24 Python
Python 安装 virturalenv 虚拟环境的教程详解
Feb 21 Python
浅谈keras2 predict和fit_generator的坑
Jun 17 Python
python爬取youtube视频的示例代码
Mar 03 Python
Python数据清洗工具之Numpy的基本操作
Apr 22 Python
Python爬虫:从m3u8文件里提取小视频的正确操作
May 14 Python
Python如何实现后端自定义认证并实现多条件登陆
Jun 22 #Python
零基础小白多久能学会python
Jun 22 #Python
Keras-多输入多输出实例(多任务)
Jun 22 #Python
python和c语言哪个更适合初学者
Jun 22 #Python
Virtualenv 搭建 Py项目运行环境的教程详解
Jun 22 #Python
终于搞懂了Keras中multiloss的对应关系介绍
Jun 22 #Python
keras 多任务多loss实例
Jun 22 #Python
You might like
动态网站web开发 PHP、ASP还是ASP.NET
2006/10/09 PHP
PHP使用Alexa API获取网站的Alexa排名例子
2014/06/12 PHP
php在linux下检测mysql同步状态的方法
2015/01/15 PHP
学习PHP的数组总结【经验】
2016/05/05 PHP
PHP Socket网络操作类定义与用法示例
2017/08/30 PHP
php判断/计算闰年的方法小结【三种方法】
2019/07/06 PHP
php上传后台无法收到数据解决方法
2019/10/28 PHP
仅IE支持clearAttributes/mergeAttributes方法使用介绍
2012/05/04 Javascript
讨论html与javascript在浏览器中的加载顺序问题
2013/11/27 Javascript
node.js入门教程
2014/06/01 Javascript
jQuery不使用插件及swf实现无刷新文件上传
2014/12/08 Javascript
node.js中的fs.open方法使用说明
2014/12/17 Javascript
jQuery插件cxSelect多级联动下拉菜单实例解析
2016/06/24 Javascript
js替换字符串中所有指定的字符(实现代码)
2016/08/17 Javascript
js 下拉菜单点击旁边收起实现(踩坑记)
2019/09/29 Javascript
ElementUI中el-tree节点的操作的实现
2020/02/27 Javascript
js this 绑定机制深入详解
2020/04/30 Javascript
Python Nose框架编写测试用例方法
2017/10/26 Python
配置 Pycharm 默认 Test runner 的图文教程
2018/11/30 Python
对python读写文件去重、RE、set的使用详解
2018/12/11 Python
Python数据结构与算法(几种排序)小结
2019/06/22 Python
python调用自定义函数的实例操作
2019/06/26 Python
Python函数装饰器原理与用法详解
2019/08/16 Python
pytorch常见的Tensor类型详解
2020/01/15 Python
python中常见错误及解决方法
2020/06/21 Python
CSS3使用transition实现的鼠标悬停淡入淡出
2015/01/09 HTML / CSS
英国Boots旗下太阳镜网站:Boots Designer Sunglasses
2018/07/07 全球购物
eDreams意大利:南欧领先的在线旅行社
2018/11/23 全球购物
计算机专业毕业生的自我评价
2013/11/18 职场文书
机关工会开展学习雷锋活动总结
2014/03/01 职场文书
电子商务求职信
2014/06/15 职场文书
婚宴新郎致辞
2015/07/28 职场文书
2016猴年春节慰问信
2015/11/30 职场文书
经销商会议开幕词
2016/03/04 职场文书
Python爬虫实战之爬取京东商品数据并实实现数据可视化
2021/06/07 Python
多线程Spring通过@Scheduled实现定时任务
2022/05/25 Java/Android