예제 #1
0
def parse_functions(code):
    """
    Parses code and returns a list of functions.

    Usage:
        For this function to work properly
        input code needs to be prepared first,
        function prepare_for_parsing(code)
        has to be called to pre-process the code first.

    Input arguments:
        code -- code from which we are parsing functions

    Output:
        List of functions.

        Each function has:
            type -- type of function (located from beginnig to ' ')
            name -- name of function (located between ' ' and '(')
            arguments -- list of arguments (located between '(' and ')'
                                            and separated by comma)

            Each argument has:
                type -- type of the argument
                name -- name of the argument
                array -- array lenght of argument (optional)

    Example:
        >>> code = '''actarray_t* actarray_init(
        ...                           file_t *file,
        ...                           int      nb_actions);
        ...
        ...           actarray_t* mixed_actarray_init(
        ...                           file_t **files,
        ...                           int nb_actions);
        ...
        ...           typedef enum mode {
        ...                     NEVER = 0,
        ...                     ALWAYS = 2
        ...           } mode_t;'''
        >>> code = prepare_for_parsing(code)
        >>> print code
        actarray_t* actarray_init(file_t* file,int nb_actions);
        actarray_t* mixed_actarray_init(file_t** files,int nb_actions);
        typedef enum mode {NEVER = 0,ALWAYS = 2}mode_t;
        >>> print parse_functions(code)
        [{'type': 'actarray_t*', 'name': 'actarray_init', 'arguments':
        [{'type': 'file_t*', 'name': 'file'}, {'type': 'int', 'name': '
        nb_actions'}]}, {'type': 'actarray_t*', 'name': 'mixed_actarray
        _init', 'arguments': [{'type': 'file_t**', 'name': 'files'}, {'
        type': 'int', 'name': 'nb_actions}]}]
    """
    def parse_arguments(arguments):
        """
        Parses arguments and returns a list of arguments

        Input arguments:
            arguments -- arguments which we are parsing

        Output:
            A list of arguments.

            Each argument has:
                type -- type of the argument
                name -- name of the argument
                array -- array lenght of argument (optional)

        Example:
            >>> arguments = 'file_t* file,int nb_actions'
            >>> print parse_element(element)
            [{'type': 'file_t*', 'name': 'file'}, {'type': 'int', 'name
            ': 'nb_actions'}]
        """
        return [({
            'type': argument[:argument.rfind(' ')],
            'name': argument[argument.rfind(' ') + 1:]
        } if ' ' in argument else {
            'type': argument
        }) if '[' not in argument else {
            'type':
            argument[:argument[:argument.find('[')].rfind(' ')],
            'name':
            argument[argument[:argument.find('[')].rfind(' ') +
                     1:argument.find('[')],
            'array':
            argument[argument.find('[') + 1:argument.find(']')]
        } for argument in arguments.split(',') if argument != '']

    # remove same lines from code
    lines = remove_duplicates(code.splitlines())
    return [
        {
            'type': line[:line[:line.find('(')].rfind(' ')],
            'name': line[line[:line.find('(')].rfind(' ') + 1:line.find('(')],
            'arguments':
            parse_arguments(line[line.find('(') + 1:line.find(')')])
        } for line in lines if '(' in line  #
        if 'typedef' not in line  # line is a function
        if '{' not in line  #
        if 'INIT_NAME' not in line  # INIT_NAME     not supported  TODO
        if 'bindresvport6' not in line  # bindresvport6 not supported  TODO
        if 'cb_func' not in line  # cb_func       not supported  TODO
        if '*__func' not in line  # *__func       not supported  TODO
        if '_IO' not in line  # _IO           not supported  TODO
        if '...' not in line
    ]  # ...           not supported  TODO
예제 #2
0
 def test_remove_duplicates_4(self):
     """remove_duplicates Unit test 4: Removing mixed duplicates"""
     input4 = ['a', 'a', '1', '1']
     output4 = ['a', '1']
     self.assertEqual(remove_duplicates(input4), output4)
예제 #3
0
 def test_remove_duplicates_3(self):
     """remove_duplicates Unit test 3: No duplicates"""
     input3 = ['a', 'b', 'c']
     output3 = ['a', 'b', 'c']
     self.assertEqual(remove_duplicates(input3), output3)
