summaryrefslogtreecommitdiffstats
path: root/python/mozbuild/mozbuild/pythonutil.py
blob: a3540647f94da079d96aad19f28ba395f61e8107 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.

import os
import sys


def iter_modules_in_path(*paths):
    paths = [os.path.abspath(os.path.normcase(p)) + os.sep for p in paths]
    for name, module in sys.modules.items():
        if getattr(module, "__file__", None) is None:
            continue
        if module.__file__ is None:
            continue
        path = module.__file__

        if path.endswith(".pyc"):
            path = path[:-1]
        path = os.path.abspath(os.path.normcase(path))

        if any(path.startswith(p) for p in paths):
            yield path