xmltools/timeshift.py at main · broken-droid/xmltools · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from xmlmerge import open_xml, write_xml, read_yaml_input
from lxml import etree
from datetime import datetime, timedelta
# Pull out channels/programs with matching ids from an xml file and timeshift the
# specified hours. Write those channels/programs to 'output' with altered name and id.
# needs xmlmerge.py for open_xml, write_xml, read_yaml_input

# timeshift.yaml for input
cache_path = 'cache/' # path to input files, edit this
output_path = 'cache/' # path to output, I want to merge this with other files
output = f'{output_path}shift.xml' # file to write, edit this
output_root = etree.Element('tv') # root node for output
output_programs = [] # hold program elements to write
output_channels = [] # hold channel elements to write
channels = {} # hold input data from yaml

def timeshift():
    global channels, cache_path
    channels = read_yaml_input('timeshift.yaml')['channels']
    for fname, data in channels.items():
        process_file(fname, data)
    finish_and_write()

def modify_programs(id, plist, new_id, shift_hours):
    for program in plist:
        start = program.get('start')
        stop = program.get('stop')
        start = datetime.strptime(start, '%Y%m%d%H%M%S %z')
        stop = datetime.strptime(stop, '%Y%m%d%H%M%S %z')
        start = start + timedelta(hours=shift_hours)
        stop = stop + timedelta(hours=shift_hours)
        start = start.strftime('%Y%m%d%H%M%S %z')
        stop = stop.strftime('%Y%m%d%H%M%S %z')
        program.set('start', start)
        program.set('stop', stop)
        program.set('channel', new_id)
    return plist

def finish_and_write():
    global output_root, output_programs, output
    for channel in output_channels:
        output_root.append(channel)
    for prog_list in output_programs:
        output_root.extend(prog_list)
    write_xml(output, False, output_root)

def process_file(input, data):
    global output_root, output_channels, output_programs, cache_path
    input_root = open_xml(input, cache_path)
    for channel in data:
        channel_name = channel[0]
        channel_id = channel[1]
        new_id = channel[2]
        channel_shift = int(channel[3])
        channel_el = input_root.find(f'.//channel[@id="{channel_id}"]')
        channel_el.set('id', new_id)
        disp_name = channel_el.find('display-name')
        disp_name.text = channel_name
        icons = channel_el.findall('icon')
        if len(icons) > 1:
            channel_el.remove(icons[1])
        output_channels.append(channel_el)
        input_programs = input_root.findall(f'.//programme[@channel="{channel_id}"]')
        modify_programs(channel_id, input_programs, new_id, channel_shift)
        output_programs.append(input_programs)

if __name__ == '__main__':
    timeshift()