示例#1
0
class Client:

	def importScripts(self):
		self.decEncHelper = DecodingEncodingHelper()
		self.inputHandler = InputHandler(self.output)
		self.fileHelper = FileHelper(self.output)
		self.guiHelper = GUIHelper(self.output)

	def setConfig(self):
		Config = self.fileHelper.getConfig()
		self.ipV4 = Config.ip
		self.port = Config.port

	def inizializeClient(self, username, password):
		self.clientObject = ClientObject(username, None, self.ipV4, self.port, "Welcome_Channel") 
		self.connected = False
		self.password = password

	def tryConnect(self):
		try:
			self.clientObject.socketObject = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
			self.clientObject.socketObject.connect((self.clientObject.ip, self.clientObject.port))
			threading.Thread(target=ServerHandler,args=[self.clientObject,self.windows]).start()
			self.clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("011" + self.clientObject.username + ":" + self.password))
			self.connected = True
		except:
			self.connected = False

	def sendInput(self, message):
		if self.connected:
			if str(message).startswith("/"):
					self.inputHandler.handleInput(str(message[1:]), self.clientObject)
			else:
				self.output.append("you: " + message)
				try:
					
					self.clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("001" + message))
				except:
					self.connected = False
		else:
			self.guiHelper.printOutput("not connected")

	def __init__(self, username, password, windows):
		#Imports
		self.windows = windows
		self.output = self.windows[0].output
		self.importScripts()
		#Config
		self.setConfig()
		#Client initializations
		self.inizializeClient(username, password)
		#Client trying to establish a connection
		self.tryConnect()
		
