Python socket实现聊天服务器（核心编程学习第二章）

Python socket实现聊天服务器（核心编程学习第二章）

这是第二章书后作业–实现一个聊天服务器，客户端与客户端可进行聊天

功能：注册，登录，全域广播（相当于一个群），获取服务器时间
下面是服务器端（Server）的代码
–注意，其中获取用户数据的代码是我自己写的sqlite3类库和MyLog类库。请前往底部获取。

import socket
import threading
import time
import sqlite3
import MyPackage.MySQLite as SQL
import os
from MyPackage.MyLog import *

class Server:
    def __init__(self,Host:str,Port:int,BufferSize:int):
        self.tcpsok = socket.socket(socket.AF_INET, socket.SOCK_STREAM) #创建服务器套接字
        self.BufferSize=BufferSize
        Addr = (Host, Port)
        self.tcpsok.bind(Addr) #将地址（主机名，端口号对）绑定到套接字上
        ULog=Log("UserLog.db","Log") #连接数据库
        self.ServerLog={}#记录连接用户
        self.Log=Log("UserLog.db","Log")#连接数据库
    def StartListen(self,Count:int):
        self.tcpsok.listen(Count)#开始监听
    def StartAccept(self):
        while True:
            print("Waiting...")
            tcpclient,addr=self.tcpsok.accept()
            username=self.Log.SignIn(tcpclient)
            self.ServerLog[username]=tcpclient
            locals()[addr[1]]=threading.Thread(target=Server.NewCommnubication,args=(self,tcpclient,username)).start()#创建新进程以开启一个会话,记录到一个动态生成的变量中
    def NewCommnubication(self,tcpclient:socket.socket,username):
          print("A New Connected by:  ",username)
          tcpclient.send(bytes('Welcom\nThere have some Users: %s  \n'%(self.ServerLog.keys()),'utf-8'))#发送欢迎信息
          threading.Thread(target=Server.GlobleBroadCast,args=(self,"{0} Online".format(username))).start()
          while True:#实现服务器和客户端的交互
              try:
                  data=tcpclient.recv(self.BufferSize)
                  data=Server.Command(self,tcpclient,data)
              except ConnectionResetError:
                  for i in self.ServerLog:
                      if self.ServerLog[i] == tcpclient:
                          self.ServerLog[i].close()
                          del(self.ServerLog[i])
                          self.GlobleBroadCast(i+" Offline")
                          print(i+" Offline")
                          break
                  break

    def CloseServer(self,tcpsok:socket.socket):
        tcpsok.close()
    def PrintUsers(self):
        print(self.Log)
    def GlobleBroadCast(self,Text:str):
        ctext=time.ctime()+">>"+Text
        for u in self.ServerLog:
            self.ServerLog[u].send(bytes(ctext,'utf-8'))
    def Command(self,tcpclient:socket.socket,Comm:str):
        CommandList=['discon','time']
        if Comm==b'time':
            print("Return Time")
            tcpclient.send(bytes(time.ctime(),'UTF-8'))
        if Comm.startswith(b'rChat'):
            threading.Thread(target=Server.Chat,args=(self,tcpclient,Comm)).start()

    def Chat(self,tcpclient:socket.socket,Comm):
        Comm=str(Comm,'utf-8')
        Comm=Comm[6:]
        for i in self.ServerLog:
            if self.ServerLog[i]==tcpclient:
                name=i
                name=name+": "+Comm
                Comm=name
                del (name)
        Server.GlobleBroadCast(self,Comm)

下面是客户端代码(Client)

import socket
from threading import Thread
class MyClient:
    def __init__(self,Host:str,Port:int,BufferSize:int):
        self.Host=Host
        self.Port=Port
        self.BufferSize=BufferSize
        self.Tcpclient=socket.socket(socket.AF_INET,socket.SOCK_STREAM)
        self.Addr=(self.Host,self.Port)
    def Connenct(self):
        tcp=self.Tcpclient.connect(self.Addr)

        t1=Thread(target=MyClient.Recv,args=(self,0))
        t2=Thread(target=MyClient.Send,args=(self,0))
        t1.start()
        t2.start()
    def Close(self):
        self.Tcpclient.close()
    def Recv(self,u):
        while True:
            try:
                recv = self.Tcpclient.recv(self.BufferSize)
                print(str(recv, 'UTF-8'))
            except ConnectionAbortedError:
                print("Connenction Closed")
                break;
    def Send(self,u):
        while True:
            recv = input("> ")
            if recv.startswith("r "):
                recv=recv[2:]
                data="rChat "
                data=data+recv
                recv=data
                del(data)
            if recv=='exit':
                MyClient.Close(self)
                return 0
            self.Tcpclient.send(bytes(recv, 'UTF-8'))

