pyrot/commands/public.py

from common import userstatus
from commands.common import CommandHelpers as CH
bold = "\x02"
italic = "\x1D"
underline = "\x1F"
reverse = "\x16" 	# swap background and foreground colors ("reverse video")
reset = "\x0F"
blue = "\x0302"
green = "\x0303"
red = "\x0304"
grey = "\x0314"

def do_command(self, connection, event):
    cmdtype, trigger, command, replyto = CH.disect_command(self, event)
    
    # Bot crashes once in a while when a user joins a channel, with "IndexError: list index out of range"  on "command.split()[0] == "joins"". 
    try:
        command.split()[0]
    except:
        return  # Stop, there is no command.
    
    # The actual commands:
    if command == "test":
        if cmdtype == "help":    #Display help text.
            if len(command.split()) is not 1:
                return
            connection.privmsg(replyto, "Strictly for testing purposes only!")
        elif cmdtype == "cmd":
            
            connection.privmsg(replyto, event)
            
    elif command == "cmd" or command == "commands":
        if cmdtype == "help":    #Display help text.
            if len(command.split()) is not 1:
                return
            connection.privmsg(replyto, "Displays a list of commands.")
        elif cmdtype == "cmd":
            connection.privmsg(replyto, grey + "Commands: " + CH.ccc(self, "cmd") + CH.ccc(self, "help") + CH.ccc(self, "joins") + CH.ccc(self, "stopgreet")[:-2] + ".")
    
    elif command == "help":
        if cmdtype == "help":    #Display help text.
            if len(command.split()) is not 1:
                return
            connection.privmsg(replyto, "Explains how to get help on any specific command and hints the user to the commandlist.")
        elif cmdtype == "cmd":
            connection.privmsg(replyto, "Replace the " + italic + "! " + reset + "prefix of any comand with " + italic + self.helpchar + " " + reset + "for help with a specific command. Request the command list with: " + blue + "!cmd")
            connection.privmsg(replyto, grey + "Example: " + reset + blue + self.helpchar + "help")
    
    
    elif command.split()[0] == "joins":
        if cmdtype == "help":    #Display help text.
            if len(command.split()) is not 1:
                return
            connection.privmsg(replyto, "Display amount of joins of user and or channel. Channel and user optional.")
            connection.privmsg(replyto, grey + "Usage: " + blue + "!joins " + reset + italic + "channel user")
        elif cmdtype == "cmd":
            
            # Parse user input
            if len(command.split()) == 1:    # Command contains only !joins.
                user = event.source.nick
                if connection.get_nickname() is not event.target:   # Channel message.
                    channel = event.target
            elif len(command.split()) == 2:  # Command has one argument.
                if command.split()[1] in self.channels:
                    channel = command.split()[1]
                else:
                    user = command.split()[1]
            elif len(command.split()) == 3: # Command has two arguments.
                if not command.split()[1] in self.channels: # Bot does not inhabit requested channel.
                    if not command.split()[2] in self.channels: # User did not revert channel and user in command syntax.
                        connection.action(replyto, "does not inhabit " + red + command.split()[1] + reset + ".")
                        return
                    else:   # User reverted user and channel in command syntax.
                        user = command.split()[1]
                        channel = command.split()[2]
                else: # Bot does inhabit requested channel.
                    user = command.split()[2]
                    channel = command.split()[1]
            elif len(command.split()) < 5:  # To many arguments
                connection.privmsg(replyto, "To many arguments. For help type " + blue + self.helpchar + "joins" + reset + ".")
                return
            
            try:
                if user and channel:
                    userjoins = str(self.db.all("SELECT joins FROM joins WHERE channel_network='" + self.network + "' AND \"user\"='" + user + "' AND user_network='" + self.network + "'"))
                    userchanneljoins = str(self.db.one("SELECT joins FROM joins WHERE channel='" + channel + "' AND channel_network='" + self.network + "' AND \"user\"='" + user + "' AND user_network='" + self.network + "'"))
                    channeljoins = str(sum(self.db.all("SELECT joins FROM joins WHERE channel='" + channel + "' AND channel_network='" + self.network + "' AND user_network='" + self.network + "'")))
                    connection.privmsg(replyto, red + user + reset + " has " + green + str(sum(list(map(int, userjoins[1:-1].split(", "))))) + reset + " joins. Of which " + green + str(userchanneljoins) + reset + " have been in " + red + channel + reset + ", that has " + green + str(channeljoins) + reset + " joins" + reset + " in total.")
                    return
            except:
                pass
            try:
                if user:
                    userjoins = str(self.db.all("SELECT joins FROM joins WHERE channel_network='" + self.network + "' AND \"user\"='" + user + "' AND user_network='" + self.network + "' AND user_network='" + self.network + "'"))
                    connection.privmsg(replyto,  red + user + reset + " has " + green + str(sum(list(map(int, userjoins[1:-1].split(", "))))) + reset + " joins in channels I monitor.")
                    return
            except:
                pass
            try:
                if channel:
                    channeljoins = str(sum(self.db.all("SELECT joins FROM joins WHERE channel='" + channel + "' AND channel_network='" + self.network + "' AND user_network='" + self.network + "'")))
                    connection.privmsg(replyto,  red + channel + reset + " has been joined " + green + channeljoins + reset + " times.")
                    return
            except:
                    pass
            connection.privmsg(replyto, "tBKwtWS Was wondering if this programming was sloppy. When you see this message, it was.")
    
    elif command.split()[0] == "stopgreet":
        if cmdtype == "help":    #Display help text.
            if len(command.split()) is not 1:
                return
            connection.privmsg(replyto, "Stops the bot from greeting you or a specific user. Channel, user and option to resume optional.")
            connection.privmsg(replyto, grey + "Usage: " + blue + "!stopgreet " + red + reset + italic + "resume " + red + "channel user")
        elif cmdtype == "cmd":
            
            # Check for resume variation of command.
            resume = False
            try:
                if command.split()[1] == "resume":
                    resume = True
                    command = command.split(' ', 1)[1]  # Remove the resume argument. Which makes the following logic easier.
            except:
                pass
            
            if len(command.split()) == 1:    # No arguments.
                if event.target == connection.get_nickname():   # PM.
                    connection.privmsg(replyto, "Specify at least a channel. For help type " + blue + self.helpchar + "stopgreet" + reset + ".")
                    return
                user = event.source.nick
                channel = event.target
                
            if len(command.split()) == 2:    # One argument.
                if command.split()[1] not in self.channels: # Argument is not a channel the bot inhabits.
                    if event.target == connection.get_nickname():   # PM.
                        connection.action(replyto, "does not inhabit " + red + command.split()[1] + reset + ".")
                        return
                    # Channel message
                    if not userstatus.atleast_halfop(self, event.source.nick, self.homechannel) and not userstatus.atleast_halfop(self, event.source.nick, event.target):  # Insufficient rights.
                        connection.privmsg(replyto, "Denied. You need to have at least halfop status in " + red + self.homechannel + reset + " or " + red + event.target + reset + ".")
                        return
                    # Stopgreet for x user in current channel
                    user = command.split()[1]
                    channel = event.target
                else:   # Bot inhabit channel.
                    user = event.source.nick
                    channel = command.split()[1]
            if len(command.split()) == 3:    # Two arguments.
                if command.split()[1] not in self.channels: # Bot does not inhabit channel.
                    connection.action(replyto, "does not inhabit " + red + command.split()[1] + reset + ".")
                    return
                if not userstatus.atleast_halfop(self, event.source.nick, self.homechannel) and not userstatus.atleast_halfop(self, event.source.nick, command.split()[1]):  # Insufficient rights.
                    connection.privmsg(replyto, "Denied. You need to have at least halfop status in " + red + self.homechannel + reset + " or " + red + command.split()[1] + reset + ".")
                    return
                user = command.split()[2]
                channel = command.split()[1]
            if len(command.split()) > 3:    # Too many arguments.
                connection.privmsg(replyto, "Too many arguments. For help type " + blue + self.helpchar + "stopgreet" + reset + ".")
                return
            
            # Check for database record.
            result = self.db.one("SELECT id, stopgreet FROM joins WHERE \"user\"='" + user + "' AND user_network='" + self.network + "' AND channel='" + channel + "' AND channel_network='" + self.network + "'")
            if not result:  # No record in database.
                connection.action(replyto, "has not yet had the pleasure of greeting " + red + user + reset + " in " + red + channel + reset + ".")
                return
            
            if resume:
                stopgreet = False
                message = "has already every intention to greet "
            else:
                stopgreet = True
                message = "has already stopped greeting "
            if result[1] == stopgreet:
                connection.action(replyto, message + red + user + reset + " in " + red + channel + reset + ".")
                return
            print(str(stopgreet) + str(user) + str(channel))
            #try:
            print("UPDATE joins SET stopgreet='" + str(stopgreet) + "' WHERE \"user\"='" + user + "' AND user_network='" + self.network + "' AND channel='" + channel + "' AND channel_network='" + self.network + "'")
            self.db.run("UPDATE joins SET stopgreet='" + str(stopgreet) + "' WHERE \"user\"='" + user + "' AND user_network='" + self.network + "' AND channel='" + channel + "' AND channel_network='" + self.network + "'")
            #except:
                #connection.privmsg(replyto, "Failed to update database.")