他拍了拍你！来这里看socket “ftp”

关注公众号"栈了一堆"，观看更多精彩文章！

socket小白看这里，'小黑’直接看代码。

socket介绍：

Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。

Python中，我们用socket（）函数来创建套接字，语法格式如下：

socket.socket([family[,type[, proto]]])

今天咱们先了解一下基本使用与方法，之后再看一个示例，好久之前写的一个模拟ftp的put,get的示例，好了，瞅下面吧！

参数说明：

family: 套接字家族可以使AF_UNIX或者AF_INET
type: 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM，
# SOCK_STREAM是基于TCP的，数据传输比较有保障。SOCK_DGRAM是基于UDP的，专门用于局域网，
protocol: 一般不填默认为0.

socket对象方法：

服务器端套接字

• s.bind() 绑定地址（host,port）到套接字， 在AF_INET下,以元组（host,port）的形式表示地址。

• s.listen() 开始TCP监听。backlog指定在拒绝连接之前，操作系统可以挂起的最大连接数量。该值至少为1，大部分应用程序设为5就可以了。

• s.accept() 被动接受TCP客户端连接,(阻塞式)等待连接的到来

客户端套接字

• s.connect() 主动初始化TCP服务器连接，。一般address的格式为元组（hostname,port），如果连接出错，返回socket.error错误。
• s.connect_ex() connect()函数的扩展版本,出错时返回出错码,而不是抛出异常

公共用途的套接字函数

• s.recv() 接收TCP数据，数据以字符串形式返回，bufsize指定要接收的最大数据量。flag提供有关消息的其他信息，通常可以忽略。
• s.send() 发送TCP数据，将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。
• s.sendall() 完整发送TCP数据，完整发送TCP数据。将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。
• s.recvfrom() 接收UDP数据，与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。
• s.sendto() 发送UDP数据，将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。
• s.close() 关闭套接字
• s.getpeername() 返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。
• s.getsockname() 返回套接字自己的地址。通常是一个元组(ipaddr,port)
• s.setsockopt(level,optname,value) 设置给定套接字选项的值。
• s.getsockopt(level,optname[.buflen]) 返回套接字选项的值。
• s.settimeout(timeout) 设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如connect()）
• s.gettimeout() 返回当前超时期的值，单位是秒，如果没有设置超时期，则返回None。
• s.fileno() 返回套接字的文件描述符。
• s.setblocking(flag) 如果flag为0，则将套接字设为非阻塞模式，否则将套接字设为阻塞模式（默认值）。非阻塞模式下，如果调用recv()没有发现任何数据，或send()调用无法立即发送数据，那么将引起
• socket.error异常。
• s.makefile() 创建一个与该套接字相关联的文件

大家看烦了吧，撸码哈！

server.py

from socket import *
import os

Basedir = os.getcwd() + '\server_file\\'
def send_file(csock, filename):
   fp = filename if os.path.isabs(filename) else os.path.join(Basedir, filename)
   if not os.path.exists(fp):
      csock.send("error".encode('utf-8'))
   else:
      f_size = os.path.getsize(fp)
      csock.send(str(f_size).encode('utf-8'))
      with open(fp, 'rb') as f:
         print('start send file size is {}'.format(f_size))
         r_data = bytes()
         while True:
            data = f.read(4096)
            r_data += data
            if len(r_data) == int(f_size):
               csock.send(r_data)
               print('\033[31mSend file ok !\033[0m')
               return
   return
   
def recv_file(csock, filename):
   fp = filename if os.path.isabs(filename) else os.path.join(Basedir, filename)
   csock.send('start_upload'.encode('utf-8'))
   fsize = csock.recv(1024)
   sumsize = fsize.decode('utf-8')
   if len(sumsize) > 0:
      with open(fp, 'wb') as f:
         print('start save file :', fp)
         r_data = bytes()
         while True:
            # print(sumsize)
            ndata = csock.recv(4096)
            r_data += ndata
            # print(r_data)
            if len(r_data) == int(sumsize):
               f.write(r_data)
               print('save file ok')
               break
   return
   
def sls_fun():
   list_data = os.listdir(Basedir)
   return list_data
   
