1

import types

     2

import sys

     4

     5

     6

class JSONRPCServiceBase:

     7

     8

    def __init__(self):

     9

        self.methods = {}

    10

    11

    def response(self, id, result):

    12

        return simplejson.dumps({'version': '1.1', 'id': id,

    13

                                 'result': result, 'error': None})

    14

    15

    def error(self, id, code, message):

    16

        return simplejson.dumps({

    17

            'id': id,

    18

            'version': '1.1',

    19

            'error': {'name': 'JSONRPCError',

    20

                      'code': code,

    21

                      'message': message}

    22

        })

    23

    24

    def add_method(self, name, method):

    25

        self.methods[name] = method

    26

    27

    def process(self, data):

    28

        data = simplejson.loads(data)

    29

        id, method, params = data["id"], data["method"], data["params"]

    30

        if method in self.methods:

    31

            try:

    32

                result = self.methods[method](*params)

    33

                return self.response(id, result)

    34

            except BaseException:

    35

                etype, eval, etb = sys.exc_info()

    36

                return self.error(id, 100, '%s: %s' % (etype.__name__, eval))

    37

            except Exception:

    38

                etype, eval, etb = sys.exc_info()

    39

                return self.error(id, 100, 'Exception %s: %s' % (etype, eval))

    40

        else:

    41

            return self.error(id, 100, 'method "%s" does not exist' % method)

    42

    43

    def listmethods(self):

    44

        return self.methods.keys()