generators.py

#!/usr/bin/env python2.7
# -*- coding: utf-8 -*-

# Generators module
# Copyright (C) 2014 Musikhin Andrey <melomansegfault@gmail.com>

import os
from subprocess import Popen, PIPE
from pprint import pformat



class CmdDictGenerator(object):
    """ Command list dictionary generator  """

    def __init__(self, mplayer='mplayer'):
        self._cmdlist = [line for line in Popen(
                      [mplayer, '-input', 'cmdlist'], stdout=PIPE).stdout][:-1]
        self._doc_file = open(
                       os.path.join(os.path.dirname(os.path.abspath(__file__)), 
                       'data/mplayer_cmdlist_docs'), 'r').readlines()
        self._cmdlist_dict = {}

    def _get_cmd_dict(self):
        """ Convert txt file to python dictionary """
        # Creating the comments dictionary
        doc_string = 'MPlayer doc:\n'
        doc_key = ''
        doc_dict = {}
        for line in self._doc_file:
            if line == '\n':
                doc_dict[doc_key] = doc_string
                doc_string = 'MPlayer doc: '
                continue
            if line[0] == ' ':
                doc_string += line[4:]
            else: 
                doc_key = line.strip().split(' ')[0]
                doc_string += line
        # Creating the command dictionary
        cmds_dict = {}
        for line in self._cmdlist:
            line_dict = {}
            list_line = line.split()
            line_dict['command'] = list_line[0]
            if list_line[0] not in doc_dict.keys():
                line_dict['comment'] = self._new_cpt(line)[0]
                line_dict['pycommand'] = self._new_cpt(line)[1]
                line_dict['types'] = self._new_cpt(line)[2]
            else:
                line_dict['comment'] = '%s\n%s' % (self._new_cpt(line)[0],
                                                   doc_dict[list_line[0]])
                line_dict['pycommand'] = self._new_cpt(line)[1]
                line_dict['types'] = self._new_cpt(line)[2]
            cmds_dict[list_line[0]] = line_dict
        self._cmdlist_dict = cmds_dict

    def _new_cpt(self, comment):
        """ 
        Return mplayer command string, python method string and types list
        """
        comment_list =  comment.split()
        _comment = ' '.join(comment_list)
        types = []
        _python_comment = ''
        for num, item in enumerate(comment_list[1:]):
            if 'Integer' in item:
                string = 'int'
            elif 'String' in item:
                string = 'str'
            elif 'Float' in item:
                string = 'float'
            else: pass
            types.append(string)
            if '[' in item and num!=0: 
                string = '[, <%s>]' % string
            elif '[' in item and num==0: 
                string = '[<%s>]' % string
            elif num != 0: string  = ', <%s>' % string
            elif num == 0: string = '<%s>' % string 
            _python_comment += string
        command_comment = 'MPlayer command: %s' % _comment
        python_comment = '%s(%s)' % (comment_list[0], _python_comment)
        return [command_comment, python_comment, types]

    def get_cmdlist(self):
        self._get_cmd_dict()
        return self._cmdlist_dict



class PropDictGenerator(object):
    '''
    Generates a dictionary of MPlayer properties
    '''
    def __init__(self):
        self.txt_file = open(
                       os.path.join(os.path.dirname(os.path.abspath(__file__)),
                       'data/mplayer_properties'), 'r').readlines()

    def _get_prop_dict(self):
        '''
        Take and parse method. Return the properties dictionary
        '''
        def convert_min_max(string):
            # Converting max/min values
            try:
                if '.' in string:
                    value = float(string)
                    return value
                else:
                    value = int(string)
                    return value
            except: return False

        def convert_type(string):
            # Converting type string
            if 'pos' in string:
                string = 'int'
            if 'time' in string:
                string = 'float'
            if 'str list' in string:
                string = 'str'
            if 'string' in string:
                string = 'str'
            return string

        property_dict = {}
        for line in self.txt_file:
            command = line[:19].strip()
            if command:
                prop_dict = {
                    'command':  command,
                    'type':     convert_type(line[19:29].strip()),
                    'min':      convert_min_max(line[29:37].strip()),
                    'max':      convert_min_max(line[37:45].strip()),
                    'get':      line[45:49].strip()=='X',
                    'set':      line[49:53].strip()=='X',
                    'step':     line[53:58].strip()=='X',
                    'comment':  line[58:].strip(),
                }
            else:
                command = prop_dict['command']
                prop_dict['comment']+= '\n' + line[58:].strip()
            property_dict[command] = prop_dict
        print property_dict
        return property_dict

    def run(self):
        '''
        Run generator method
        '''
        output_prop_file = open(
                       os.path.join(os.path.dirname(os.path.abspath(__file__)),
                       'player_properties.py'), 'w')
        text = 'property_dict = %s'
        output_prop_file.write(text % pformat(self._get_prop_dict()))
        output_prop_file.close()



if __name__ == '__main__':
    DG = CmdDictGenerator()
    print pformat(DG.get_cmdlist())
    PG = PropDictGenerator()
    PG.run()