IRC-Speak/irc_speak.py

#!/usr/bin/env python2

"""
Copyright (C) 2011 by Wesley Kerfoot

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""

try:
    import simplejson as json
except ImportError as exception:
    print exception
    import json

import xchat
import argparse
import os
from espeak import espeak

options_path = os.path.expanduser("~/.xchat2/options.json")

__module_name__ = "IRC Speak" 
__module_version__ = "1.1" 
__module_description__ = "Allows you to make certain users' text run through a voice synthesizer"

editor = argparse.ArgumentParser(description="""
Specify a user. If that user already exists their options will be updated. If they do not exist they will be added.
If you do not specify arg(s) then the default values will be used for each of them.
""")

editor.add_argument("--name", nargs=1, type=str, required=True, help="e.g. /ircspeak set -volume=<int> -pitch=<int> --name=foo")
editor.add_argument("-volume", nargs=1, type=int, default=[70], help="default=70")
editor.add_argument("-pitch", nargs=1, type=int, default=[50], help="default=50")
editor.add_argument("-rate", nargs=1, type=int, default=[175], help="default=175")
editor.add_argument("-vrange", nargs=1, type=int, default=[50], help="default=50")
editor.add_argument("-language", nargs=1, type=str, help="type \"/ircspeak langlist\" to get a list of languages", default=["english"])

options = json.load(open(options_path, "rw"))

aliases = {"rate" : 1,
            "volume" : 2,
            "pitch" : 3,
            "vrange" : 4}

def edit_users(*args, **new_options):
    """Edits the options dictionary to add/update a user's settings"""
    
    nick = new_options['name'][0]
    if not options.has_key(nick):
        options[nick] = {"args": {}}
        
    for item in new_options:
        if (new_options[item] and (item!="name") and (item!="language")):
        # don't want to use name or language, because language needs to get set separately
        # and name is only used to create the main dictionary for their options
            options[nick]["args"][item] = new_options[item][0]
    
    if new_options["language"]:
        options[nick]["language"] = new_options["language"][0]
    return options    

def irc_speak(word, word_eol, users):
    """Checks to see if a user is set to be voiced, and if so synthesizes their text"""
    
    if options.has_key(word[0]): # check to see if the user is in the options dictionary (word[0] == their nick)
        [espeak.set_parameter(aliases[arg], options[word[0]]["args"][arg]) for arg in options[word[0]]["args"]]
        # options[word[0]]["args"][arg] is the same as options[nickname]["args"][arg] (where arg is some one of the names in aliases
        # which corresponds to some integer value in options[nick][args])
        espeak.set_voice(name=options[word[0]]["language"])
        return espeak.synth(word[1])
    return # return nothing because they weren't in the options dictionary

def set_user(arguments):
    try:
        return edit_users(**vars(editor.parse_args(arguments[1:])))
        # argparse raises a SystemExit error if you use --help or -h, or have the wrong arguments
    except SystemExit as exception:
        return "Exited with status %s" % (exception)
              
def save(arguments):
    with open(options_path, mode='w') as jsonfile:
        json.dump(options, jsonfile)
    return "Saved!"

def commands(word, word_eol, users):
    """Function for doing different commands with XChat"""
    
    arguments = word[1:]

    if not arguments:
        return xchat.EAT_XCHAT
            
    commands = {"set" : set_user,
            "langlist" : (lambda x:", ".join([item["name"] for item in espeak.list_voices()])),
            "save" : save}
    try:
        print commands[arguments[0]](arguments)
    except KeyError as exception:
        print "No such option", exception
            
    return xchat.EAT_XCHAT

xchat.hook_print("Channel Message", irc_speak)
xchat.hook_command("ircspeak",commands, userdata=None, help="/ircspeak set --help")