def main(): args = parser.parse_args() shpr = shipper.get_shipper(args.shipper)(args) msgs = io.messages(sys.stdin) if not args.no_stamp: msgs = stamp(msgs) if args.tags: msgs = tag(msgs, args.tags) for msg in msgs: payload = json.dumps(msg) if args.bulk: command = json.dumps({"index": {"_index": args.bulk_index, "_type": args.bulk_type}}) payload = "{0}\n{1}\n".format(command, payload) shpr.ship(payload)
def main(): args = parser.parse_args() shpr = shipper.get_shipper(args.shipper)(args) msgs = io.messages(sys.stdin) if not args.no_stamp: msgs = stamp(msgs) if args.tags: msgs = tag(msgs, args.tags) if args.fields: msgs = fields(msgs, args.fields) for msg in msgs: payload = json.dumps(msg) if args.bulk: command = json.dumps({'index': {'_index': args.bulk_index, '_type': args.bulk_type}}) payload = '{0}\n{1}\n'.format(command, payload) shpr.ship(payload)
def test_shipper_registration(): register_shipper('myshipper', MyShipper) assert_equal(get_shipper('myshipper'), MyShipper) unregister_shipper('myshipper') assert_equal(get_shipper('myshipper'), None)