def accept_write(ssock):
   csockfd, addres = ssock.accept()
   print('connect client \033[31m{}\033[0m...'.format(addres))
   while True:
      data = csockfd.recv(1024)
      ndata = data.decode('utf-8')
      ndata = ndata.split()
      if ndata[0] == 'close':
         break
      elif ndata[0] == 'sls':
         list2 = sls_fun()
         csockfd.send(str(list2).encode('utf-8'))
      elif ndata[0] == 'put':
         recv_file(csockfd, ndata[1])
      elif ndata[0] == 'get':
         send_file(csockfd, ndata[1])
      else:
         break
         
if __name__ == '__main__':
   sock = socket(AF_INET, SOCK_STREAM)
   sock.bind(('192.168.1.2', 8008))
   sock.listen(5)
   print('waiting connect ...')
   accept_write(sock)

client.py

from socket import *
import os, time

Basedir = os.getcwd() + '\client_file\\'

def put_file(csock, filename, ncmd):
   if not os.path.isabs(filename):
      file_path = os.path.join(Basedir, filename)
   else:
      file_path = filename
   print(file_path)
   if not os.path.exists(file_path):
      print('本地没有该文件！')
      return
   else:
      csock.send(ncmd.encode('utf-8'))
      result = csock.recv(1024)
      if result.decode('utf-8') == 'start_upload':
         f_size = os.path.getsize(file_path)
         csock.send(str(f_size).encode('utf-8'))
         with open(file_path, 'rb') as f:
            print('start put file...')
            r_data = bytes()
            while True:
               data = f.read(4096)
               r_data += data
               if len(r_data) == int(f_size):
                  csock.send(r_data)
                  print('put file ok {}'.format(len(r_data)))
                  break
   return
   
def get_file(csock, filename, ncmd):
   csock.send(ncmd.encode('utf-8'))
   data = csock.recv(1024)
   if data.decode('utf-8') == 'error':
      print('file not found!')
   elif int(data.decode('utf-8')) > 0:
      sumsize = data.decode('utf-8')
      fp = os.path.join(Basedir, filename)
      print('start get file')
      with open(fp, 'wb') as f:
         rdata = bytes()
         while True:
            ndata = csock.recv(4096)
            rdata += ndata
            if len(rdata) == int(sumsize):
               f.write(rdata)
               print('\033[31mGet file ok !\033[0m')
               break
      return
      
def ls_fun():
   ls_data = os.listdir(Basedir)
   return ls_data
   
def conn_s(csock):
   csock.connect(('192.168.1.2', 8008))
    print('''connect server ...
    Please input command:
    ls : 列出客户端文件。
    sls: 列出服务端文件。
    put fn:上传‘fn’文件。
    get fn:下载‘fn’文件。''')
   while True:
      ncmd = input('>>>')
      cmd = ncmd.split()
      if not cmd or cmd[0] == 'ls':
         list1 = ls_fun()
         print(list1)
      elif cmd[0] == 'sls':
         csock.send(cmd[0].encode('utf-8'))
         sdata = csock.recv(4096)
         print(sdata.decode('utf-8'))
      elif cmd[0] == 'put' and cmd[1]:
         put_file(csock, str(cmd[1]), ncmd)
      elif cmd[0] == 'get' and cmd[1]:
         get_file(csock, str(cmd[1]), ncmd)
      elif cmd[0] == 'del' and cmd[1]:
         csock.send(cmd[0].encode('utf-8'))
      elif cmd[0] == 'sdel' and cmd[1]:
         csock.send(cmd[0].encode('utf-8'))
      elif cmd[0] == 'close':
         csock.send(cmd[0].encode('utf-8'))
         break
      else:
         pass
         
if __name__ == '__main__':
   csock = socket(AF_INET, SOCK_STREAM)
   conn_s(csock)

讲一下大概功能哈，先运行server.py，后运行client.py，在client端可以查看当地文件夹内的文件，和server端的文件夹内的文件，可以put与get两端的文件，这是最初学习socket时弄的，大家有问题请关注公众号’栈了一堆’再留言，希望和大家一起学习成长，你们的时间都是很宝贵的，这里就不留大家吃饭了哈。