Python修炼之路-Socket
网络编程
socket套接字
socket通常也称作"套接字"，用于描述IP地址和端口，是一个通信链的句柄，应用程序通常通过“套接字”向网络发出请求或者应答网络请求。
socket模块是针对 服务器端 和 客户端Socket 进行【打开】【读写】【关闭】
TCP协议Socket
tcp是基于链接的，必须先启动服务端，然后再启动客户端去连接服务端。
server端
import socket s = socket.socket()
s.bind(('127.0.0.1',8888)) # 把地址绑定到套接字
s.listen() # 监听链接
conn,addr = s.accept() # 接受客户端链接, conn是客户端连接过来而在服务器端为其生成的一个连接实例
print(addr)
print(conn)
data = conn.recv(1024) # 接收客户端信息
print(data) # 打印客户端信息
conn.send(b'hi') # 向客户端发送信息
conn.close() # 关闭客户端套接字
s.close() # 关闭服务器套接字(可选)
client端
import socket s = socket.socket() # 创建客户套接字
s.connect(('127.0.0.1',8888)) # 尝试连接服务器
s.send(b'hello!')
res = s.recv(1024) # 对话(发送/接收)
print(res)
s.close() # 关闭客户套接字
UDP协议Socket
udp是无链接的，启动服务之后可以直接接受消息，不需要提前建立链接。
server
import socket
udp_s = socket.socket(type=socket.SOCK_DGRAM) # 创建一个服务器的套接字
udp_s.bind(('127.0.0.1',9000)) # 绑定服务器套接字
msg,addr = udp_s.recvfrom(1024)
print(msg)
udp_s.sendto(b'hi',addr) # 对话(接收与发送)
udp_s.close() # 关闭服务器套接字
client
import socket
ip_port = ('127.0.0.1',9000)
udp_c=socket.socket(type=socket.SOCK_DGRAM)
udp_c.sendto(b'hello',ip_port)
back_msg,addr=udp_c.recvfrom(1024)
print(back_msg.decode('utf-8'),addr)
socket参数
创建一个socket
socket.socket( family=AF_INET, type=SOCK_STREAM, proto=0, fileno=None)
socket参数解析
| 参数 | 解析 |
| family | 地址簇：socket.AF_INET : IPv4（默认） socket.AF_INET6: IPv6 socket.AF_UNIX: 只能够用于单一的Unix系统进程间通信，使用本地 socket 文件来通信 |
| type | socket.SOCK_STREAM: 默认类型，数据流，TCP协议，有保障的（即能保证数据正确传送到对方）面向连接的SOCKET，多用于资料传送。 socket.SOCK_DGRAM: 数据报式，UDP协议，无保障的面向消息的socket，多用于在网络上发广播信息。 |
| proto | 协议号，通常为零,可以省略,与特定的地址家族相关的协议,如果是 0，则系统就会根据地址格式和套接类别,自动选择一个合适的协议；在地址族为AF_CAN的情况下,协议应为CAN_RAW或CAN_BCM之一。 |
| fileno | 如果指定了fileno,则其他参数将被忽略,导致带有指定文件描述符的套接字返回。与socket.fromfd()不同,fileno将返回相同的套接字,而不是重复的。这可能有助于使用socket.close()关闭一个独立的插座。 |
socket常用方法
| 方法 | 说明 |
| s.bind(address) | 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下，以元组（host,port）的形式表示地址 |
| s.listen(backlog) | 开始监听传入连接。backlog指定在拒绝连接之前，可以挂起的最大连接数量。 backlog等于5，表示内核已经接到了连接请求，但服务器还没有调用accept进行处理的连接个数最大为5。 这个值不能无限大，因为要在内核中维护连接队列。 |
| s.setblocking(bool) | 是否阻塞，默认True，如果设置False，那么accept和recv时一旦无数据，则报错。 |
| s.accept() | 接受连接并返回（conn,address）,其中conn是新的套接字对象，可以用来接收和发送数据。address是连接客户端的地址。 接收TCP 客户的连接（阻塞式）等待连接的到来。 |
| s.connect(address) | 连接到address处的套接字。一般，address的格式为元组（hostname,port）,如果连接出错，返回socket.error错误。 |
| s.connect_ex(address) | 同上，只不过会有返回值，连接成功时返回 0 ，连接失败时候返回编码，例如：10061。 |
| s.close() | 关闭套接字。 |
| s.recv(bufsize[,flag]) | 接受套接字的数据。数据以字符串形式返回，bufsize指定最多可以接收的数量。flag提供有关消息的其他信息，通常可以忽略。 |
| s.recvfrom(bufsize[.flag]) | 与recv()类似，但返回值是（data,address）。其中data是包含接收数据的字符串，address是发送数据的套接字地址。 |
| s.send(string[,flag]) | 将string中的数据发送到连接的套接字。返回值是要发送的字节数量，该数量可能小于string的字节大小。即：可能未将指定内容全部发送。 |
| s.sendall(string[,flag]) | 将string中的数据发送到连接的套接字，但在返回之前会尝试发送所有数据。成功返回None，失败则抛出异常。 内部通过递归调用send，将所有内容发送出去。 |
| s.sendto(string[,flag],address) | 将数据发送到套接字，address是形式为（ipaddr，port）的元组，指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。 |
| s.settimeout(timeout) | 设置套接字操作的超时期，timeout是一个浮点数，单位是秒。值为None表示没有超时期。一般，超时期应该在刚创建套接字时设置，因为它们可能用于连接的操作（如 client 连接最多等待5s ） |
| s.getpeername() | 返回连接套接字的远程地址。返回值通常是元组（ipaddr,port）。 |
| s.getsockname() | 返回套接字自己的地址。通常是一个元组(ipaddr,port) |
| s.fileno() | 套接字的文件描述符 |
黏包现象
黏包：同时执行多条命令之后，得到的结果很可能只有一部分，在执行其他命令的时候又接收到之前执行的另外一部分结果，这种显现就是黏包。
TCP协议会出现黏包问题，UDP协议不会出现黏包。
server
from socket import *
import subprocess ip_port=('127.0.0.1',8888)
BUFSIZE=1024 s = socket(AF_INET,SOCK_STREAM)
s.setsockopt(SOL_SOCKET,SO_REUSEADDR,1)
s.bind(ip_port)
s.listen(5) while True:
conn,addr = s.accept()
print('客户端',addr)
while True:
cmd = conn.recv(BUFSIZE)
if len(cmd) == 0:break
res = subprocess.Popen(cmd.decode('utf-8'),shell=True,
stdout=subprocess.PIPE,
stdin=subprocess.PIPE,
stderr=subprocess.PIPE)
stderr = res.stderr.read()
stdout = res.stdout.read()
conn.send(stderr)
conn.send(stdout)
client
import socket
BUFSIZE=1024
ip_port=('127.0.0.1',8888) c = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
res = c.connect_ex(ip_port)
while True:
msg=input('>>: ').strip()
if len(msg) == 0:continue
if msg == 'quit':break
c.send(msg.encode('utf-8'))
act_res=c.recv(BUFSIZE)
print(act_res.decode('utf-8'),end='')
黏包产生
发送方的缓存机制
发送端需要等缓冲区满才发送出去，造成粘包（发送数据时间间隔很短，数据了很小，会合到一起，产生粘包）
# server
from socket import *
ip_port=('127.0.0.1',8080) s = socket(AF_INET,SOCK_STREAM)
s.bind(ip_port)
s.listen(5) conn,addr = s.accept() data1=conn.recv(10)
data2=conn.recv(10)
print('----->',data1.decode('utf-8'))
print('----->',data2.decode('utf-8'))
conn.close() # client
import socket
BUFSIZE=1024
ip_port=('127.0.0.1',8080) s=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
res=s.connect_ex(ip_port) s.send('hello'.encode('utf-8'))
s.send('egg'.encode('utf-8'))
接收方的缓存机制
接收方不及时接收缓冲区的包，造成多个包接收（客户端发送了一段数据，服务端只收了一小部分，服务端下次再收的时候还是从缓冲区拿上次遗留的数据，产生粘包）
# 服务端
from socket import *
ip_port=('127.0.0.1',8080) s = socket(AF_INET,SOCK_STREAM)
s.bind(ip_port)
s.listen(5) conn,addr = s.accept() data1=conn.recv(2) # 一次没有收完整
data2=conn.recv(10) # 下次收的时候,会先取旧的数据,然后取新的 print('----->',data1.decode('utf-8'))
print('----->',data2.decode('utf-8'))
conn.close() # 客户端
import socket
BUFSIZE=1024
ip_port=('127.0.0.1',8080)
c = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
res = c.connect_ex(ip_port)
c.send('hello egg'.encode('utf-8'))
注：
黏包现象只发生在tcp协议中：
- 从表面上看，黏包问题主要是因为发送方和接收方的缓存机制、tcp协议面向流通信的特点。
- 实际上，主要还是因为接收方不知道消息之间的界限，不知道一次性提取多少字节的数据所造成的
黏包解决方案
在发送数据前，把将要发送的字节流总大小给接收端，接收端通过循环多次接受并判断是否接收完所有数据。
注意
接收端判断总接收字节大小时（recv_size < length），会出现接收到的总字节数大于发送端发送过来的总字节数。这是因为，len(一个汉字)=1，len('国'.encode())=3；如果客户端计算字节数不是统计encode()之后的数目，而客户端统计encode后的数目，那么客户端统计的总字节数大于发送端发送过来的总字节数。这也是为什么判断recv_size < length退出循环，而不是判断两者相等退出循环。
# server
import socket,subprocess
ip_port=('127.0.0.1',8080)
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
s.bind(ip_port)
s.listen(5)
while True:
conn,addr=s.accept()while True:
msg = conn.recv(1024)
if not msg:break
res=subprocess.Popen(msg.decode('utf-8'),shell=True,stdin=subprocess.PIPE,stderr=subprocess.PIPE,stdout=subprocess.PIPE)
err = res.stderr.read()
if err:
ret = err
else:
ret = res.stdout.read()
data_length = len(ret)
conn.send(str(data_length).encode('utf-8'))
data = conn.recv(1024).decode('utf-8')
if data == 'recv_ready':
conn.sendall(ret)
conn.close() # client
import socket,time
c = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
res = c.connect_ex(('127.0.0.1',8080))
while True:
msg=input('>>: ').strip()
if len(msg) == 0:continue
if msg == 'quit':break
c.send(msg.encode('utf-8'))
length = int(s.recv(1024).decode('utf-8'))
s.send('recv_ready'.encode('utf-8'))
recv_size=0
data = b''
while recv_size < length:
data+=s.recv(1024)
recv_size+=len(data)
print(data.decode('utf-8'))
端口占用问题
Windows
# 加入一条socket配置，重用ip和端口
import socket
from socket import SOL_SOCKET,SO_REUSEADDR
sk = socket.socket()
sk.setsockopt(SOL_SOCKET,SO_REUSEADDR,1) # 就是它，在bind前加
sk.bind(('127.0.0.1',8898)) # 把地址绑定到套接字
sk.listen() # 监听链接
conn,addr = sk.accept() # 接受客户端链接
ret = conn.recv(1024) # 接收客户端信息
print(ret) # 打印客户端信息
conn.send(b'hi') # 向客户端发送信息
conn.close() # 关闭客户端套接字
sk.close() # 关闭服务器套接字(可选)
Linux
Unix系统可以找到占用端口的进程，kill掉进程
socketserver
四种基本server类型
- class
socketserver.TCPServer(server_address, RequestHandlerClass, bind_and_activate=True) - class
socketserver.UDPServer(server_address, RequestHandlerClass, bind_and_activate=True) - class
socketserver.UnixStreamServer(server_address, RequestHandlerClass, bind_and_activate=True) - class
socketserver.UnixDatagramServer(server_address, RequestHandlerClass,bind_and_activate=True)
ThreadingTCPServer使用：
- 创建一个继承socketserver.BaseRequestHandler的类：class MyTCPHandler(socketserver.BaseRequestHandler)
- 重写handler方法： def handle(self):
- 实例化ThreadingTCPServer： server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
- 启动server： server.serve_forever()
server
import socketserver class Myserver(socketserver.BaseRequestHandler):
def handle(self):
self.data = self.request.recv(1024).strip()
print("{} wrote:".format(self.client_address[0]))
print(self.data)
self.request.sendall(self.data.upper()) if __name__ == "__main__":
HOST, PORT = "127.0.0.1", 9999 # 设置allow_reuse_address允许服务器重用地址
socketserver.TCPServer.allow_reuse_address = True
# 创建一个server, 将服务地址绑定到127.0.0.1:9999
server = socketserver.TCPServer((HOST, PORT),Myserver)
# 让server永远运行下去，除非强制停止程序
server.serve_forever()
client
import socket HOST, PORT = "127.0.0.1", 9999
data = "hello" # 创建一个socket链接，SOCK_STREAM代表使用TCP协议
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
sock.connect((HOST, PORT)) # 链接到客户端
sock.sendall(bytes(data + "\n", "utf-8")) # 向服务端发送数据
received = str(sock.recv(1024), "utf-8")# 从服务端接收数据 print("Sent: {}".format(data))
print("Received: {}".format(received))
实例
import os
import json
import hashlib
import socketserver
from conf import setting
from core import sql class MyTCPHandler(socketserver.BaseRequestHandler): def handle(self):
while True:
try:
self.server_command = self.request.recv(1024).strip()
print('receive command: ',self.server_command.decode() )
self.server_command = json.loads(self.server_command .decode())
if hasattr(self,self.server_command['command_key']):
getattr(self,self.server_command['command_key'])()
except ConnectionResetError as e:
print('\033[31;1mConnection[%s] failed...\033[0m' % (
str(self.client_address[0])))
#print('\033[31;1mConnection[%s:%s] failed...\033[0m' % (str(self.client_address[0]), str(self.client_address[1])))
break
except Exception as e:
print('\033[31;1mConnection[%s] failed...\033[0m' % (
str(self.client_address[0])))
break def register(self):
"""
function: register new account
:return:
"""
account_path = '%s/%s' % (setting.DATABASE['db'], self.server_command['command_param'])
if not os.path.isfile(account_path): # judge this account if exist
self.request.send('False'.encode('utf-8'))
password = self.request.recv(1024).decode() # receive register password
table_name = '%s.%s' % (setting.DATABASE['db'], self.server_command['command_param'])
account_home = '%s/%s' % (setting.DATABASE['home'], self.server_command['command_param'])
os.makedirs(account_home) # make dir for new user
user_data = {
'account': self.server_command['command_param'], # 用户账号
'password': password, # 密码
'space_size_total': 102400000, # 默认分配空间大小
'space_size_used': 0, # 已经使用空间
'home_dir': account_home # 家目录
}
sql.Sql('insert into %s value %s' % (table_name, user_data)) # save account info
self.request.send("create account successful".encode('utf-8'))
else:
self.request.send('True'.encode('utf-8')) def login(self, *args):
"""
function: login ftp server
:return:
"""
self.table_name = '%s.%s' % (setting.DATABASE['db'], self.server_command['command_param'])
self.read_execute = sql.Sql('select * from %s' % self.table_name)
if self.read_execute.status:
# print('user info: ', self.read_execute.status[0], type(self.read_execute.status[0]))
self.request.send(self.read_execute.status[0].encode('utf-8'))
else:
self.request.send('None'.encode('utf-8')) def ls(self):
"""
列出当前目录或所查找目录的下一级内容
:return:
""" dir_dic = {}
if os.path.isdir(self.server_command['command_param']):
chitem = os.listdir(self.server_command['command_param'])
for path in chitem:
chpath = '%s/%s' % (self.server_command['command_param'], path)
if os.path.isdir(chpath):
dir_dic[path] = 'dir'
else:
dir_dic[path] = 'file'
self.request.send(json.dumps(dir_dic).encode('utf-8')) def upload(self):
"""
上传文件
:return:
""" filepath = self.server_command['command_filelocation'] # 文件上传路径
filename = os.path.basename(self.server_command['command_param']) # 文件名
file_size = self.server_command['command_filesize'] # 文件大小
file_info = '%s/%s' %(filepath, filename) # 带路径的文件名
self.read_execute.status[0]= eval(self.read_execute.status[0]) # 字符串转换为字典
if file_size > int(self.read_execute.status[0]['space_size_total'])- int(self.read_execute.status[0]['space_size_used']): # check space大小
self.request.send('false'.encode('utf-8')) # 判断磁盘空间不足
else:
self.request.send('true'.encode('utf-8')) # 磁盘空间足够大
check = self.request.recv(1024).decode()
if check == 'false': # 可用磁盘不足，退出
return 1
if os.path.isfile(file_info): # 文件存在，则查看文件大小
file_offset = os.stat(file_info).st_size
else:
file_offset = 0
self.request.send(str(file_offset).encode('utf-8')) # 发送server端文件大小
f = open(file_info, "ab+") # 打开文件
received_size = 0
m_check = hashlib.md5() # used hashlib.md5 to encryption
while received_size < file_size - file_offset: # 判断接收到的文件大小
data = self.request.recv(1024) # 接受最大为1024大小文件
f.write(data) # 写入文件
received_size += len(data) # 累加收到文件大小
m_check.update(data) # 更新md5校验
self.request.send(m_check.hexdigest().encode()) # 发送校验码，以及防粘连
else:
self.read_execute.status[0]['space_size_used'] += file_size # 上传成功修改使用空间大小
sql.Sql("update %s from %s" % (self.read_execute.status[0], self.table_name))
f.close() def info(self):
"""
返回个人信息
:return:
""" self.table_name = '%s.%s' % (setting.DATABASE['db'],self.server_command['command_param'] ) # 解析用户数据库名
self.read_execute = sql.Sql('select * from %s' % self.table_name) # 获取用户信息
if self.read_execute.status: # 用户存在
self.request.send(self.read_execute.status[0].encode('utf-8'))
else: # 用户不存在
self.request.send(json.dumps('None').encode('utf-8')) def download(self):
""""""
if os.path.isfile(self.server_command['command_param']): # 文件是否在本地存在
file_size = os.stat(self.server_command['command_param']).st_size # 文件大小
self.request.send(str(file_size).encode('utf-8')) # 发送文件大小
else:
file_size = 0 # 如果文件不存在，在大小为0
self.request.send(str(file_size).encode('utf-8'))
return 1
client_file_size = int(self.request.recv(1024).decode()) # client端目前文件大小
if client_file_size == file_size:
return 1
m_filecheck = hashlib.md5()
file_obj = open(self.server_command['command_param'], 'rb') # 打开文件
file_obj.seek(client_file_size) # 文件位置跳转，用于断点续传
send_size = 1024 # 每次发送文件大小
while True:
file_data = file_obj.read(send_size) # 读取文件
if file_data: # 判断文件是否读取完成
self.request.send(file_data) # 发送读取文件内容
file_check = self.request.recv(1024) # 接受server确认信息
m_filecheck.update(file_data) # 文件校验
else:
self.request.send(m_filecheck.hexdigest().encode())
file_obj.close() # 关闭文件
break def cd(self):
"""
切换目录
:return:
"""
if os.path.isdir(self.server_command['command_param']):
cd_dir = self.server_command['command_param']
else:
cd_dir = 'None'
self.request.send(json.dumps(cd_dir).encode('utf-8')) def mkdir(self):
"""
新建文件夹
:return:
"""
if os.path.isdir(self.server_command['command_param']): # 判断新建目录是否存在
check_data = 'File[%s] is already exist ' %self.server_command['command_param']
else:
os.mkdir(self.server_command['command_param']) # 不存在则新建目录
check_data = 'Create successful'
self.request.send(check_data.encode('utf-8')) def change_space(self):
"""
配置用户磁盘配额
:return:
"""
new_info = self.server_command['command_param'].split() # 参数解析
self.table_name = '%s.%s' % (setting.DATABASE['db'], new_info[0]) # 解析用户数据库名
print('tablename:',self.table_name)
self.read_execute = sql.Sql('select * from %s' % self.table_name) # 获取用户信息
if self.read_execute.status: # 用户存在
account_data = eval(self.read_execute.status[0])
print('info::',self.read_execute.status[0])
print('info type:', type(account_data))
account_data['space_size_total'] = str(new_info[1]) # 修改磁盘大小
sql.Sql("update %s from %s" % (account_data, self.table_name))
self.request.send('chang successful'.encode('utf-8'))
else: # 用户不存在
self.request.send('No such user, chang failed'.encode('utf-8')) if __name__ == "__main__":
pass
else:
HOST, PORT = "localhost", 9000
# Create the server, binding to localhost on port 9999
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)
server.serve_forever()
FTP server
import json
import hashlib
from conf import setting
from core import auth
import os
import re
import sys
import socket def admin_required(func):
"""
验证用户是否登录，装饰器
:param func:
:return:
"""
def wrapper(self,*args,**kwargs):
if self.user_data['account_id'] == 'admin': # 判断用户是否为admin
return func(self,*args, **kwargs)
else:
print("No permission... ")
return False
return wrapper class FtpClient(object): def __init__(self):
self.client = socket.socket()
self.command = {}
self.user_data = {}
self.user_data = { # user data
'account_id': None,
'access_authenticated': None, # judge user if authenticated
'dir': {}, # user operation dir info
'home': None, # user home dir
}
self.connect(setting.SERVER_IP, setting.SERVER_PORT) def login(self):
self.init_data = auth.login(self) # call login authenticated function
# self.init_data = {'password': 'e99a18c428cb38d5f260853678922e03', 'space_size_used': 0, 'home_dir': 'C:\\D\\personal_data\\workspace\\FTP\\server/home/greath', 'account': 'greath', 'space_size_total': 102400000}
# print('receive: ',self.init_data )
if self.init_data:
self.user_data['dir']['current_dir'] = self.init_data['home_dir'] # 初始化当前目录
#self.user_data['dir']['cddir'] = self.init_data['home_dir'] # 查找目录
#self.user_data['dir']['chdir'] = {} # 打印查找目录
self.user_data['home'] = self.init_data['home_dir'] # 登入成功后记录用户家目录
self.command['command_key'] = 'ls'
self.login_interview() # 登入成功后显示界面
else:
print('\033[31;0m login failed\033[0m ') def login_interview(self):
"""
登入成功后，显示界面
:return:
"""
while True:
try:
if hasattr(self, self.command['command_key']):
getattr(self,self.command['command_key'])()
self.command['command_key'] = ' '
self.command['command_param'] = ' '
else:
print('\033[41;0mEnter[%s] error...\033[0m' % self.command) except Exception as e:
print('\033[0;31mError Found[%s]' %e)
exit()
command_pro = re.search(r'/home/.*', self.user_data['dir']['current_dir'].replace('\\', '/'))
if not command_pro:
pass # exit("can't find file[%s]" % self.dir['current_dir'])
self.command_pro = command_pro.group() # 输入提示
while True:
command = input('[%s:%s]# ' % (self.user_data['account_id'], self.command_pro)).strip()
if len(command) > 0:
break
self.command_paras(command) @admin_required
def change_space(self):
"""
改变用户磁盘配额
:return:
"""
self.client.send(json.dumps(self.command).encode('utf-8'))
data = self.client.recv(1024).decode()
print(data) def command_paras(self, command):
"""
输入命令解析
:param command:
:return:
"""
command_l = command.split(maxsplit=1) # 提取命令关键字和参数
# print(command_l)
self.command['command_key'] = command_l[0] # 关键字
if len(command_l) > 1:
self.command['command_param'] = command_l[1] #参数
else:
self.command['command_param'] = None def mkdir(self):
"""
新建文件夹
:return:
"""
self.command['command_param'] = '%s/%s' %(self.user_data['dir']['current_dir'],self.command['command_param']) # 相对路径创建文件夹
self.client.send(json.dumps(self.command).encode('utf-8'))
data = self.client.recv(1024).decode()
print(data) def ls(self):
""""""
if not self.command.get('command_param') or self.command['command_param'] == None:
self.command['command_param'] = self.user_data['dir']['current_dir']
self.client.send(json.dumps(self.command).encode('utf-8')) # 发送ls命令
ls_data = json.loads(self.client.recv(10240).decode()) # 接受查询数据
print("\033[36;1m|-- %s/\033[0m" % self.user_data['dir']['current_dir'])
for path in ls_data:
if ls_data[path] == 'dir':
print("\033[32;1m |-- %s\033[0m" % path)
elif ls_data[path] == 'file':
print('\033[30;1m |-- %s\033[0m' % path)
print() def cd(self):
"""
切换目录
:return:
"""
if self.command['command_param'] == '.': # 返回当前目录
self.command['command_param'] = self.user_data['dir']['current_dir']
elif self.command['command_param'] == '..': # 返回上一层目录
if self.command_pro == '/home/%s' % self.user_data['account_id']: # 用户只能访问自己的家目录
print('No access permission to /home')
return 1
self.command['command_param'] = os.path.dirname(self.user_data['dir']['current_dir'].strip('/'))
elif self.command['command_param'] not in self.user_data['dir']['current_dir'].replace('\\', '/'):
self.command['command_param']='%s/%s' %(self.user_data['dir']['current_dir'],self.command['command_param'])
self.client.send(json.dumps(self.command).encode('utf-8')) # 发送cd命令
cd_data = json.loads(self.client.recv(10240).decode()) # 接受查询数据
if cd_data != 'None':
self.user_data['dir']['current_dir'] = cd_data
self.command['command_param'] = self.user_data['dir']['current_dir']
self.command['command_key'] = 'ls'
self.ls()
else:
print('\033[31;1mNo such dir[%s]\033[0m' % (self.command['command_param'])) def upload(self):
"""
上传文件
:return:
"""
if os.path.isfile(self.command['command_param']): # 文件是否在本地存在
self.command['command_filesize'] = int(os.stat(self.command['command_param']).st_size) # 文件大小
# if not self.command.get('command_filelocation'): # 如果没有输入保存路径
self.command['command_filelocation'] = self.user_data['dir']['current_dir'] # 文件保存位置，默认为当前路径
self.client.send(json.dumps(self.command).encode('utf-8')) # 发送upload命令
check = self.client.recv(1024).decode()
if check == 'false':
print('space is not enough, upload failed')
self.client.send('false'.encode('utf-8'))
return 1
else:
self.client.send('true'.encode('utf-8'))
file_offset = int(self.client.recv(1024).decode()) # 接受server端的确认信息
if file_offset < self.command['command_filesize'] : # 判断server端文件大小
self.file_upload(file_offset) # 开始
else:
print('file is already exist.')
else: # 文件在本地不存在
print('\033[31;0mFile [%s] is not exist...\033[0m' % self.command['command_param'])
return 1
del self.command['command_filesize'] # 清除命令无关key
del self.command['command_filelocation'] def file_upload(self, total_send_size):
"""
上传文件
:param init_offset:
:return:
"""
total_send_size = int(total_send_size)
file_obj = open(self.command['command_param'], 'rb') # 打开文件
file_obj.seek(total_send_size) # 文件位置跳转，用于断点续传
send_size = 1024 # 发送文件大小
m_check = hashlib.md5() # 文件校验
file_check_server = 0 # 初始化server端文件校验码
while total_send_size < int(self.command['command_filesize']):# 可以修改size == 0的判断
file_data = file_obj.read(send_size) # 读取文件
self.client.send(file_data) # 发送读取文件内容
file_check_server = self.client.recv(1024).decode() # 接受server确认信息
m_check.update(file_data) # 文件校验
total_send_size += send_size
# 传输过程进度条
percent = '{0:.0%}'.format(total_send_size/ self.command['command_filesize']) #打印进度条
sys.stdout.write(
'\r [%-100s] %s' % ('#' * int(total_send_size / self.command['command_filesize'] * 100), percent))
sys.stdout.flush()
if total_send_size > self.command['command_filesize'] -1024: # 判断最后一次传输大小
send_size = self.command['command_filesize'] - total_send_size
else:
print()
if m_check.hexdigest() == file_check_server: # 相同则上传成功
print('upload file completed!!!')
else:
print('upload failed...')
file_obj.close() # 关闭文件 def info(self):
"""
打印个人信息
:return:
"""
if not self.command['command_param']:
self.command['command_param'] = self.user_data['account_id']
self.client.send(json.dumps(self.command).encode('utf-8'))
data = self.client.recv(10240).decode()
if data != 'None':
data = eval(data)
info = """
-----------------info of {account}------------------
Account:{account}
Home:{home}
Space_size:{space_size}
Space_used:{space_used}
""".format(
account = data['account'],
home = data['home_dir'],
space_size = data['space_size_total'],
space_used = data['space_size_used']
)
print(info)
else:
print("get info failed") def download(self):
""""""
filename = self.command['command_param']
self.command['command_param'] = '%s/%s' %(self.user_data['dir']['current_dir'], self.command['command_param'])
self.client.send(json.dumps(self.command).encode('utf-8')) # 发送download命令
total_file_size = int(self.client.recv(1024).decode()) # 接收文件大小
if total_file_size == 0: # 接收来自server的确认文件是否存在
print("file is not exist in server...")
return 1
received_size = 0 # 初始化client端文件大小
if os.path.isfile(filename): # 判断文件是否存在
file_size = os.stat(filename).st_size
if total_file_size == received_size :
print("this file is already exist") # 文件存在并且大小一致，返回
self.client.send(str(total_file_size).encode('utf-8'))
return 0
self.client.send(str(received_size).encode('utf-8'))
m_filecheck = hashlib.md5()
size = 1024
with open(filename, 'ab+') as f:
while received_size < total_file_size:
if received_size > total_file_size -1024:
size = total_file_size - received_size # 设置每次接受的大小
data = self.client.recv(size)
f.write(data) # 写入数据
# 传输过程进度条
percent = '{0:.0%}'.format(received_size / total_file_size) # 打印进度条
sys.stdout.write(
'\r [%-100s] %s' % ('#' * int(received_size / total_file_size * 100), percent))
sys.stdout.flush()
self.client.send('receive_check'.encode('utf-8')) # 防粘连
received_size += size
m_filecheck.update(data)
else:
print() # 输出换行
file_check = self.client.recv(1024).decode() # 接收server校验码
if file_check == m_filecheck.hexdigest():
print('download successful...') def exit(self):
"""
退出
:return:
"""
exit() def help(self):
msg = '''
ls # list file or folder info
cd dir_param # [..] Return to the previous level directory.
download filename # offer filename in relative paths
upload filename # offer filename in absolute paths
mkdir folder_name # Creat new folder
info account_id # return user info
chang_space account_id new_size # chang user space size configure
exit # exit system
'''
print(msg) def register(self):
"""
function: register new account
:return:
"""
self.command['command_key'] = 'register'
account = input('please input user name: ').strip() # get register account
self.command['command_param'] = account
self.client.send(json.dumps(self.command).encode('utf-8')) # send register command to server
account_exist = self.client.recv(1024).decode()
if account_exist == 'False': # judge this account if exist
password = self.password_get()
self.client.send(password.encode('utf-8')) # send user password to server
self.client.recv(1024) # 防粘连
print("\033[33;0mAccount: [%s] register successful.\033[0m" % account)
else:
print("\033[31;0mAccount: [%s] is already exist.\033[0m" % account)
input('Enter any key to back') def password_get(self):
"""
获取密码，并且hashlib加密
:return:
"""
while True:
password = input('input password: ').strip()
if len(password) == 0:
continue
if self.command['command_key'] == 'register': # 注册需要确认密码
while True:
password_confirm = input('confirm password: ').strip()
if password == password_confirm:
break
else:
print('\033[31;0mPassword is inconsistent...\033[0m')
m = hashlib.md5() # used hashlib.md5 to encryption
m.update(password.encode('utf-8'))
password = m.hexdigest()
return password @staticmethod
def exit_system():
"""
function: exit ftp system
:return:
"""
exit() @staticmethod
def user_choice(menu, menu_dic, *args, **kwargs):
"""
function: provide one interface for user to choice which channel.
:param menu: menu
:param menu_dic: function
:param args:
:param kwargs:
:return:
"""
exit_flag = False
while not exit_flag:
print(menu) # print current menu
user_option = input(">>>: ").strip()
if user_option in menu_dic:
exit_flag = menu_dic[user_option]() # call menu function which user choice
else:
print("\033[31;1mOption does not exist!\033[0m")
input("Enter any key to back") def system_interactive(self):
"""
function: provide interactive between user and ftp system.
:return:
"""
menu = u'''
************** FTP **************
\033[32;1m
1. 登入
2. 注册
3. 退出
\033[0m'''
menu_dict = {
'': self.login,
'': self.register,
'': self.exit_system
}
self.user_choice(menu, menu_dict) def connect(self, ip, port):
'''
连接server
:param ip:
:param port:
:return:
'''
self.client.connect((ip, port)) def main():
ftp = FtpClient()
ftp.system_interactive()
FTP client
Python修炼之路-Socket的更多相关文章
- Python学习之路--Socket
Socket socket通常也称作"套接字",用于描述IP地址和端口,是一个通信链的句柄,应用程序通常通过"套接字"向网络发出请求或者应答网络请求. sock ...
- Python修炼之路-文件操作
Python编程之文件操作 文件操作流程 打开文件,得到文件句柄并赋值给一个变量 通过句柄对文件进行操作 关闭文件 每次文件打开.读取信息时,Python自动记录所达到的位置,好比一个书签,之后每一次 ...
- Python学习之路——socket
一.Socket通常也称作"套接字",用于描述IP地址和端口,是一个通信链的句柄,可以用来实现不同虚拟机或不同计算机之间的通信. socket服务端示例: import socke ...
- python修炼之路——控制语句
Python编程之print python2.x和python3.x的print函数区别:python3.x的print函数需要加括号(),python2.x可以不加. #-*- coding:utf ...
- Python修炼之路-数据类型
Python编程之列表 列表是一个使用一对中括号"[ ]" 括起来的有序的集合,可以通过索引访问列表元素,也可以增加和删除元素. 列表的索引:第一个元素索引为0,最后一个元素索 ...
- Python修炼之路-异常
异常处理 在程序出现bug时一般不会将错误信息直接显示给用户,而是可以自定义显示内容或处理. 常见异常 AttributeError # 试图访问一个对象没有的树形,比如foo.x,但是foo没有属性 ...
- Python修炼之路-函数
Python编程之函数 程序的三种方式 面向对象:类------->class 面向过程:过程------>def 函数式编程:函数------>def 定义函数 函数:逻辑结构化与 ...
- Python修炼之路-模块
模块 模块与包 模块:用来从逻辑上组织python代码(可以定义变量.函数.类.逻辑:实现一个功能),本质就是.py结尾的python文件. 例如,文件名:test.py,对应的模块名为:test 包 ...
- python修炼之路---面向对象
面向对象编程 面向对象编程:使用类和对象实现一类功能. 类与对象 类:类就是一个模板,模板里可以包含多个函数,函数里实现一些功能. 对象:是根据模板创建的实例,通过实例对象可以执行类中的函数. 面向对 ...
随机推荐
- 清除表单input输入框内数据
清除表单input输入框内数据 1. $(':input','#addVoucherType') //'#addVoucherType'表单id .not(':button') .val('') .r ...
- JavaScript日常学习4
JavaScript事件 1.<button id="btn1" onclick="document.getElementById("btn1" ...
- docker(常见调试技巧)：docker打包镜像调试技巧
写Dockerfile可以先不指定CMD.ENTRYPOINT等启动命令,只要拷贝就好了 如下: # Dockerfile for basic-app-client # Build with: # d ...
- unity快捷放置物体操作
https://connect.unity.com/p/zui-jia-shi-jian-dui-xiang-fang-zhi-he-wu-li-xiao-guo 最佳实践系列文章将探讨我们在与客户合 ...
- Unity Shader 基础
推荐: https://www.cnblogs.com/nanwei/p/7277417.html 上面链接作者的整个系列都写的不错 https://www.cnblogs.com/nanwei/ca ...
- TCP/IP笔记——TCP特点、首部格式、滑动窗口
这次总结一下TCP相关的知识. TCP主要特点 面向连接:在通信前必须建立连接(只是逻辑上存在,而不是物理连接) 只能有两个端点:即只能一对一通信(所以通常p2p是用UDP实现的) 提供可靠交付服务: ...
- 红帽学习记录[RHCE] ISCSI远程块储存
目录 iSCSI 定义 组件术语 启动器 目标 ACL 发现 IQN 登录 LUN 节点 门户 TPG 搭建一个iSCSI服务 提供一个iSCSI目标 配置iSCSI客户端 iSCSI 定义 SCSI ...
- Solrcloud+tomcat+zookeeper
准备两台服务器,目录结构如下 主机名 IP地址 tomcat安装路径 zookeeper安装路径 solr安装路径 java安装路径 sht-sgmhadoopnn-01 172.16.101.55 ...
- hbase 查看hfile文件
emp表数据结构 hbase(main):098:0> scan 'emp' ROW COLUMN+CELL row1 column=mycf:depart, timestamp=1555846 ...
- Spring MVC 中使用AOP 进行统一日志管理--注解实现
1.AOP简介 AOP称为面向切面编程 AOP的基本概念 (1)Aspect(切面):通常是一个类,里面可以定义切入点和通知 (2)JointPoint(连接点):程序执行过程中明确的点,一般是方法的 ...