示例#2
0
class InputHandler:

    commandList = list()

    def importScripts(self):
        self.decEncHelper = DecodingEncodingHelper()
        self.channelManager = ChannelManager()
        self.clientManager = ClientManager()
        self.fileHelper = FileHelper()
        self.logHelper = LogHelper()

        self.mysqlHelper = MysqlHelper()

    def createCommand(self, name, syntax, arguments, description):
        command = Command(name, syntax, arguments, description)
        self.commandList.append(command)
        return command

    def initializeCommands(self):
        self.cmdListClients = self.createCommand(
            "listClients", "/listClients", "NONE",
            "Lists all connected clients with their name, ip and channel their in."
        )
        self.cmdClear = self.createCommand("Clear", "/clear", "NONE",
                                           "Clears your interpreter console.")
        self.cmdHelp = self.createCommand(
            "Help", "/help", "NONE", "Shows a list of available commands.")
        self.cmdKick = self.createCommand(
            "Kick", "/kick <name/ip>", "<NAME/IP>",
            "Kicks the given IP from the server.")
        self.cmdBan = self.createCommand(
            "Ban", "/ban <name/ip> <time>", "<NAME/IP> <TIME>",
            "Bans the specified client for the given amount of time in minutes."
        )
        self.cmdMonitorMode = self.createCommand("monitorMode", "/monitorMode",
                                                 "NONE",
                                                 "Switches to monitor mode.")
        self.cmdChangeRank = self.createCommand(
            "changeRank", "/changeRank <name/ip> <rank>", "<NAME/IP> <RANK>",
            "Changes the rank of the given client.")
        self.cmdListChannel = self.createCommand(
            "listChannel", "/listChannel", "NONE",
            "Lists all channels with their belonging clients.")
        self.cmdCreateChannel = self.createCommand(
            "createChannel",
            "/createChannel <name> <description> <password> <accessLevel>",
            "<NAME/DESCRIPTION/PASSWORD/ACCESSLEVEL>",
            "Creates a temporary Channel.")
        self.cmdRemoveChannel = self.createCommand(
            "removeChannel", "/removeChannel <name>", "<NAME>",
            "Removes the give Channel.")

    def __init__(self, upTime):
        self.upTime = upTime
        #Imports
        self.importScripts()
        #Create Commands
        self.initializeCommands()

    def handleInput(self, command):
        command = command.split()

        if command[0] == self.cmdClear.name:
            os.system('cls' if os.name == 'nt' else 'clear')

        elif command[0] == self.cmdListClients.name:
            if len(self.clientManager.clientList) < 1:
                self.logHelper.log("error", "No clients connected")
            else:
                self.logHelper.log("error", "Connected clients:")
                for clientObject in self.clientManager.clientList:
                    self.logHelper.log(
                        "info", clientObject.ip + " with name " +
                        clientObject.username + " in " +
                        clientObject.channelObject.name)

        elif command[0] == self.cmdHelp.name:
            self.logHelper.log("info", "Commands:")
            self.logHelper.log(
                "info",
                "----------------------------------------------------------")
            for command in self.commandList:
                self.logHelper.log(
                    "info", command.syntax + " : " + command.description)
            self.logHelper.log(
                "info",
                "----------------------------------------------------------")

        elif command[0] == self.cmdKick.name:
            if len(self.clientManager.clientList) < 1:
                self.logHelper.log("error", "No clients connected")
            else:
                client = None
                try:
                    client = command[1]
                except IndexError:
                    self.logHelper.log("error",
                                       "Syntax: " + self.cmdKick.syntax)
                if client != None:
                    if self.clientManager.ipExists(client):
                        for clientObject in self.clientManager.clientList:
                            if clientObject.ip == client:
                                self.logHelper.log(
                                    "info", clientObject.ip + " : " +
                                    clientObject.username + " got kicked")
                                clientObject.socketObject.sendall(
                                    self.decEncHelper.stringToBytes("402"))
                                clientObject.socketObject.close()
                    elif self.clientManager.usernameExists(client):
                        for clientObject in self.clientManager.clientList:
                            if clientObject.username.lower() == client:
                                self.logHelper.log(
                                    "info", clientObject.ip + " : " +
                                    clientObject.username + " got kicked")
                                clientObject.socketObject.sendall(
                                    self.decEncHelper.stringToBytes("402"))
                                clientObject.socketObject.close()
                    else:
                        self.logHelper.log(
                            "error", "Your given Ip/Name doesn't exist.")

        elif command[0] == self.cmdBan.name:
            if len(self.clientManager.clientList) < 1:
                self.logHelper.log("error", "No clients connected")
            else:
                client = None
                banTime = None
                try:
                    client = command[1]
                    banTime = int(command[2])
                except IndexError:
                    if client or banTime == None:
                        self.logHelper.log("error",
                                           "Syntax: " + self.cmdBan.syntax)
                if client != None:
                    if self.clientManager.ipExists(client):
                        for clientObject in self.clientManager.clientList:
                            if clientObject.ip == client:
                                if banTime != None:
                                    if banTime == 0:
                                        self.fileHelper.addClientToBanList(
                                            clientObject.ip)
                                        self.logHelper.log(
                                            "info", clientObject.ip + " : " +
                                            clientObject.username +
                                            " got permanantly banned")
                                        clientObject.socketObject.sendall(
                                            self.decEncHelper.stringToBytes(
                                                "405" +
                                                "You got permanantly banned by the console"
                                            ))
                                        clientObject.socketObject.close()
                                    else:
                                        currentTimeStamp = datetime.datetime.now(
                                        ).timestamp()
                                        self.fileHelper.addClientToBanList(
                                            clientObject.ip + ":" +
                                            str(currentTimeStamp +
                                                int(banTime) * 60))
                                        self.logHelper.log(
                                            "info", clientObject.ip + " : " +
                                            clientObject.username +
                                            " got banned for " + str(banTime) +
                                            "minutes")
                                        clientObject.socketObject.sendall(
                                            self.decEncHelper.stringToBytes(
                                                "405" + "You got banned for " +
                                                str(banTime) +
                                                " minutes by the console"))
                                        clientObject.socketObject.close()
                                else:
                                    self.fileHelper.addClientToBanList(
                                        clientObject.ip)
                                    self.logHelper.log(
                                        "info", clientObject.ip + " : " +
                                        clientObject.username +
                                        " got permanantly banned")
                                    clientObject.socketObject.sendall(
                                        self.decEncHelper.stringToBytes(
                                            "405" +
                                            "You got permanantly banned by the console"
                                        ))
                                    clientObject.socketObject.close()
                    elif self.clientManager.usernameExists(client):
                        for clientObject in self.clientManager.clientList:
                            if clientObject.username.lower() == client:
                                if banTime != None:
                                    if banTime == 0:
                                        self.fileHelper.addClientToBanList(
                                            clientObject.ip)
                                        self.logHelper.log(
                                            "info", clientObject.ip + " : " +
                                            clientObject.username +
                                            " got permanantly banned")
                                        clientObject.socketObject.sendall(
                                            self.decEncHelper.stringToBytes(
                                                "405" +
                                                "You got permanantly banned by the console"
                                            ))
                                        clientObject.socketObject.close()
                                    else:
                                        currentTimeStamp = datetime.datetime.now(
                                        ).timestamp()
                                        self.fileHelper.addClientToBanList(
                                            clientObject.ip + ":" +
                                            str(currentTimeStamp +
                                                int(banTime) * 60))
                                        self.logHelper.log(
                                            "info", clientObject.ip + " : " +
                                            clientObject.username +
                                            " got banned for " + str(banTime) +
                                            "minutes")
                                        clientObject.socketObject.sendall(
                                            self.decEncHelper.stringToBytes(
                                                "405" + "You got banned for " +
                                                str(banTime) +
                                                " minutes by the console"))
                                        clientObject.socketObject.close()
                                else:
                                    self.fileHelper.addClientToBanList(
                                        clientObject.ip)
                                    self.logHelper.log(
                                        "info", clientObject.ip + " : " +
                                        clientObject.username +
                                        " got permanantly banned")
                                    clientObject.socketObject.sendall(
                                        self.decEncHelper.stringToBytes(
                                            "405" +
                                            "You got permanantly banned by the console"
                                        ))
                                    clientObject.socketObject.close()
                    else:
                        print(
                            "[Server/Error] Your given Ip/Name doesn't exist.")

        elif command[0] == self.cmdListChannel.name:
            self.logHelper.log("info", "Channels:")
            self.logHelper.log(
                "info",
                "----------------------------------------------------------")
            for channel in self.channelManager.channelList:
                self.logHelper.log(
                    "info", "-" + channel.name + " : Description:" +
                    channel.description)
                self.logHelper.log("info", " Clients:")
                if len(channel.clientList) < 1:
                    self.logHelper.log("info", " -channel is empty")
                else:
                    for client in channel.clientList:
                        self.logHelper.log(
                            "info", " -" + client.ip + " : " + client.username)
            self.logHelper.log(
                "info",
                "----------------------------------------------------------")

        elif command[
                0] == self.cmdCreateChannel.name:  #TODO: add things to remove user error when acces level isnst int and description contains spaces
            name = None
            description = None
            password = None
            accessLevel = None
            try:
                name = command[1]
                description = command[2]
                password = command[3]
                accessLevel = int(command[4])
                self.channelManager.addChannel(
                    Channel(name, description, password, accessLevel, list()))
                self.logHelper.log("info", "Channel " + name + " created.")
            except:
                if name or description or password or accessLevel == None:
                    self.logHelper.log(
                        "error", "Syntax: " + self.cmdCreateChannel.syntax)

        elif command[0] == self.cmdRemoveChannel.name:
            name = None
            try:
                name = command[1]
                for channel in self.channelManager.channelList:
                    if channel.name == name:
                        self.channelManager.removeChannel(channel)
                self.logHelper.log("info", "Channel " + name + " was removed.")
            except:
                if name == None:
                    self.logHelper.log(
                        "error", "Syntax: " + self.cmdRemoveChannel.syntax)

        elif command[
                0] == self.cmdChangeRank.name:  #TODO: add things to remove user error for rank and check if rank isnt even a rank
            if len(self.clientManager.clientList) < 1:
                self.logHelper.log("error", "No clients connected")
            else:
                client = None
                rank = None
                try:
                    client = command[1]
                    rank = command[2]
                except IndexError:
                    self.logHelper.log("error",
                                       "Syntax: " + self.cmdChangeRank.syntax)
                if client != None:
                    if rank != None:
                        if self.clientManager.ipExists(client):
                            for clientObject in self.clientManager.clientList:
                                if clientObject.ip == client:
                                    prevRank = clientObject.rank
                                    clientObject.rank = rank
                                    self.mysqlHelper.updateAccountRank(
                                        clientObject)
                                    self.logHelper.log(
                                        "info", "Changed " + clientObject.ip +
                                        ":" + str(clientObject.port) + " " +
                                        clientObject.username +
                                        " 's rank from " + prevRank + " to " +
                                        rank)

                        elif self.clientManager.usernameExists(client):
                            for clientObject in self.clientManager.clientList:
                                if clientObject.username.lower() == client:
                                    prevRank = clientObject.rank
                                    clientObject.rank = rank
                                    self.mysqlHelper.updateAccountRank(
                                        clientObject)
                                    #clientObject.sendall(self.decEncHelper.stringToBytes("904" + rank))TODO:
                                    self.logHelper.log(
                                        "info", "Changed " + clientObject.ip +
                                        ":" + str(clientObject.port) + " " +
                                        clientObject.username +
                                        " 's rank from " + prevRank + " to " +
                                        rank)
                        else:
                            self.logHelper.log(
                                "error", "Your given Ip/Name doesn't exist.")

        elif command[0] == self.cmdMonitorMode.name:
            monitor = True
            config = self.fileHelper.getConfig("Server Config")
            ip = config.ip + ":" + str(config.port)
            if len(ip) != 20:
                ip = " " * (20 - len(ip)) + ip
            while monitor:
                clearCount = 0
                connectedClients = str(len(self.clientManager.clientList))
                connectedAdmins = str(len(self.clientManager.getAdmins()))
                if len(connectedClients) != 3:
                    connectedClients = "0" * (
                        3 - len(connectedClients)) + connectedClients
                if len(connectedAdmins) != 3:
                    connectedAdmins = "0" * (
                        3 - len(connectedAdmins)) + connectedAdmins
                os.system('cls' if os.name == 'nt' else 'clear')
                try:
                    print(
                        "##############Monitoring##############\n#Server Ip/Port: "
                        + ip + "#\n#Uptime:                     " +
                        time.strftime(
                            '%H:%M:%S',
                            time.gmtime(int(time.time() - self.upTime))) +
                        "#\n#Connected Clients:               " +
                        connectedClients +
                        "#\n#Connected Admins:                " +
                        connectedAdmins +
                        "#\n##############Monitoring##############")
                    while clearCount != 5:
                        sys.stdout.write("\x1b[1A")
                        sys.stdout.write("\x1b[2K")
                        clearCount = clearCount + 1
                    time.sleep(0.5)
                except KeyboardInterrupt:
                    monitor = False
                    os.system('cls' if os.name == 'nt' else 'clear')
                    self.logHelper.log("info", "Exited monitor mode.")

        else:
            self.logHelper.log("error", "Unknown command: " + command[0])
            self.logHelper.log("error", "type /help for a list of commands")
