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)
def test_shipper_registration():
register_shipper('myshipper', MyShipper)
assert_equal(get_shipper('myshipper'), MyShipper)
unregister_shipper('myshipper')
assert_equal(get_shipper('myshipper'), None)
