#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : hgh
import hashlib
import os
import json
from conf import settings
class User_auth(object):
 def auth(self, account_info):
 """
 #此功能是進行用戶的登錄信息驗證，如果登錄成功，那么返回用戶對應的http狀態(tài)碼及賬戶信息,否則只返回http狀態(tài)碼
 :param account_info: 用戶的賬戶信息：用戶名，密碼
 :return:
 """
 name = account_info.split(":")[0]
 pwd = account_info.split(":")[1]
 pwd = self.hash(pwd.encode()) # 將用戶名的密碼轉換成hash值
 user_db_file = settings.DATABASE + r"\%s.db" % name # 也可以寫成 "\\%s.db" or "/%s.db"
 if os.path.isfile(user_db_file): # 輸入的用戶名存在
  with open(user_db_file) as fr:
  user_db_info = json.loads(fr.read()) # or josn.load(fr)
  if pwd == user_db_info['password']:
   return "200", user_db_info # 確定，客戶請求成功
  else:
   return "403.11", None # 密碼錯誤
 else:
  return "400", None # 用戶名不存在，用戶認證失敗
 def hash(self, pwd):
 """
  用戶的密碼加密
 :param self:
 :param pwd: 用戶密碼
 :return:
 """
 m = hashlib.md5()
 m.update(pwd)
 return m.hexdigest()

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : hgh
import hashlib
import sys
class Breakpoint(object):
 # 本模塊確認用戶上傳或下載的文件是否存在，如果存在是否需要斷點續(xù)傳
 def transfer(self, filename, has_send_size, total_size, conn):
 """
 進行續(xù)傳
 :param filename:
 :param has_send_size: 已經發(fā)送的文件大小
 :param total_size: 需要傳輸文件總大小
 :param conn: 客戶端和服務端進行數據交換的接口
 :return:
 """
 with open(filename, 'rb') as fr:
  fr.seek(has_send_size) # 定位到續(xù)傳的位置
  print("has_send", has_send_size, "total", total_size)
  m = hashlib.md5()
  if has_send_size == total_size:
  self.progress_bar(has_send_size, total_size)
  for line in fr:
  conn.send(line)
  m.update(line)
  has_send_size += len(line)
  # self.progress_bar(has_send_size,total_size)
 return m.hexdigest()
 def progress_bar(self, has_send_size, total_size):
 bar_width = 50 # 進度條長度
 process = has_send_size / total_size
 send_bar = int(process * bar_width + 0.5) # 發(fā)送的數據占到的進度條長度，四舍五入取整
 sys.stdout.write("#" * send_bar + "=" * (bar_width - send_bar) + "\r") # 注意點：只能這么寫才能達到要求
 sys.stdout.write("\r%.2f%%: %s%s" % (process * 100, "#" * send_bar, "=" * (bar_width - send_bar))) # 注意點：在pycharm中要加\r\n
         # 用sublime只要\r否則換行
 sys.stdout.flush()

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : hgh
import sys
import os
from core import socket_server
path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(path)
if __name__ == "__main__":
 HOST, PORT = "192.168.40.1", 9901
 server = socket_server.socketserver.ThreadingTCPServer((HOST, PORT), socket_server.MyTCPServer)
 server.serve_forever()

 #!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author : hgh
from core import socket_client
import os
import sys
path = os.path.dirname(os.path.abspath(__file__))
sys.path.append(path)
if __name__ == "__main__":
 host, port = "192.168.40.1", 9901
 myClient = socket_client.MySocketClient(host, port)
 myClient.start()