示例#3
0
class Server:

	def importScripts(self):
		self.decEncHelper = DecodingEncodingHelper()
		self.channelManager = ChannelManager()
		self.clientManager = ClientManager()
		self.inputHandler = InputHandler(self.upTime)
		self.fileHelper = FileHelper()
		self.logHelper = LogHelper()

	def setConfig(self):
		config = self.fileHelper.getConfig("Server Config")
		self.port = config.port
		self.ipV4 = config.ip

	def inizializeChannel(self):
		self.welcomeChannel = Channel("Welcome_Channel", "welcome to the server", "No", 0, list())
		self.channel1 = Channel("Channel_1", "Description of channel 1", "No", 0, list())
		self.channel2 = Channel("Channel_2", "Description of channel 1", "No", 0, list())
		self.channel3 = Channel("Channel_3", "Description of channel 1", "No", 0, list())
		self.channelManager.addChannel(self.welcomeChannel)
		self.channelManager.addChannel(self.channel1)
		self.channelManager.addChannel(self.channel2)
		self.channelManager.addChannel(self.channel3)

	def inizializeServer(self):
		self.server = ServerThread((self.ipV4, self.port), ClientHandler)
		serverThread = threading.Thread(target=self.server.serve_forever)
		serverThread.daemon = True
		serverThread.start()
		self.logHelper.log("info" ,"Started server (version: " + self.fileHelper.getConfig("Version") + ") on ip: " + self.ipV4 + " port: " + str(self.port))

	def askForInput(self):
		while True:
			try:
				command = input()
			except KeyboardInterrupt:
				self.logHelper.log("info", "Gracefully stopping server...")
				if len(self.clientManager.clientList) < 1:
					self.logHelper.log("info", "Gracefully stopped server")
					break
				else:
					for clientObject in self.clientManager.clientList:
						clientObject.socketObject.sendall(self.decEncHelper.stringToBytes("403"))
					self.logHelper.log("info", "Gracefully stopped server")
					break
			if str(command).startswith("/"):
				try:
					self.inputHandler.handleInput(str(command[1:]).lower())
				except IndexError:
					self.logHelper.log("info", "type /help for a list of commands.")	
			else:
				self.logHelper.log("info", "Commands always start with (/)")	

	def __init__(self):
		self.upTime = time.time()
		#Imports
		self.importScripts()
		#Config
		self.setConfig()
		#Channel initialization
		self.inizializeChannel()
		#Server initializations
		self.inizializeServer()
		#Console Input
		self.askForInput()
