1 #!usr/bin/env python
2 # -*- coding:utf-8 -*-
3 # auther:Mr.chen
4 # 描述：
5
6 import socket
7 import os
8 import threading
9 import time
10 import json
11 from user_users import PersonInfo
12
13 DIR = os.path.dirname(os.path.abspath(__file__))
14 DIR = DIR+'/Folder/'
15
16 TAG = True
17
18
19
20 def tcplink(conn,addr):
21 """
22 tcp请求分析函数
23 :param conn: tcp连接对象
24 :param addr: 连接地址
25 :return:
26 """
27 print ("收到来自{0}的连接请求".format(addr))
28 conn.send('与主机通信中...')
29 while TAG:
30 try:
31 data = conn.recv(4096)
32 time.sleep(1)
33 if not data:
34 break
35 else:
36 print (data)
37 if data == 'ls':
38 P.view_file(conn)
39 continue
40 action,filename = data.strip().split()
41 action = action.lower()
42 if action == 'put':
43 re = P.Recvfile(conn,filename)
44 if re == True:
45 print ("文件接收成功！")
46 else:
47 print ("文件接收失败！")
48 elif action == 'get':
49
50 P.Sendfile(conn,filename)
51 elif action == 'login':
52 name, password = filename.split(',')
53 P = PersonInfo(name, password)
54 re = P.login()
55 if re == True:
56 conn.send('Ready!')
57 else:
58 conn.send('False!')
59 elif action == 'register':
60 name,password = filename.split(',')
61 P = PersonInfo(name, password)
62 re = P.register()
63 if re == True:
64 conn.send('Ready!')
65 else:
66 conn.send('False!')
67 else:
68 print ("请求方的输入有错！")
69 continue
70 except Exception,e:
71 print "tcplink处理出现问题",e
72 break
73
74
75
76
77 if __name__ == '__main__':
78 host = 'localhost'
79 port = 8888
80 s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
81 s.bind((host,port))
82 s.listen(5)
83 print ("服务运行中：正在监听{0}地址的{1}端口：".format(host,port))
84 while TAG:
85 # 接受一个新连接
86 conn,addr = s.accept()
87 # 创建一个新线程处理TCP连接
88 t = threading.Thread(target=tcplink,args=(conn,addr))
89 t.start()
1 #!usr/bin/env python
2 # -*- coding:utf-8 -*-
3 # auther:Mr.chen
4 # 描述：
5
6 import time
7 import os
8 import pickle
9
10
11 class PersonInfo:
12 """
13 用户模型类
14 """
15 DIR = os.path.dirname(os.path.abspath(__file__))
16 DIR = DIR + '/Folder/'
17 ConfigDir = DIR.replace('Folder','db')
18
19 def __init__(self,name,password):
20 self.Name = name #用户名
21 self.Password = password #密码
22 self.DIR = PersonInfo.DIR + self.Name +'/' #用户家目录
23
24
25 def login(self):
26 """
27 用户登陆
28 :return:
29 """
30 dict = PersonInfo.config_read()
31 if dict == None:
32 return False
33 if self.Name in dict:
34 if dict[self.Name] == self.Password:
35 return True
36 return False
37
38
39
40 def register(self):
41 """
42 用户注册
43 :return:
44 """
45 if os.path.exists(self.DIR) != True:
46 os.system('mkdir'+' '+ self.DIR)
47 try:
48 dict = PersonInfo.config_read()
49 if dict == None:
50 dict = {}
51 if self.Name not in dict:
52 dict[self.Name] = self.Password
53 else:
54 print ("姓名重复")
55 return False
56 re = PersonInfo.config_write(dict)
57 if re == True:
58 return True
59 except Exception,e:
60 print "注册出现异常！",e
61 return False
62
63
64
65 def view_file(self,conn):
66 """
67 查看用户家目录
68 :param conn:
69 :return:
70 """
71 data = os.popen('ls'+' '+ self.DIR).read()
72 conn.sendall(data)
73
74 def Recvfile(self,conn,filename):
75 """
76 接收文件方法
77 :param conn:tcp连接对象
78 :param filename:目标文件名
79 :return:
80 """
81 print ("开始接收文件...")
82 conn.send('Ready!')
83 buffer = []
84 while True:
85 d = conn.recv(4096)
86 if d == 'exit':
87 break
88 else:
89 buffer.append(d)
90 data = ''.join(buffer)
91 if data == '':
92 return False
93 print (data)
94 print (filename)
95 print (self.DIR)
96 with open(self.DIR + filename, 'w') as f:
97 f.write(data)
98 return True
99
100
101 def Sendfile(self,conn,filename):
102 """
103 放送文件方法
104 :param conn: tcp连接对象
105 :param filename: 目标文件名
106 :return:
107 """
108
109 if os.path.exists(self.DIR + filename):
110 print ("开始放送文件...")
111 conn.send('Ready!')
112 time.sleep(1)
113 with open(self.DIR + filename, 'r') as f:
114 while True:
115 data = f.read(4096)
116 print (data)
117 if not data:
118 break
119 conn.sendall(data)
120 time.sleep(1)
121 conn.send('exit')
122 print ("文件放送成功！")
123 else:
124 conn.send('False!')
125
126
127 @staticmethod
128 def config_read():
129 """
130 配置文件全部读取
131 :return:
132 """
133 if os.path.exists(PersonInfo.ConfigDir+'user_config'):
134 with open(PersonInfo.ConfigDir+'user_config','r') as f:
135 dict = pickle.load(f)
136 return dict
137
138
139 @staticmethod
140 def config_write(dict):
141 """
142 配置文件全部写入
143 :param dict:
144 :return:
145 """
146 with open(PersonInfo.ConfigDir + 'user_config', 'w') as f:
147 pickle.dump(dict,f)
148 return True
#!usr/bin/env python
# -*- coding:utf-8 -*-
# auther:Mr.chen
# 描述：
import socket,os
import time
TAG =True
DIR = os.path.dirname(os.path.abspath(__file__))
DIR = DIR+'/Folder/'
HOST = 'localhost'
PORT = 8888
def Recvfile(s,filename):
"""
接收文件方法函数
:param s: 套接字封装对象
:param filename: 目标文件名
:return:
"""
print ("开始下载文件...")
buffer = []
while TAG:
d = s.recv(4096)
if d == 'exit':
break
buffer.append(d)
data = ''.join(buffer)
with open(DIR+filename,'w') as f:
f.write(data)
print ("文件下载完毕！")
def Sendfile(s,filename):
"""
放送文件方法函数
:param s: 套接字封装对象
:param filename: 目标文件名
:return:
"""
print ("开始上传文件！")
if os.path.exists(DIR+filename):
with open(DIR+filename,'r') as f:
while TAG:
data = f.read(4096)
if not data:
break
s.sendall(data)
time.sleep(1)
s.send('exit')
print ("文件上传完毕")
else:
print ("你的目录里没有这个文件")
time.sleep(1)
s.send('exit')
def Confirm(s,command):
"""
验证与服务器连接是否正常；
把用户命令发过去，让服务器做好相应准备准备
:param s: 套接字封装对象
:param command: 用户输入的命令
:return:
"""
s.sendall(command)
re = s.recv(4096)
if re == 'Ready!':
return True
elif re == 'False!':
return False
else:
print ("与服务器连接出现异常！")
def File_transfer(s):
"""
用户指令函数
:param s:
:return:
"""
while TAG:
command = raw_input("请输入你想执行的命令>>")
if not command:
continue
if command.lower().strip() == 'help':
print ("请用'put'+'空格'+'文件名'的格式上传文件")
print ("请用'get'+'空格'+'文件名'的格式下载文件")
print ("输入'ls'查看用户服务器家目录")
continue
if command.lower().strip() == 'ls':
s.send('ls')
data = s.recv(4096)
print (data)
continue
try:
action,filename = command.strip().split()
action = action.lower()
except:
print ("您的输入有误!输入help查看帮助文档")
continue
if action == 'put':
re = Confirm(s,command)
if re == True:
Sendfile(s, filename)
else:
print ("对方服务器没有准备好！")
break
elif action == 'get':
re = Confirm(s,command)
if re == True:
Recvfile(s, filename)
elif re == False:
print ("服务器家目录没有这个文件")
else:
print ("对方服务器没有准备好！")
break
else:
print ("你输入的命令有误！输入help查看帮助文档")
def Login(s):
"""
用户登录
:param s:
:return:
"""
name = raw_input("请输入你的用户名：")
password = raw_input("请输入你的密码：")
command = 'login'+' '+ name + ',' + password
re = Confirm(s, command)
if re == True:
print ("登陆成功！")
File_transfer(s)
elif re == False:
print ("您的输入有误，请重新输入！")
Login(s)
else:
print ("与服务器连接出现异常！")
def Register(s):
"""
用户注册
:param s:
:return:
"""
name = raw_input("请输入你的用户名：")
password = raw_input("请输入你的密码：")
Password = raw_input("请再次输入密码：")
if password != Password:
print ("你的密码两次输入不一致，请重新输入！")
Register(s)
command = 'register' + ' ' + name + ',' + password
print (command)
re = Confirm(s,command)
if re == True:
File_transfer(s)
elif re == False:
print ("用户名重复，请重新输入！")
Register(s)
else:
print ("与服务器连接出现异常！")
def Main(s,log = '未联通主机...'):
"""
用户登陆界面
:param s:
:param log:
:return:
"""
text = """
用户登陆界面 {0}
1，用户登陆
2，用户注册
""".format(log)
print (text)
choose = raw_input("请输入索引进行选择：")
if choose == '1':
Login(s)
elif choose == '2':
Register(s)
else:
print ("你的选择有误！")
if __name__ == "__main__":
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
try:
s.connect((HOST,PORT))
Main(s,s.recv(1024))
except Exception,e:
print "服务器连接不上....",e
finally:
s.close()