예제 #4
0
 def test_remove_duplicates_2(self):
     """remove_duplicates Unit test 2: Removing string duplicates"""
     input2 = ['a', 'b', 'c', 'c']
     output2 = ['a', 'b', 'c']
     self.assertEqual(remove_duplicates(input2), output2)
예제 #5
0
 def test_remove_duplicates_1(self):
     """remove_duplicates Unit test 1: Removing int duplicates"""
     input1 = [1, 2, 2, 6, 4, 6, 1]
     output1 = [1, 2, 6, 4]
     self.assertEqual(remove_duplicates(input1), output1)
예제 #6
0
파일: parse.py 프로젝트: maxeler/maxskins
def parse_functions(code):
    """
    Parses code and returns a list of functions.

    Usage:
        For this function to work properly
        input code needs to be prepared first,
        function prepare_for_parsing(code)
        has to be called to pre-process the code first.

    Input arguments:
        code -- code from which we are parsing functions

    Output:
        List of functions.

        Each function has:
            type -- type of function (located from beginnig to ' ')
            name -- name of function (located between ' ' and '(')
            arguments -- list of arguments (located between '(' and ')'
                                            and separated by comma)

            Each argument has:
                type -- type of the argument
                name -- name of the argument
                array -- array lenght of argument (optional)

    Example:
        >>> code = '''actarray_t* actarray_init(
        ...                           file_t *file,
        ...                           int      nb_actions);
        ...
        ...           actarray_t* mixed_actarray_init(
        ...                           file_t **files,
        ...                           int nb_actions);
        ...
        ...           typedef enum mode {
        ...                     NEVER = 0,
        ...                     ALWAYS = 2
        ...           } mode_t;'''
        >>> code = prepare_for_parsing(code)
        >>> print code
        actarray_t* actarray_init(file_t* file,int nb_actions);
        actarray_t* mixed_actarray_init(file_t** files,int nb_actions);
        typedef enum mode {NEVER = 0,ALWAYS = 2}mode_t;
        >>> print parse_functions(code)
        [{'type': 'actarray_t*', 'name': 'actarray_init', 'arguments':
        [{'type': 'file_t*', 'name': 'file'}, {'type': 'int', 'name': '
        nb_actions'}]}, {'type': 'actarray_t*', 'name': 'mixed_actarray
        _init', 'arguments': [{'type': 'file_t**', 'name': 'files'}, {'
        type': 'int', 'name': 'nb_actions}]}]
    """
    def parse_arguments(arguments):
        """
        Parses arguments and returns a list of arguments

        Input arguments:
            arguments -- arguments which we are parsing

        Output:
            A list of arguments.

            Each argument has:
                type -- type of the argument
                name -- name of the argument
                array -- array lenght of argument (optional)

        Example:
            >>> arguments = 'file_t* file,int nb_actions'
            >>> print parse_element(element)
            [{'type': 'file_t*', 'name': 'file'}, {'type': 'int', 'name
            ': 'nb_actions'}]
        """
        return [({'type': argument[ : argument.rfind(' ')],
                  'name': argument[argument.rfind(' ') + 1 : ]}
                 if ' ' in argument
                 else {'type': argument})
                if '[' not in argument else
                {'type': argument[ : argument[:argument.find('[')].rfind(' ')],
                 'name': argument[argument[:argument.find('[')].rfind(' ') + 1:
                                  argument.find('[')],
                 'array': argument[argument.find('[') + 1 :
                                   argument.find(']')]}
                for argument
                in arguments.split(',')
                if argument != '']

    # remove same lines from code
    lines = remove_duplicates(code.splitlines())
    return [{'type': line[ : line[ : line.find('(')].rfind(' ')],
             'name': line[line[ : line.find('(')].rfind(' ') + 1
                          : line.find('(')],
             'arguments': parse_arguments(line[line.find('(') + 1 :
                                               line.find(')')])}
            for line
            in lines
            if '(' in line                 #
            if 'typedef' not in line       # line is a function
            if '{' not in line             #
            if 'INIT_NAME' not in line     # INIT_NAME     not supported  TODO
            if 'bindresvport6' not in line # bindresvport6 not supported  TODO
            if 'cb_func' not in line       # cb_func       not supported  TODO
            if '*__func' not in line       # *__func       not supported  TODO
            if '_IO' not in line           # _IO           not supported  TODO
            if '...' not in line]          # ...           not supported  TODO