import socket, string, sys

def split_and_remove_quotes(str):

    pos = string.find(str, " ")

    return (str[ : pos], str[pos + 2 : -1])

class DictError(Exception):

    def __init__(self, msg, code):

        self._code = code

    def get_code(self):

        return self._code

class DictConnection:

    def __init__(self, server, port = 2628, identify = 1):

        self._server = server

        self._port = port

        self._conn = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

        self._conn.connect((socket.gethostbyname(self._server),

                              self._port))

        self._conn.recv(1024) # eat welcome message

#          if identify:

#              self._send("CLIENT dict.py/0.00", 1)

    def get_databases(self):

        return map(split_and_remove_quotes,

                   self._interpret_multiline(self._send("SHOW DATABASES")))

    def get_strategies(self):

        return map(split_and_remove_quotes,

                   self._interpret_multiline(self._send("SHOW STRAT")))

    def get_database_info(self, db):

        return self._interpret_multiline(self._send("SHOW INFO %s" % db))

    def get_server_info(self):

        return self._interpret_multiline(self._send("SHOW SERVER"), "\r\n")

    def get_definition(self, word, db = "*"):

        try:

            return self._interpret_multiresp(self._send("DEFINE %s %s" %

                                                        (db, word),

                                                        "\r\n.\r\n250"))

        except DictError, e:

            if e.get_code() == 552:

                return []

            raise

    def get_words(self, search, db = "*", strategy = "prefix"):

        try:

            return map(split_and_remove_quotes,

                       self._interpret_multiline(self._send("MATCH %s %s %s" %

                                                     (db, strategy, search))))

        except DictError, e:

            if e.get_code() == 552:

                return []

            raise

    def _send(self, cmd, resp_term = "\r\n.\r\n"):

        self._conn.send(cmd + "\r\n")

        self._verify(self._conn.recv(1024 * 16))

        return self._receive_response(resp_term)

    def _verify(self, resp):

        if resp[0] < "1" or resp[0] > "3":

            raise DictError(resp[4 : ], int(resp[ : 3]))

    def _interpret_multiline(self, resp):

        lines = string.split(resp, "\r\n")

        return lines[ : lines.index(".")]

    def _interpret_multiresp(self, resp):

        resps = []

        lines = []

        for line in string.split(resp, "\r\n"):

            if line == ".":

                resps.append(lines)

                lines = []

            else:

                lines.append(line)

        return resps

    def _receive_response(self, look_for):

        resp = self._conn.recv(1024 * 16)

        while string.find(resp, look_for) == -1:

            resp = resp + self._conn.recv(1024 * 16)

        return resp

# ---

if __name__ == "__main__":

    from pprint import pprint

    if len(sys.argv) > 1:

        word = sys.argv[1]

    else:

        word = raw_input('Word: ')

    dict = DictConnection("dict.org")

    #pprint(dict.get_server_info())

    #pprint(dict.get_database_info(word))

    #pprint(dict.get_databases())

    #pprint(dict.get_strategies())

    pprint(dict.get_definition(word))

    #pprint(dict.get_words(word))