Convert goreplay file to yandex.tank ammo file

goreplay_to_tank_ammo_file.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

import sys

gor_delimeter = "🐵🙈🙉"

def make_ammo(req, case):
    """ makes phantom ammo """

    # phantom ammo template
    ammo_template = (
        "%d %s\n"
        "%s"
    )

    return ammo_template % (len(req), case, req)


def main():
    i = 0
    req_string = ""

    for stdin_line in sys.stdin:
        i += 1
        if i == 1:
            continue
        if stdin_line.strip().startswith("Connection:"):
            continue
        if stdin_line.strip() == "":
            req_string += "Connection: keep-alive\n" # only modification is keep-alive
        if stdin_line.strip() == gor_delimeter:
            sys.stdout.write(make_ammo(req_string, "any"))
            i = 0
            req_string = ""
            continue

        req_string += stdin_line

if __name__ == "__main__":
    main()