Tvheadend

import sys, datetime, json

dict_chs = {}

dict_pr = {}

dict_des = {}

dict_output = {}

chans_read = False

packet = sys.stdin.buffer.read(3)

while sys.stdin.buffer.read !="":

    if len(packet) > 2 :

        tid = packet[0]

        ssi = packet[1] & 0x80

        tlen = (packet[1] & 0xF) << 8 | packet[2]

        packet = sys.stdin.buffer.read(tlen)

        if tid == 0xc8 and not chans_read :

            t_ext = packet[1]

            chans = packet[117]

            sid_pointer = 118

            ch_pointer = 118 + (chans*8)

            chan_count = 0

            if t_ext == 0 :

                if len(packet) > ch_pointer :

                    while chan_count < chans :

                        nid = packet[sid_pointer] << 8 | packet[sid_pointer+1]

                        tsid = packet[sid_pointer+2] << 8 | packet[sid_pointer+3]

                        sid = packet[sid_pointer+4] << 8 | packet[sid_pointer+5]

                        name_len = packet[ch_pointer] & 0x3f

                        name = packet[ch_pointer+1:ch_pointer+1+name_len].decode('iso8859-9')

                        dict_chs[chan_count+1] = {}

                        dict_chs[chan_count+1]['nid'] = nid

                        dict_chs[chan_count+1]['tsid'] = tsid

                        dict_chs[chan_count+1]['sid'] = sid

                        dict_chs[chan_count+1]['name'] = name

                        chan_count += 1

                        sid_pointer += 8

                        ch_pointer = ch_pointer+1+name_len

                    chans_read = True

        elif tid == 0xdc :

            pos = 5

            while pos+19 < len(packet) :

                ch_id = packet[4]+1

                mhw2_mjd_hi = packet[pos+11]

                mhw2_mjd_lo = packet[pos+12]

                mhw2_hours = (((packet[pos+13] & 0xf0) >> 4) * 10) + (packet[pos+13] & 0x0f)

                mhw2_minutes = (((packet[pos+14] & 0xf0) >> 4) * 10) + (packet[pos+14] & 0x0f)

                mhw2_seconds = (((packet[pos+15] & 0xf0) >> 4) * 10) + (packet[pos+15] & 0x0f)

                mhw2_epoch = (((mhw2_mjd_hi << 8 | mhw2_mjd_lo) - 40587) * 86400) + (mhw2_hours * 3600) + (mhw2_minutes * 60) + mhw2_seconds

                duration = ((packet[pos+16] << 8)|packet[pos+17]) >> 4

                mhw2_duration_hi = (duration&0xFF00) >> 8

                mhw2_duration_lo = duration&0xFF

                title_id = (packet[pos+7] << 24) | (packet[pos+8] << 16) | (packet[pos+9] << 8) | packet[pos+10]

                summary_id = (packet[pos+4] << 16) | (packet[pos+5] << 8) | packet[pos+6]

                slen = packet[pos+18] & 0x7f

                pr_title = packet[pos+19:pos+19+slen].decode('iso8859-9')

                if title_id not in dict_pr :

                    dict_pr[title_id] = {}

                dict_pr[title_id]['channel'] = ch_id

                dict_pr[title_id]['start'] = datetime.datetime.fromtimestamp(mhw2_epoch).strftime('%d/%m/%Y %H:%M:%S')

                dict_pr[title_id]['end'] = datetime.datetime.fromtimestamp(mhw2_epoch+(duration*60)).strftime('%d/%m/%Y %H:%M:%S')

                dict_pr[title_id]['title'] = pr_title

                dict_pr[title_id]['desc_id'] = summary_id

                pos = pos+19+slen+1

        elif tid == 0x96 :

            if ssi == 0x80 :

                #PID 644 TID 150 Short descriptions

                summary_id = (packet[6] << 16) | (packet[7] << 8) | packet[8]

                subt_len = packet[11]

                if not (len(packet) < (14+subt_len)):

                    subt_text = packet[12:12+subt_len].decode('iso8859-9')

                    desc_len = packet[12+subt_len+1]

                    if not (len(packet) < (26+subt_len+desc_len)) :

                        desc_text = packet[14+subt_len:14+subt_len+desc_len].decode('iso8859-9')

                        title_id = (packet[14+subt_len+desc_len+9] << 24) | (packet[14+subt_len+desc_len+10] << 16) | (packet[14+subt_len+desc_len+11] << 8) | packet[14+subt_len+desc_len+12]

                        if summary_id not in dict_des :

                            dict_des[summary_id] = {}

                        dict_des[summary_id]['title_id'] = title_id

                        dict_des[summary_id]['subtitle'] = subt_text

                        dict_des[summary_id]['summary'] = desc_text

            elif ssi == 0 :

                #PID 642 TID 150 Long descriptions

                summary_id = (packet[3] << 16) | (packet[4] << 8) | packet[5]

                subt_len = packet[16]

                if not (len(packet) < (19+subt_len)):

                    subt_text = packet[17:17+subt_len].decode('iso8859-9')

                    desc_len = ((packet[17+subt_len] & 0xf) << 8) | packet[17+subt_len+1]

                    if not (len(packet) < (30+subt_len+desc_len)) :

                        desc_text = packet[19+subt_len:19+subt_len+desc_len].decode('iso8859-9')

                        title_id = (packet[18+subt_len+desc_len+9] << 24) | (packet[18+subt_len+desc_len+10] << 16) | (packet[18+subt_len+desc_len+11] << 8) | packet[18+subt_len+desc_len+12]

                        if summary_id not in dict_des :

                            dict_des[summary_id] = {}

                        dict_des[summary_id]['title_id'] = title_id

                        dict_des[summary_id]['subtitle'] = subt_text

                        dict_des[summary_id]['summary'] = desc_text

        packet = sys.stdin.buffer.read(3)

    else :

        break

#Parse data together

for key in dict_des.keys() :

    for prog in dict_pr.copy() :

        if (dict_des[key]['title_id'] == prog) or (key == dict_pr[prog]['desc_id']) :

            dict_output[prog] = dict_pr[prog]

            dict_output[prog]['channel'] = dict_chs[dict_pr[prog]['channel']]

            dict_output[prog]['subtitle'] = dict_des[key]['subtitle']

            dict_output[prog]['summary'] = dict_des[key]['summary']

            dict_pr.pop(prog)

            dict_output[prog].pop('desc_id')

for prog in dict_pr.copy() :

    dict_output[prog] = dict_pr[prog]

    dict_output[prog]['channel'] = dict_chs[dict_pr[prog]['channel']]

    dict_pr.pop(prog)

print(json.dumps(dict_output, ensure_ascii=False))