# Copyright (c) 2009 Nokia Corporation

#

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

#     http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

import os

excluded_modules = []

module_paths = []

pyd_path_name = []

def get_mod_list(arg, dirname, names):

    for name in names:

        # This check is done for not including the test_ files for modules

        if not name.startswith('test_') and name.endswith('.py'):

            path = os.path.join(dirname, name)

            path = path.replace(lib_path + '\\', '')

            # This check for 'test' is done to remove those files which fall

            # under test or crashers

            if not 'test' in path:

                module_paths.append(path)

def gen_excluded_list():

    global excluded_modules

    py_module = []

    module_type = []

    pyd_modules = []

    builtin_modules = []

    modules_cfg_path = \

            os.path.abspath('..\\newcore\\Symbian\\src\\modules.cfg')

    config_file = open(modules_cfg_path, 'r')

    for line in config_file:

        line = line.strip()

        if line in ['PYD', 'PY_MODULES', 'BUILTIN', 'EXT-MODULES',

                    'EXT-PYD', 'EXT-MOD-CFGS']:

            module_type = line

            continue

        if line and line[0] != '#':

            if module_type == 'PYD':

                pyd_modules.append(line.split(',')[0])

            elif module_type == 'BUILTIN':

                builtin_modules.append(line.split('=')[0])

            elif module_type == 'PY_MODULES':

                py_module.append(line.split(':')[1])

    excluded_modules = py_module + builtin_modules + pyd_modules

def get_pyd_list(arg, dirname, names):

    for name in names:

        if name.endswith('.c'):

            path = os.path.join(dirname, name)

            buf = open(path, 'r').read()

            if "Py_InitModule" in buf:

                pyd_path_name.append(path)

def add_pyd():

    pyd_name = []

    for pyd in pyd_path_name:

        fp = open(pyd, 'r')

        file_name = os.path.basename(pyd)

        for line in fp:

            if "Py_InitModule" in line:

                try:

                    pyd_name.append(line.split('"')[1])

                except:

                    # Some files do not have their pyd names in Py_InitModule,

                    # a split on these files would not yield the names so their

                    # names are hard coded

                    if file_name == 'unicodedata.c':

                        pyd_name.append('unicodedata')

                    elif file_name == 'posixmodule.c':

                        pyd_name.append('posix')

                    elif file_name == 'pyexpat.c':

                        pyd_name.append('pyexpat')

                    elif file_name == 'socketmodule.c':

                        pyd_name.append('_socket')

                    elif file_name == '_bsddb.c':

                        pyd_name.append('_bsddb')

                    continue

        fp.close()                             

    module_paths.extend(pyd_name)

def gen_mod_dep_cfg_template():

    # These are the module dependency map for modules that are not

    # found by the module_config_parser and hence they need an explicit mention

    additional_modules = {'__main__': {'type': 'base', 'deps': []},

                          '__builtin__': {'type': 'base', 'deps': []},

                          'sys': {'type': 'base', 'deps': []},

                          'nt': {'type': 'excluded', 'deps': []},

                          'win32evtlogutil':{'type': 'excluded', 'deps': []},

                          'win32evtlog': {'type': 'excluded', 'deps': []}}

    for module in excluded_modules:

        if module in module_paths:

            module_paths.remove(module)

    # The exclude py_files() are parsed here

    dep_map = {}

    mod_dep_cfg_template_path = \

    os.path.abspath(

                '..\\newcore\\Symbian\\src\\module_dependency.cfg.template')

    dep_map = eval(open(mod_dep_cfg_template_path, 'r').read())

    dep_map.update(additional_modules)

    mod_map = {}

    for mod in module_paths:

        mod_map['type'] = 'excluded'

        mod_map['deps'] = []

        mod_name = os.path.splitext(mod)[0].replace('\\', '.')

        dep_map[mod_name] = mod_map

    open(mod_dep_cfg_template_path, 'w').write(str(dep_map))

    fp = open(mod_dep_cfg_template_path, 'r')

    buf = fp.read()

    buf = buf.replace("}, '", "}, \n'")

    fp.close()

    open(mod_dep_cfg_template_path, 'w').write(buf)

if __name__ == "__main__":

    # Walk through the Lib path for inbuilt modules

    lib_path = os.path.abspath("..\\newcore\\Lib")

    os.path.walk(lib_path, get_mod_list, None)

    gen_excluded_list()

    os.path.walk(os.path.abspath("..\\newcore\\Modules"), get_pyd_list, None)

    add_pyd()

    gen_mod_dep_cfg_template()