twitch_observer_type_perf_test.py

import timeit

d = {"TWITCHCHATJOIN": 'JOIN',
     "TWITCHCHATLEAVE": 'PART',
     "TWITCHCHATMESSAGE": 'PRIVMSG',
     "TWITCHCHATMODE": 'MODE',
     "TWITCHCHATCLEARCHAT": 'CLEARCHAT',
     "TWITCHCHATHOSTTARGET": 'HOSTTARGET',
     "TWITCHCHATNOTICE": 'NOTICE',
     "TWITCHCHATRECONNECT": 'RECONNECT',
     "TWITCHCHATROOMSTATE": 'ROOMSTATE',
     "TWITCHCHATUSERNOTICE": 'USERNOTICE',
     "TWITCHCHATUSERSTATE": 'USERSTATE',
     "TWITCHCHATWHISPER": 'WHISPER',
     "TWITCHCHATCOMMAND": 'COMMAND'}

def from_command(command):
    return d.get(command) or 'COMMAND'

class TwitchChatEventType(object):
    TWITCHCHATJOIN = 'JOIN'
    TWITCHCHATLEAVE = 'PART'
    TWITCHCHATMESSAGE = 'PRIVMSG'
    TWITCHCHATMODE = 'MODE'
    TWITCHCHATCLEARCHAT = 'CLEARCHAT'
    TWITCHCHATHOSTTARGET = 'HOSTTARGET'
    TWITCHCHATNOTICE = 'NOTICE'
    TWITCHCHATRECONNECT = 'RECONNECT'
    TWITCHCHATROOMSTATE = 'ROOMSTATE'
    TWITCHCHATUSERNOTICE = 'USERNOTICE'
    TWITCHCHATUSERSTATE = 'USERSTATE'
    TWITCHCHATWHISPER = 'WHISPER'
    TWITCHCHATCOMMAND = 'COMMAND'

    @classmethod
    def from_command(cls, command):
        if hasattr(cls, command):
            return getattr(cls, command)
        return cls.TWITCHCHATCOMMAND

def from_command_2(command):
    if command == "TWITCHCHATJOIN":
        return 'JOIN'
    elif command == "TWITCHCHATLEAVE":
        return 'PART'
    elif command == "TWITCHCHATMESSAGE":
        return 'PRIVMSG'
    elif command == "TWITCHCHATMODE":
        return 'MODE'
    elif command == "TWITCHCHATCLEARCHAT":
        return 'CLEARCHAT'
    elif command == "TWITCHCHATHOSTTARGET":
        return 'HOSTTARGET'
    elif command == "TWITCHCHATNOTICE":
        return 'NOTICE'
    elif command == "TWITCHCHATRECONNECT":
        return 'RECONNECT'
    elif command == "TWITCHCHATROOMSTATE":
        return 'ROOMSTATE'
    elif command == "TWITCHCHATUSERNOTICE":
        return 'USERNOTICE'
    elif command == "TWITCHCHATUSERSTATE":
        return 'USERSTATE'
    elif command == "TWITCHCHATWHISPER":
        return 'WHISPER'

    return 'COMMAND'

def a():
    from_command("TWITCHCHATMODE")

def b():
    TwitchChatEventType.from_command("TWITCHCHATMODE")

def c():
    from_command_2("TWITCHCHATMODE")

count = 10000
total_1 = timeit.timeit(a, number=count)
print("Cached dictionary:")
print(total_1 / count)
print()

total_2 = timeit.timeit(b, number=count)
print("Getattr:")
print(total_2 / count)
print()

total_3 = timeit.timeit(c, number=count)
print("If statement:")
print(total_3 / count)
print()