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 相关文章推荐
通过C++学习Python
Jan 20 Python
python使用wxPython打开并播放wav文件的方法
Apr 24 Python
Python在线运行代码助手
Jul 15 Python
Python中异常重试的解决方案详解
May 05 Python
Python爬虫实现网页信息抓取功能示例【URL与正则模块】
May 18 Python
Python用imghdr模块识别图片格式实例解析
Jan 11 Python
Python生成任意范围任意精度的随机数方法
Apr 09 Python
Python实现的多进程和多线程功能示例
May 29 Python
python 提取tuple类型值中json格式的key值方法
Dec 31 Python
Python list列表中删除多个重复元素操作示例
Feb 27 Python
python基于json文件实现的gearman任务自动重启代码实例
Aug 13 Python
详解Python中import机制
Sep 11 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
阿拉伯的咖啡与水烟
2021/03/03 咖啡文化
PHP下用rmdir实现删除目录的三种方法小结
2008/04/20 PHP
PHP set_time_limit(0)长连接的实现分析
2010/03/02 PHP
PHP命名空间(Namespace)简明教程
2014/06/11 PHP
thinkPHP使用pclzip打包备份mysql数据库的方法
2016/04/30 PHP
php版微信开发之接收消息,自动判断及回复相应消息的方法
2016/09/23 PHP
PHP实现防盗链的方法分析
2017/07/25 PHP
php弹出提示框的是实例写法
2019/09/26 PHP
判断页面是关闭还是刷新的js代码
2007/01/28 Javascript
关于extjs4如何获取grid修改后的数据的问题
2013/08/07 Javascript
createTextRange()的使用示例含文本框选中部分文字内容
2014/02/24 Javascript
鼠标拖拽移动子窗体的JS实现
2014/02/25 Javascript
js实现简单的可切换选项卡效果
2015/04/10 Javascript
JavaScript中的toLocaleLowerCase()方法使用详解
2015/06/06 Javascript
javascript点击按钮实现隐藏显示切换效果
2016/02/03 Javascript
微信小程序开发之录音机 音频播放 动画实例 (真机可用)
2016/12/08 Javascript
jQuery实现select模糊查询(反射机制)
2017/01/14 Javascript
vue实现商城上货组件简易版
2017/11/27 Javascript
详解Angular路由之路由守卫
2018/05/10 Javascript
使用vue-router在Vue页面之间传递数据的方法
2019/07/15 Javascript
React-redux实现小案例(todolist)的过程
2019/09/29 Javascript
JavaScript中数组去重的5种方法
2020/07/04 Javascript
python3利用Dlib19.7实现人脸68个特征点标定
2018/02/26 Python
解决python3运行selenium下HTMLTestRunner报错的问题
2018/12/27 Python
解决yum对python依赖版本问题
2019/07/05 Python
python urllib爬虫模块使用解析
2019/09/05 Python
使用python制作游戏下载进度条的代码(程序说明见注释)
2019/10/24 Python
python numpy数组中的复制知识解析
2020/02/03 Python
如何将tensorflow训练好的模型移植到Android (MNIST手写数字识别)
2020/04/22 Python
python使用列表的最佳方案
2020/08/12 Python
HTML5进度条特效
2014/12/18 HTML / CSS
公司营业员的自我评价
2014/03/04 职场文书
事业单位工作人员2015年度思想工作总结
2015/10/15 职场文书
励志语录:时光飞逝,请学会珍惜所有的人和事
2020/01/16 职场文书
Python实现DBSCAN聚类算法并样例测试
2021/06/22 Python
Mysql数据库按时间点恢复实战记录
2021/06/30 MySQL