89 lines
3.8 KiB
Python
89 lines
3.8 KiB
Python
|
import os
|
||
|
|
import time
|
||
|
|
import socket
|
||
|
|
import json
|
||
|
|
import threading
|
||
|
|
from .Process_Client import *
|
||
|
|
|
||
|
|
|
||
|
|
class link_client(ProcessClient):
|
||
|
|
def __init__(self):
|
||
|
|
ProcessClient.__init__(self)
|
||
|
|
self.server_host = "127.0.0.1"
|
||
|
|
self.server_port = 7868
|
||
|
|
self.server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||
|
|
self.server_socket.bind((self.server_host, self.server_port))
|
||
|
|
self.server_socket.listen()
|
||
|
|
self.client_socket_dict = {}
|
||
|
|
|
||
|
|
current_file_path = __file__
|
||
|
|
current_file_name = os.path.basename(current_file_path).split('.')[0]
|
||
|
|
self.Process_client_send("Server", "Name", current_file_name)
|
||
|
|
|
||
|
|
self.link_client_Thread = threading.Thread(target=self.link_client)
|
||
|
|
self.link_client_Thread.start()
|
||
|
|
|
||
|
|
def link_client(self):
|
||
|
|
while True:
|
||
|
|
try:
|
||
|
|
client_socket, client_address = self.server_socket.accept()
|
||
|
|
recv_client_Thread = threading.Thread(target=self.recv_client(client_socket), daemon=True)
|
||
|
|
recv_client_Thread.start()
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
def send_client(self, client_socket, genre, target, content):
|
||
|
|
try:
|
||
|
|
data = {"genre": genre, "target": target, "data": content,
|
||
|
|
"datetime": time.strftime('%Y-%m-%d %H:%M:%S', time.localtime())}
|
||
|
|
data_json = json.dumps(data)
|
||
|
|
client_socket.send(data_json.encode("utf-8"))
|
||
|
|
except:
|
||
|
|
client_socket.close()
|
||
|
|
|
||
|
|
def pick_data(self, client_socket, data):
|
||
|
|
match data['genre']:
|
||
|
|
case '登录':
|
||
|
|
content = {'client_socket': client_socket, 'account': data['data']['account'],
|
||
|
|
'password': data['data']['password']}
|
||
|
|
self.Process_client_send("Database_formula", "check_account_state", content)
|
||
|
|
case '注册':
|
||
|
|
content = {'client_socket': client_socket, 'account': data['data']['account'],
|
||
|
|
'password': data['data']['password']}
|
||
|
|
self.Process_client_send("Database_formula", "sign_account", content)
|
||
|
|
|
||
|
|
def recv_client(self, client_socket):
|
||
|
|
state = True
|
||
|
|
while state:
|
||
|
|
try:
|
||
|
|
data_json = client_socket.recv(1024).decode('utf-8')
|
||
|
|
data = json.loads(data_json)
|
||
|
|
self.pick_data(client_socket, data)
|
||
|
|
except:
|
||
|
|
try:
|
||
|
|
client_socket.close()
|
||
|
|
finally:
|
||
|
|
state = False
|
||
|
|
for key, value in self.client_socket_dict.items():
|
||
|
|
if value == client_socket:
|
||
|
|
del self.client_socket_dict[key]
|
||
|
|
self.Process_client_send("Database_formula", "alter_state_database",
|
||
|
|
{"Id": key, "sate": "在线"})
|
||
|
|
break
|
||
|
|
|
||
|
|
def Process_client_pick(self, data):
|
||
|
|
if data['target'] in ['ALL', 'Session_client']:
|
||
|
|
match data['function']:
|
||
|
|
case 'send_client':
|
||
|
|
client_socket = data['content']['client_socket']
|
||
|
|
genre = data['content']['genre']
|
||
|
|
target = data['content']['target']
|
||
|
|
content = data['content']['content']
|
||
|
|
self.send_client(client_socket, genre, target, content)
|
||
|
|
if target == '登录':
|
||
|
|
if data['content']['status'] == 0:
|
||
|
|
self.client_socket_dict[data['content']['account']] = data['data']['client_socket']
|
||
|
|
self.Process_client_send("Database_formula", "alter_state_database",
|
||
|
|
{"Id": data['content']['account'],
|
||
|
|
"sate": "在线"})
|