"""Format input files for dynamodb batch-get-items.

Create a file asset_final_ids.txt with one asset final id per line.
"""

import json


def main():
    """Start process."""
    with open('./asset_final_ids.txt') as f:
        lines = f.read().split('\n')
    counter = 0
    index = 0
    items = []
    for line in lines:
        if not line:
            continue
        counter += 1
        if counter >= 100:
            index += 1
            counter = 0
        if len(items) < (index + 1):
            items.append(
                {'prod_content_review_match_audio_match': {'Keys': []}})
        items[index]['prod_content_review_match_audio_match']['Keys'].append({
            'asset_final_id': {'N': line}
        })
    file_suffix = 0
    for item in items:
        file_suffix += 1
        with open(f'./req{file_suffix}.json', 'w') as f:
            f.write(json.dumps(item))


if __name__ == '__main__':
    main()