Entitlement Stripper TSV

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
""" Entitlements Stripper by Al-Azif
https://github.com/Al-Azif/
MIT License
"""

import json
import sys


def main():
    output = 'CID\tName\tPKG Link\tSize\r\n'
    with open(sys.argv[1], 'rb') as buf:
        raw_json = json.loads(buf.read())

    for entitlement in raw_json['entitlements']:
        cid = entitlement['id']
        name = entitlement['game_meta']['name']

        try:
            pkg = entitlement['entitlement_attributes'][0]['reference_package_url']
            size = entitlement['entitlement_attributes'][0]['package_file_size']
        except:
            pass

        try:
            pkg = entitlement['drm_def']['drmContents'][0]['contentUrl']
            size = entitlement['drm_def']['drmContents'][0]['contentSize']
        except:
            pass

        try:
            if not size:
                size = "UNK"
        except:
            size = "UNK"

        try:
            if pkg:
                output += '{}\t{}\t{}\t{}\r\n'.format(cid, name, pkg, size)
        except:
            pass

    with open('output.txt', 'w+', encoding='utf-8') as buf:
        buf.write(output)


if __name__ == '__main__':
    main()