Ejemplo n.º 1
0
import yaml
import sys

from yq import parser
from yq.operators.match_error import MatchError
from yq.output import output


def main(op_str, input):
    op = parser.parse(op_str)
    data = yaml.load(input)
    try:
        for item in op.apply([data]):
            yield item
    except MatchError as ex:
        print ex


if __name__ == '__main__':
    if len(sys.argv) < 2:
        print >> sys.stderr, 'Usage: {} <yaml>'.format(sys.argv[0])
        sys.exit(2)
    op = sys.argv[1]
    input = sys.stdin.read()
    for item in main(op, input):
        print output(item)
Ejemplo n.º 2
0
import yaml
import sys

from yq import parser
from yq.operators.match_error import MatchError
from yq.output import output


def main(op_str, input):
    op = parser.parse(op_str)
    data = yaml.load(input)
    try:
        for item in op.apply([data]):
            yield item
    except MatchError as ex:
        print ex


if __name__ == '__main__':
    if len(sys.argv) < 2:
         print >> sys.stderr, 'Usage: {} <yaml>'.format(sys.argv[0])
         sys.exit(2)
    op = sys.argv[1]
    input = sys.stdin.read()
    for item in main(op, input):
        print output(item)
Ejemplo n.º 3
0
 def test_output_dict(self, mock_yaml):
     data = {'foo': 'bar'}
     self.assertEqual(output(data), mock_yaml.dump.return_value)
     mock_yaml.dump.assert_called_once_with(data)
Ejemplo n.º 4
0
 def test_output_string(self):
     self.assertEqual(output('foo'), 'foo')
Ejemplo n.º 5
0
 def test_output_int(self):
     self.assertEqual(output(1), '1')