Change Stars Example Solution

#!/usr/bin/python2

'''
Proof of concept. Change the stars rating for a single recording

Save this in /usr/local/bin/change_stars.py. chmod 755 on that filename .

First:
    change_stars.py --help

List all recordings (to find the recordedid in the 4th column):
    change_stars.py

Then:
    change_stars.py --recordedid=12345 --stars=6

To print three specific titles, using regular expressions/or not:
    change_stars.py --titles 'madam sec' '^ncis$' element
'''

from __future__ import print_function
import argparse
import re
import sre_constants
import sys
from MythTV import (exceptions, MythBE, Recorded)

BEG = '\n\033[92m'
END = '\033[0m'


def main():
    '''
    Old tool modified to print the starttime, title, subtitle,
    recordedid and stars. Optionally, allow changing a single
    recording's stars value.
    '''

    choices = range(11)

    parser = argparse.ArgumentParser(description='MythTV "Stars" tool',
                                     epilog='No arguments = all recordings')

    parser.add_argument('--recordedid', type=int, metavar='<n>',
                        help='value that identifies a single recording')

    parser.add_argument('--titles', type=str, nargs='+', metavar='<title>',
                        help='one or more titles, regular expressions ' +
                        'are allowed')

    parser.add_argument('--stars', type=int, metavar='<n>',
                        choices=choices,
                        help='the number of stars {}'.format(choices))

    args = parser.parse_args()

    if (args.recordedid and (args.stars is None)) or \
            ((args.stars) and args.recordedid is None):
        sys.exit('\n--recordedid and --stars must both be used to change data')

    recs = Recorded.getAllEntries()
    sorted_recordings = sorted(recs, key=lambda recs: recs.starttime)

    title_re_lst = list()

    if args.titles:
        for one_title in args.titles:
            try:
                title_re_lst.append(re.compile(one_title, flags=re.IGNORECASE))
            except sre_constants.error:
                sys.exit('\nCan\'t compile regular expression "{}", maybe ".*?'
                         .format(one_title))

    for recording in sorted_recordings:

        title_match = False

        for title_re in title_re_lst:
            if args.titles and not title_re.search(recording.title):
                continue
            else:
                title_match = True

        if title_match or (args.titles is None):
            if args.recordedid and (args.stars is not None):
                db_stars_value = float(args.stars) / 10
                if recording.recordedid == args.recordedid:
                    recording.update({'stars': db_stars_value})
                    print('\n"{}" stars value set to {}\n'
                          .format(recording.title, args.stars))
                    break

            else:
                print('  {}   {:<30.30}\t{:<16.16}  {:06} {}'
                      .format(recording.starttime.strftime("%b %d %H:%M"),
                              recording.title.encode('utf-8'),
                              recording.subtitle.encode('utf-8'),
                              recording.recordedid,
                              int(recording.stars * 10)))


if __name__ == '__main__':

    try:
        MythBE()
    except exceptions.MythError:
        sys.exit('Unable to connect to the backend.')

    main()

# vim: set expandtab tabstop=4 shiftwidth=4 smartindent colorcolumn=80: