r"""CouchDB view function handler implementation for Python.

:param input: the readable file-like object to read input from

:param output: the writable file-like object to write output to

"""

functions = []

def _writejson(obj):

obj = json.encode(obj)

if hasattr(output, 'encoding'):

output.write(obj)

output.write('\n')

else:

if isinstance(obj, util.utype):

obj = obj.encode('utf-8')

output.write(obj)

output.write(b'\n')

output.flush()

def _log(message):

if not isinstance(message, util.strbase):

message = json.encode(message)

_writejson({'log': message})

def reset(config=None):

del functions[:]

return True

def compile_fun(string):

string = BOM_UTF8 + string.encode('utf-8')

globals_ = {}

try:

util.pyexec(string, {'log': _log}, globals_)

except Exception as e:

return {'error': {

'id': 'map_compilation_error',

'reason': e.args[0]

}}

err = {'error': {

'id': 'map_compilation_error',

'reason': 'string must eval to a function '

'(ex: "def(doc): return 1")'

}}

if len(globals_) != 1:

return err

function = list(globals_.values())[0]

if type(function) is not FunctionType:

return err

return function

def add_fun(string):

functions.append(compile_fun(string))

return True

def map_doc(doc):

results = []

for function in functions:

try:

results.append([[key, value] for key, value in function(doc)])

except Exception as e:

log.error('runtime error in map function: %s', e,

exc_info=True)

results.append([])

_log(traceback.format_exc())

return results

def reduce(*cmd, **kwargs):

code = BOM_UTF8 + cmd[0][0].encode('utf-8')

args = cmd[1]

globals_ = {}

try:

util.pyexec(code, {'log': _log}, globals_)

except Exception as e:

log.error('runtime error in reduce function: %s', e,

exc_info=True)

return {'error': {

'id': 'reduce_compilation_error',

'reason': e.args[0]

}}

err = {'error': {

'id': 'reduce_compilation_error',

'reason': 'string must eval to a function '

'(ex: "def(keys, values): return 1")'

}}

if len(globals_) != 1:

return err

function = list(globals_.values())[0]

if type(function) is not FunctionType:

return err

rereduce = kwargs.get('rereduce', False)

results = []

if rereduce:

keys = None

vals = args

else:

if args:

keys, vals = zip(*args)

else:

keys, vals = [], []

if util.funcode(function).co_argcount == 3:

results = function(keys, vals, rereduce)

else:

results = function(keys, vals)

return [True, [results]]

def rereduce(*cmd):

# Note: weird kwargs is for Python 2.5 compat

return reduce(*cmd, **{'rereduce': True})

def ddoc(*cmd):

if cmd[0] == 'new':

ddoc = cmd[2]

ddoc['updates'] = dict(

(name, compile_fun(value))

for name, value in ddoc['updates'].items())

ddocs[cmd[1]] = ddoc

return True

else:

ddoc = ddocs[cmd[0]]

action = cmd[1]

if action[0] == 'updates':

if cmd[2][1]['method'] not in ('POST', 'PUT'):

return [

'up',

None,

{'status': 405, 'body': 'Method not allowed'},

]

else:

fun = ddoc['updates'][action[1]]

doc, body = fun(*cmd[2])

res = ['up', doc, body]

sys.stderr.flush()

return res

handlers = {

'add_fun': add_fun,

'ddoc': ddoc,

'map_doc': map_doc,

'reduce': reduce,

'rereduce': rereduce,

'reset': reset,

}

try:

while True:

line = input.readline()

if not line:

break

try:

cmd = json.decode(line)

log.debug('Processing %r', cmd)

except ValueError as e:

log.error('Error: %s', e, exc_info=True)

return 1

else:

retval = handlers[cmd[0]](*cmd[1:])

log.debug('Returning %r', retval)

_writejson(retval)

except KeyboardInterrupt:

return 0

except Exception as e:

log.error('Error: %s', e, exc_info=True)

The %(name)s command runs the CouchDB Python view server.

The exit status is 0 for success or 1 for failure.

Options:

--version display version information and exit

-h, --help display a short help message and exit

--json-module=<name> set the JSON module to use ('simplejson', 'cjson',

or 'json' are supported)

--log-file=<file> name of the file to write log messages to, or '-' to

enable logging to the standard error stream

--debug enable debug logging; requires --log-file to be

specified

Report bugs via the web at <https://github.com/djc/couchdb-python/issues>.

"""Command-line entry point for running the view server."""

import getopt

from couchdb import __version__ as VERSION

try:

option_list, argument_list = getopt.gnu_getopt(

sys.argv[1:], 'h',

['version', 'help', 'json-module=', 'debug', 'log-file=']

)

message = None

for option, value in option_list:

if option in ('--version'):

message = _VERSION % dict(name=os.path.basename(sys.argv[0]),

version=VERSION)

elif option in ('-h', '--help'):

message = _HELP % dict(name=os.path.basename(sys.argv[0]))

elif option in ('--json-module'):

json.use(module=value)

elif option in ('--debug'):

log.setLevel(logging.DEBUG)

elif option in ('--log-file'):

if value == '-':

handler = logging.StreamHandler(sys.stderr)

handler.setFormatter(logging.Formatter(

' -> [%(levelname)s] %(message)s'

))

else:

handler = logging.FileHandler(value)

handler.setFormatter(logging.Formatter(

'[%(asctime)s] [%(levelname)s] %(message)s'

))

log.addHandler(handler)

if message:

sys.stdout.write(message)

sys.stdout.flush()

sys.exit(0)

except getopt.GetoptError as error:

message = '%s\n\nTry `%s --help` for more information.\n' % (

str(error), os.path.basename(sys.argv[0])

)

sys.stderr.write(message)

sys.stderr.flush()

sys.exit(1)