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)
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)
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)
def test_output_string(self):
self.assertEqual(output('foo'), 'foo')
def test_output_int(self):
self.assertEqual(output(1), '1')