`
我的sqlite3类库

import sqlite3
class SQLite:
    def __init__(self,Path:str):
        self.sql=sqlite3
        self.Path=Path
        self.conn = self.sql.connect(self.Path)
        self.conn.close()
        self.TName=''
    def ConnectDataBase(self):
        self.conn=self.sql.connect(self.Path)
    def SetTargetTable(self,TableName:str):
        self.TName=TableName
    def Close(self):
        self.conn.close()
    def CreateTable(self,TableName:str,ColumsNameAndType:str):
        SQLite.ConnectDataBase(self)
        cur=self.conn.cursor()
        cur.execute("Create Table %s(%s)"%(TableName,ColumsNameAndType))
        self.conn.commit()
        self.conn.close()
    def NewColum(self,Name,Type):
        SQLite.ConnectDataBase(self)
        cur=self.conn.cursor()
        cur.execute('Alter Table %s add %s %s'%(self.TName,Name,Type))
        self.conn.commit()
        self.conn.close()
    def Insert(self,CoulumName:str,Values:str):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Insert into {0}('{1}') values('{2}')".format(self.TName,CoulumName,Values))
        self.conn.commit()
        self.conn.close()
    def Update(self,ColumName:str,Value:str,XColumName:str,XValue:str):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Update %s set %s='%s' Where %s='%s' "%(self.TName,ColumName,Value,XColumName,XValue))
        self.conn.commit()
        self.conn.close()
    def GetOneData(self,ColumName:str,XColumName:str,XValue:str)->str:
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("select %s from %s where %s='%s'" % ( ColumName, self.TName, XColumName, XValue))
        data=cur.fetchall()
        self.conn.close()
        return data
    def GetOneColumn(self,ColumnName:str)->list:
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("select %s from %s " % (ColumnName, self.TName))
        data = cur.fetchall()
        self.conn.close()
        return data
    def GetAllTableName(self)->list:
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Select name from sqlite_master where type='table' Order by name")
        data = cur.fetchall()
        self.conn.close()
        return data
    def GetAllColumns(self):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("pragma table_info({0})".format(self.TName))
        data = list(cur.fetchall())
        self.conn.close()
        return data
    def DeleteData(self,ColumnName:str,Values:str):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("delete from {0} where {1}='{2}'".format(self.TName,ColumnName,Values))
        self.conn.commit()
        self.conn.close()
        return data
    def DeleteDataAccurately(self,ColumName:str,Value:str,XColumName:str,XValue:str):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Delete From {0} Where {1}='{2}' and {3}='{4}'".format(TableName, ColumnName,Value,XColumnName,XValue))
        self.conn.commit()
        self.conn.close()
    def DeleteColumn(self,ColumnName):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Alter table drop COLUMN {0}".format(ColumnName))
        self.conn.commit()
        self.conn.close()
    def DeleteRow(self,RowName,RowValues):
        SQLite.ConnectDataBase(self)
        cur = self.conn.cursor()
        cur.execute("Delete from '{0}' where {1}='{2}'".format(self.TName,RowName,RowValues))
        self.conn.commit()
        self.conn.close()

MyLog类库

import sqlite3
import MyPackage.MySQLite as SQL
import os
import socket
class Log:
    def __init__(self,DBPath,TableName:str):
        self.DBPath=DBPath
        self.Log={}
        self.UserName=""
        self.BufferSize=1024
        self.TableName=TableName
    def LoadLog(self,Name:str,sock:socket.socket):
        self.Log[Name]=sock


    def SignUp(self,tcpclient:socket.socket):#注册新用户
        data="Please input your UserName:"
        tcpclient.send(bytes(data,'utf-8'))
        username=str(tcpclient.recv(self.BufferSize),'utf-8')
        data="Please input your PassWord:"
        tcpclient.send(bytes(data, 'utf-8'))
        pwd= str(tcpclient.recv(self.BufferSize), 'utf-8')
        sql=SQL.SQLite(self.DBPath)
        sql.SetTargetTable(self.TableName)
        try:
            sql.Insert('Name',username)
            sql.Update('PWD',pwd,'Name',username)
        except sqlite3.IntegrityError:
            data="The UserName {0} is already exists,please rename:"
            tcpclient.send(bytes(data, 'utf-8'))
            Log.SignUp(self,tcpclient)
        data="Register Success! Please Sign-In\n---------------------------------------------"
        tcpclient.send(bytes(data, 'utf-8'))
        Log.SignIn(self,tcpclient)

    def SignIn(self,tcpclient:socket.socket):
        sql = SQL.SQLite(self.DBPath)
        sql.SetTargetTable(self.TableName)
        data="Please input your UserName:\n"
        tcpclient.send(bytes(data,'utf-8'))
        username = str(tcpclient.recv(self.BufferSize),'utf-8')
        data="Please input your PassWord:\n"
        tcpclient.send(bytes(data,'utf-8'))
        pwd = str(tcpclient.recv(self.BufferSize),'utf-8')
        inputpwd=""
        if sql.GetOneData('Name','Name',username) != []:
            while True:
                self.UserName=username
                inputpwd = str(list(sql.GetOneData("PWD", 'Name', username)[0])[0])
                if  inputpwd!=pwd:
                    print(inputpwd)
                    data="PassWord wrong,please rewrite your password:"
                    tcpclient.send(bytes(data,'utf-8'))
                    pwd=str(tcpclient.recv(self.BufferSize),'utf-8')
                else:
                    self.UserName=username
                    break
        else:
            data="No User Named {0},Rewrite(w) or Register(r)?".format(username)
            tcpclient.send(bytes(data,'utf-8'))
            if str(tcpclient.recv(self.BufferSize),'utf-8') is not 'w':
                Log.SignUp(self,tcpclient)
            else:
                Log.SignIn(self,tcpclient)
        return username