示例#4
0
class MysqlHelper:
    def __init__(self):
        self.fileHelper = FileHelper()
        self.logHelper = LogHelper()
        config = self.fileHelper.getConfig("Mysql Server Config")
        try:
            self.connection = mysql.connector.connect(user=config.username,
                                                      password=config.password,
                                                      host=config.ip,
                                                      database=config.database)
        except:
            print("Couldn't establish connection to mysql database(" +
                  config.database + ") with ip: " +
                  config.ip)  #TODO: find a way to handle with this

    def ExecuteCommand(self, command):
        result = None
        try:
            result = MysqlStatement(
                command, self.connection).execute().escape().fetchall().result
        except AttributeError:
            self.logHelper.log("info", "Created mysql table.")
            self.ExecuteCommandWithoutFetchAndResult(
                "CREATE TABLE accounts ( id INT NOT NULL AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0', PRIMARY KEY (id))"
            )

        return result

    def ExecuteCommandWithoutFetchAndResult(self, command):
        return MysqlStatement(command,
                              self.connection).execute().escape().commit()

    def tryLogin(self, clientObject,
                 password):  #TODO: give better fedback for layer 8
        result = False
        lenght = None
        try:
            lenght = len(
                self.ExecuteCommand(
                    "SELECT * FROM accounts WHERE username = '******'"))
        except:
            lenght = len(
                self.ExecuteCommand(
                    "SELECT * FROM accounts WHERE username = '******'"))
        if lenght > 0:
            if self.ExecuteCommand(
                    "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'")[0][1] != "loggedIn":
                if len(
                        self.ExecuteCommand(
                            "SELECT * FROM accounts WHERE username = '******' and password = '******'")) > 0:
                    self.ExecuteCommandWithoutFetchAndResult(
                        "UPDATE accounts SET loggedIn = 1 WHERE username = '******'")
                    result = True
        return result

    def logoutAccount(self, clientObject):
        self.ExecuteCommandWithoutFetchAndResult(
            "UPDATE accounts SET loggedIn = 0 WHERE username = '******'")

    def getAccountRank(self, clientObject):
        return self.ExecuteCommand(
            "SELECT rank FROM accounts WHERE username = '******'")[0][0]

    def updateAccountRank(self, clientObject):
        self.ExecuteCommandWithoutFetchAndResult(
            "UPDATE accounts SET rank = '" + clientObject.rank +
            "' WHERE username = '******'")
