本文介绍了Python中Socket的基本使用，包括socket()函数创建套接字，并探讨了sendto函数的返回值，以及Socket相关的超时和错误处理。通过示例代码展示了如何实现简单的FTP模拟操作。
Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。
Python中，我们用socket()函数来创建套接字，语法格式如下：
socket.socket([family[,type[, proto]]])今天咱们先了解一下基本使用与方法，之后再看一个示例，好久之前写的一个模拟ftp的put,get的示例，好了，瞅下面吧！
参数说明：
family: 套接字家族可以使AF_UNIX或者AF_INETtype: 套接字类型可以根据是面向连接的还是非连接分为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 osBasedir = 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, timeBasedir = 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)
扫一扫
846
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
