1 基于socketserver在服务器端实现执行cmd命令和上传文件（断点续传） 2 3 #server: 4 5 #!/usr/bin/env python 6 # -*- coding:utf-8 -*- 7 import subprocess 8 import socketserver 9 import os 10 11 BASEDIR = os.path.dirname((os.path.abspath(__file__))) 12 13 class Myserver(socketserver.BaseRequestHandler): 14 15 def handle(self): 16 while True: 17 conn = self.request 18 conn.sendall(bytes('连接成功', encoding='utf8')) 19 while True: 20 cmd_or_load = str(conn.recv(1024), encoding='utf8') 21 if not cmd_or_load:break 22 if cmd_or_load[0:3] == 'cmd': 23 cmd_name, cmd_str = cmd_or_load.split('|') 24 result = subprocess.getoutput(cmd_str) 25 result_lenth = len(result) 26 conn.sendall(bytes(str(result_lenth), encoding='utf8')) 27 conn.recv(1024) #解决粘包 28 conn.sendall(bytes(result, encoding='utf8')) 29 elif cmd_or_load[0:4] == 'load': 30 load_name, file_name, s_path, file_byte_size = cmd_or_load.split('|') 31 file_bytes_size = int(file_byte_size) 32 # file_path = os.path.join(BASEDIR, s_path, file_name) 33 file_path = BASEDIR+s_path+'\\'+file_name 34 35 has_rec = 0 36 37 if os.path.exists(file_path): #2001文件存在 2011 继续上传 2012重新上传 38 conn.sendall(bytes('2001', encoding='utf8')) 39 if str(conn.recv(1024), encoding='utf8') == '2011': 40 has_file_size = os.stat(file_path).st_size 41 has_rec += has_file_size 42 conn.sendall(bytes(str(has_file_size), encoding='utf8')) 43 f = open(file_path, 'ab') 44 elif str(conn.recv(1024)) == '2012': 45 f = open(file_path, 'wb') 46 else: 47 conn.sendall(bytes('2002', encoding='utf8')) 48 f = open(file_path, 'wb') 49 50 data_info = bytes() 51 while has_rec<file_bytes_size: 52 try: 53 data = conn.recv(1024) 54 if not data: 55 raise Exception 56 except Exception: 57 break 58 has_rec += len(data) 59 f.write(data) 60 f.close() 61 print('写入成功') 62 63 64 if __name__ == '__main__': 65 server = socketserver.ThreadingTCPServer(('127.0.0.1', 8009), Myserver) 66 server.serve_forever()
67 #---------------------------------------------------------------- 68 #clinet: 69 70 #!/usr/bin/env python 71 # -*- coding:utf-8 -*- 72 import socket 73 import os 74 75 76 sk = socket.socket() 77 sk.connect(('127.0.0.1', 8009)) 78 79 rec_str = str(sk.recv(1024), encoding='utf8') 80 print(rec_str) 81 82 83 def cmd(): 84 inp = input('请输入命令>>>') 85 comm = 'cmd|' 86 inp_bytes = bytes(comm + inp, encoding='utf8') 87 sk.sendall(inp_bytes) 88 cmd_lenth = int(str(sk.recv(1024), encoding='utf8')) 89 sk.sendall(bytes('1111', encoding='utf8')) # 解决粘包 90 has_recv = 0 91 cmd_info = bytes() 92 while has_recv<cmd_lenth: 93 fetch_info = sk.recv(1024) 94 has_recv += 1024 95 cmd_info += fetch_info 96 cmd_info_str = str(cmd_info ,encoding='utf8') 97 print(cmd_info_str) 98 99 def load(): 100 inp = input('请输入上传文件文件名 服务器保存路径>>>')#input:1.zip|\home 101 loadd = 'load|' 102 file_name = inp.split('|', 1)[0] 103 file_byte_size = os.stat(file_name).st_size 104 file_bytes_size = str(file_byte_size) 105 sk.sendall(bytes(loadd + inp + '|' + file_bytes_size, encoding='utf8')) 106 107 is_or_not = str(sk.recv(1024), encoding='utf8') 108 has_sent = 0 109 110 if is_or_not == '2001': 111 inp = input('文件已存在\n1、继续上传 2、重新上传>>>') 112 if inp == '1': 113 sk.sendall(bytes('2011', encoding='utf8')) 114 has_sent_size = int(str(sk.recv(1024), encoding='utf8')) 115 if has_sent_size == file_byte_size: 116 print('文件已全部发送') 117 else: 118 print('已经发送%d字节'%has_sent_size) 119 has_sent += has_sent_size 120 elif inp == '2': #2001文件存在 2011 继续上传 2012 重新上传 121 sk.sendall(bytes('2012', encoding='utf8')) 122 elif is_or_not == '2002': 123 pass 124 125 f = open(file_name, 'rb') 126 f.seek(has_sent) 127 while has_sent<file_byte_size: 128 data = f.read(1024) 129 sk.sendall(data) 130 has_sent += len(data) 131 print('\r已发送{}字节 共{}字节'.format(has_sent, file_byte_size), end='') 132 f.close() 133 print('\n发送成功') 134 135 136 while True: 137 inp = input('1、执行命令 2、上传文件>>>') 138 if inp == '1': 139 cmd() 140 elif inp == '2': 141 load()