示例#5
0
class MysqlHelper:
    def __init__(self, announce):
        self.mysqlMode = 0  # 0 = Mysql; 1 = MysqlLite

        self.fileHelper = FileHelper()
        self.logHelper = LogHelper()
        config = self.fileHelper.getConfig("Mysql Server Config")
        try:
            self.connection = mysql.connector.connect(user=config.username,
                                                      password=config.password,
                                                      host=config.ip,
                                                      database=config.database)
        except:
            self.mysqlMode = 1
            if (announce):
                print(
                    "[" + datetime.datetime.now().strftime("%H:%M:%S") +
                    " ERROR]: Couldn't establish connection to mysql database("
                    + config.database + ") with ip: " + config.ip)
                print("[" + datetime.datetime.now().strftime("%H:%M:%S") +
                      " INFO]: Falling back to MysqlLite.")

            #sqllite

            self.conn = self.create_connection("data/database.db")

            checkForAccountsTable = "SELECT * FROM accounts"

            result = self.executeCommandOnLite(self.conn,
                                               checkForAccountsTable)
            try:
                for row in result:
                    if row[0] == 1:
                        result = True
                    else:
                        result = False
            except:
                result = False
            if result == False:
                createTableStatement = "CREATE TABLE accounts (id INTEGER NOT NULL PRIMARY KEY AUTOINCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0');"
                print("[" + datetime.datetime.now().strftime("%H:%M:%S") +
                      " INFO]: Created accounts table in MysqlLite database.")
                self.executeCommandOnLite(self.conn, createTableStatement)

    def create_connection(self, db_file):
        try:
            conn = sqlite3.connect(db_file)
            return conn
        except Error as e:
            print(e)
            return e

    def executeCommandOnLite(self, connection, command):
        try:
            cursor = connection.cursor()
            result = cursor.execute(command)
            return result
        except Error as e:
            return e

    def ExecuteCommand(self, command):
        result = None
        try:
            result = MysqlStatement(
                command, self.connection).execute().escape().fetchall().result
        except AttributeError:
            self.logHelper.log("info", "Created mysql table.")
            self.ExecuteCommandWithoutFetchAndResult(
                "CREATE TABLE accounts ( id INT NOT NULL AUTO_INCREMENT, username TEXT NOT NULL, password TEXT NOT NULL, email TEXT NOT NULL, rank TEXT NOT NULL, loggedIn TINYINT NOT NULL DEFAULT '0', PRIMARY KEY (id))"
            )

        return result

    def ExecuteCommandWithoutFetchAndResult(self, command):
        return MysqlStatement(command,
                              self.connection).execute().escape().commit()

    def tryLogin(self, clientObject,
                 password):  #TODO: give better fedback for layer 8
        if self.mysqlMode == 0:

            result = False
            lenght = None
            try:
                lenght = len(
                    self.ExecuteCommand(
                        "SELECT * FROM accounts WHERE username = '******'"))
            except:
                lenght = len(
                    self.ExecuteCommand(
                        "SELECT * FROM accounts WHERE username = '******'"))
            if lenght > 0:
                if self.ExecuteCommand(
                        "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'")[0][1] != "loggedIn":
                    if len(
                            self.ExecuteCommand(
                                "SELECT * FROM accounts WHERE username = '******' and password = '******'")) > 0:
                        self.ExecuteCommandWithoutFetchAndResult(
                            "UPDATE accounts SET loggedIn = 1 WHERE username = '******'")
                        result = True
            return result

        else:

            checkLoggedIn = "select loggedIn,(case when loggedIn = 0 then 'loggedOut' when loggedIn = 1 then 'loggedIn' end) as loggedIn_status FROM accounts WHERE username = '******'"
            result = self.executeCommandOnLite(self.conn, checkLoggedIn)
            try:
                for row in result:
                    if row[0] == 1:
                        result = True
                    else:
                        result = False
            except:
                result = False
            if result == False:
                checkPW = "SELECT * FROM accounts WHERE username = '******' and password = '******'"
                result1 = self.executeCommandOnLite(self.conn, checkPW)
                result5 = False
                try:
                    for row in result1:
                        if (row[1] == clientObject.username):
                            result5 = True
                        else:
                            result5 = False
                except:
                    result5 = False
                if result5:
                    updateStatus = "UPDATE accounts SET loggedIn = 1 WHERE username = '******'"
                    self.executeCommandOnLite(self.conn, updateStatus)
                return result5

    def logoutAccount(self, clientObject):
        if self.mysqlMode == 0:
            self.ExecuteCommandWithoutFetchAndResult(
                "UPDATE accounts SET loggedIn = 0 WHERE username = '******'")
        else:
            logoutAccount = "UPDATE accounts SET loggedIn = 0 WHERE username = '******'"
            self.executeCommandOnLite(self.conn, logoutAccount)

    def getAccountRank(self, clientObject):
        if self.mysqlMode == 0:
            result = self.ExecuteCommand(
                "SELECT rank FROM accounts WHERE username = '******'")[0][0]
        else:
            getRank = "SELECT rank FROM accounts WHERE username = '******'"
            result = self.executeCommandOnLite(self.conn, getRank)
            for row in result:
                print(row[0])
                result = row[0]
        return result

    def updateAccountRank(self, clientObject):
        if self.mysqlMode == 0:
            self.ExecuteCommandWithoutFetchAndResult(
                "UPDATE accounts SET rank = '" + clientObject.rank +
                "' WHERE username = '******'")
        else:
            updateRank = "UPDATE accounts SET rank = '" + clientObject.rank + "' WHERE username = '******'"
            self.executeCommandOnLite(self.conn, updateRank)
示例#6
0
class Client:
    def importScripts(self):
        self.decEncHelper = DecodingEncodingHelper()
        self.inputHandler = InputHandler(self.output)
        self.fileHelper = FileHelper(self.output)
        self.guiHelper = GUIHelper(self.output)

    def setConfig(self):
        Config = self.fileHelper.getConfig()
        self.ipV4 = Config.ip
        self.port = Config.port

    def inizializeClient(self, username, password):
        self.clientObject = ClientObject(username, None, self.ipV4, self.port,
                                         "Welcome_Channel")
        self.connected = False
        self.password = password

    def button(self):
        if self.mainWindow.statusButton.text() != "Offline":
            self.sendInput("/disconnect")
            self.mainWindow.statusButton.setText("Offline")
            self.mainWindow.output.clear()
            self.mainWindow.channelTree.clear()
            self.connected = False
        else:
            data = CustomDialog().getData()
            if data == ":":
                self.mainWindow.close()
            else:
                data = data.split(":")
                self.inizializeClient(data[0], data[1])
                self.tryConnect()

    def tryConnect(self):
        trys = 0
        while not self.connected:
            try:
                self.clientObject.socketObject = socket.socket(
                    socket.AF_INET, socket.SOCK_STREAM)
                self.clientObject.socketObject.connect(
                    (self.clientObject.ip, self.clientObject.port))
                threading.Thread(target=ServerHandler,
                                 args=[self.clientObject,
                                       self.mainWindow]).start()
                self.clientObject.socketObject.sendall(
                    self.decEncHelper.stringToBytes(
                        "011" + self.clientObject.username + ":" +
                        self.password))
                self.connected = True
            except:
                trys = trys + 1
                os.system('cls' if os.name == 'nt' else 'clear')
                self.guiHelper.printOutput(
                    "[Client/Info] Attempting to connect to server with ip: " +
                    self.clientObject.ip + ". Attempts: " + str(trys))
                time.sleep(5)

    def sendInput(self, message):
        if self.connected:
            if str(message).startswith("/"):
                self.inputHandler.handleInput(str(message[1:]),
                                              self.clientObject)
            else:
                self.output.append("you: " + message)
                try:

                    self.clientObject.socketObject.sendall(
                        self.decEncHelper.stringToBytes("001" + message))
                except:
                    self.connected = False
        else:
            self.guiHelper.printOutput("not connected")

    def __init__(self, username, password, mainWindow):
        #Imports
        self.mainWindow = mainWindow
        self.mainWindow.statusButton.clicked.connect(self.button)
        self.output = mainWindow.output
        self.importScripts()
        #Config
        self.setConfig()
        #Client initializations
        self.inizializeClient(username, password)
        #Client trying to establish a connection
        self.tryConnect()