aboutsummaryrefslogtreecommitdiff
path: root/mammoth/results.py
blob: 59c3a4215f66c6814f6a41b1172c9ad6153cedae (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
import collections

from .lists import unique


class Result(object):
    def __init__(self, value, messages):
        self.value = value
        self.messages = unique(messages)
    
    def map(self, func):
        return Result(func(self.value), self.messages)
    
    def bind(self, func):
        result = func(self.value)
        return Result(result.value, self.messages + result.messages)


Message = collections.namedtuple("Message", ["type", "message"])


def warning(message):
    return Message("warning", message)


def success(value):
    return Result(value, [])


def combine(results):
    values = []
    messages = []
    for result in results:
        values.append(result.value)
        for message in result.messages:
            messages.append(message)
        
    return Result(values, messages)


def map(func, *args):
    return combine(args).map(lambda values: func(*values))