飞鸽传书-发送下载、接收文件到Queue
FeiQSend.py(修改部分代码)
def send_file_msg():
"""给指定的ip发送文件"""
dest_ip = input("请输入对方的ip(输入0显示当前所有的在线用户):")
if dest_ip == "0":
print("="*30)
for i, user_info in enumerate(FeiQCoreData.user_list):
print(i, user_info)
print("="*30)
try:
num = int(input("请输入用户对应的序号:"))
except:
print("输入有误...")
return
else:
dest_ip = FeiQCoreData.user_list[num]['ip']
file_name = input("请输入要发送的文件名:")
if dest_ip and file_name:
try:
file_size = os.path.getsize(file_name)
file_ctime = os.path.getctime(file_name)
except:
print("没有此文件。。。。")
else:
option = "%d:%s:%x:%x:%x:" % (0, file_name, file_size, int(file_ctime), FeiQCoreData.IPMSG_FILE_REGULAR)
option = "\0" + option
file_msg = build_msg(FeiQCoreData.IPMSG_SENDMSG|FeiQCoreData.IPMSG_FILEATTACHOPT, option)
send_msg(file_msg, dest_ip)
send_file_info = dict()
send_file_info['packet_id'] = FeiQCoreData.packet_id
send_file_info['file_id'] = 0
send_file_info['file_name'] = file_name
queue_info = dict()
queue_info['type'] = "send_file"
queue_info['data'] = send_file_info
FeiQCoreData.file_queue.put(queue_info)
def download_file():
"""下载文件"""
for i, file_info in enumerate(FeiQCoreData.download_file_list):
print(i, file_info)
try:
num = int(input("请输入要下载的文件序号:"))
except:
print("输入数据有误....")
return
else:
file_info = FeiQCoreData.download_file_list[num]
queue_info = dict()
queue_info['type'] = "download_file"
queue_info['data'] = file_info
FeiQCoreData.file_queue.put(queue_info)
main.py
def print_menu():
"""显示飞鸽传书的功能"""
print(" 飞鸽传书v1.0")
print("1:上线广播")
print("2:下线广播")
print("3:给指定的ip发送数据")
print("4:显示在线用户信息")
print("5:给指定的ip发送文件")
print("6:显示可以下载文件")
print("7:下载文件")
print("0:退出")
def main():
FeiQCoreData.file_queue = multiprocessing.Queue()
tcp_process = multiprocessing.Process(target=FeiQTcp.tcp_main, args=(FeiQCoreData.file_queue,))
tcp_process.start()
create_udp_socket()
recv_msg_thread = threading.Thread(target=FeiQRecv.recv_msg)
recv_msg_thread.start()
while True:
print_menu()
command_num = input("请输入要进行的操作:")
if command_num == "1":
FeiQSend.send_broadcast_online_msg()
elif command_num == "2":
FeiQSend.send_broadcast_offline_msg()
elif command_num == "3":
FeiQSend.send_chat_msg()
elif command_num == "4":
print_online_user()
elif command_num == "5":
FeiQSend.send_file_msg()
elif command_num == "6":
print_all_waiting_files()
elif command_num == "7":
FeiQSend.download_file()
elif command_num == "0":
FeiQSend.send_broadcast_offline_msg()
FeiQCoreData.udp_socket.close()
exit()