delete_duplicates.py

#!/usr/bin/python
# -*- coding: utf-8 -*-

from __future__ import absolute_import, division, print_function, unicode_literals
import cli
import elisaviihde
from user_input import get_input



def main():
    parser = cli.init_argparser()
    parser.add_argument('-f', '--folder_name')
    parser.add_argument('-m', '--max_auto_delete')
    params = parser.parse_args()

    username = cli.read_input(params.user, 'Elisa Viihde Username')
    password = cli.read_password(params.passfile, 'Elisa Viihde Password')

    e = elisaviihde.Elisaviihde()
    if not e.login(username, password):
        return
          
    folder_name = cli.read_input(params.folder_name, 'Look from folder (enter for all)')
    if folder_name:
        folder = e.find_folder_by_name(folder_name)
        if folder is None:
            print('Folder', folder_name, 'not found')
            return
    else:
        folder = e.find_folder_by_id(0)

    recordings = e.ls_recordings_recursive(folder, [])
    found_duplicates = find_duplicates(recordings)
    
    answer = 'n'
    if params.max_auto_delete is not None:
        if len(found_duplicates) <= int(params.max_auto_delete):
            answer = 'y'
    elif len(found_duplicates) > 0:
        answer = get_input("Enter 'y' to delete duplicates, anything else to cancel: ")
    
    if answer == 'y':
        for key, recordings in found_duplicates.items():
            print('Processing:', key[0], ';', key[1])
            first = True
            for recording in recordings:
                if first:
                    print('Kept first', recording['startTime'], recording['channel'])
                    first = False
                else:
                    status = e.delete(recording['programId'])
                    print('DELETED', status, recording['startTime'], recording['channel'])
        print('Delete duplicates done.')
    else:
        print('Delete duplicates canceled.')
        exit(-1)
        
    
def find_duplicates(recordings):
    # arrange programs as hastable {(name, description): [recording, recording], ...}
    recordings_dict = {}
    for recording in recordings:
        key = (recording['name'].strip(), recording.get('description', '').strip())
        value = recordings_dict.get(key, [])
        value.append(recording)
        recordings_dict[key] = value
    
    # hastable {(name, description): [recording, recording], ...}
    # where recordings ordered by not-HD, startTimeUTC timestamp
    duplicate_recordings_dict = {}
    
    for key, recordings in recordings_dict.items():
        if len(recordings) > 1:
            recordings.sort(key=lambda r: (not r['channel'].endswith('HD'), r['startTimeUTC']))
            duplicate_recordings_dict[key] = recordings
            print('Found duplicates for:', key[0], ';', key[1])
            for recording in recordings:
                print('*', recording['startTime'], recording['channel'])
    print('Total', len(duplicate_recordings_dict))
    return duplicate_recordings_dict


if __name__ == '__main__':
    main()