Exemplo n.º 1
0
def colourify_content(content, root="", target=False):
    """Colourify content name using LS_COLORS.
    If target is True and content is a symbolic link,
    colourify content.target instead of content.name."""

    content_name = rootjoin(content.location_key().parse_value(), root)

    codes, special_codes = parse_ls_colours()
    if not codes and not special_codes:
        return content_name

    if isinstance(content, ContentsDirEntry):
        return "\033[" + special_codes.get("di", "00") + "m" + content_name + "\033[m"
    if isinstance(content, ContentsFileEntry):
        return colourify_file(content_name, codes, special_codes)
    elif isinstance(content, ContentsSymEntry):
        if not target:
            return "\033[" + special_codes.get("ln", "00") + "m" + content_name + "\033[m"
        elif os.path.isabs(content.target_key().parse_value()):
            content_target = rootjoin(content.target_key().parse_value(), root)
            return colourify_file(content_target, codes, special_codes)
        else:
            dname = os.path.dirname(content_name)
            abstarget = rootjoin(content.target_key().parse_value(), dname)
            return colourify_file(abstarget, codes, special_codes).replace(
                        dname + os.path.sep, '')
    else:
        return content_name
Exemplo n.º 2
0
def no_colourify_content(content, root="", target=False):
    """Dummy replacement for colourify_content() with no colouring."""
    content_name = rootjoin(content.location_key().parse_value(), root)
    if target:
        if os.path.isabs(content.target_key().parse_value()):
            return rootjoin(content.target_key().parse_value(), root)
        else:
            dname = os.path.dirname(content_name)
            return rootjoin(content.target_key().parse_value(), dname).replace(
                    dname + os.path.sep, '')
    else:
        return content_name
Exemplo n.º 3
0
def search_contents(path, env, matcher="exact", ignore_case=False, requested_instances=[object]):  # {{{
    """Search filename in contents of installed packages."""

    # Get package ids of all installed packages
    ids = env[Selection.AllVersionsGroupedBySlot(Generator.Matches.All() | Filter.InstalledAtRoot(env.root))]

    if matcher == "fnmatch":
        if ignore_case:
            fnmatch_matcher = fnmatch.fnmatchcase
        else:
            fnmatch_matcher = fnmatch.fnmatch
    elif matcher == "regex":
        if ignore_case:
            regex_matcher = re.compile(path, re.IGNORECASE)
        else:
            regex_matcher = re.compile(path)

    for package_id in ids:
        if package_id.contents_key() is None:
            Log.instance.message(
                "vdb.no_contents",
                LogLevel.WARNING,
                LogContext.NO_CONTEXT,
                "'%s' does not provide a contents key." % package_id.name,
            )
        else:
            for content in package_id.contents_key().parse_value():
                if not True in [isinstance(content, i) for i in requested_instances]:
                    continue

                content_path = rootjoin(content.location_key().parse_value(), env.root)

                if matcher == "exact" and (path == content_path or path == os.path.basename(content_path)):
                    yield package_id, content
                elif matcher == "simple" and path in content_path:
                    yield package_id, content
                elif matcher == "fnmatch" and fnmatch_matcher(content_path, path):
                    yield package_id, content
                elif matcher == "regex" and regex_matcher.search(content_path):
                    yield package_id, content
Exemplo n.º 4
0
def get_contents(
    package,
    env,
    source_repos=[],
    requested_instances=[object],
    selection=Selection.AllVersionsGroupedBySlot,
    fnpattern=None,
    regexp=None,
    ignore_case=False,
):
    """Get contents of package"""

    # {{{Pattern matching
    if regexp is not None:
        if ignore_case:
            pattern = re.compile(regexp, re.IGNORECASE)
        else:
            pattern = re.compile(regexp)
    # }}}

    # {{{Get PackageDepSpec
    filter_installed = Filter.InstalledAtRoot(env.system_root_key)
    allow_wildcards = UserPackageDepSpecOption.ALLOW_WILDCARDS
    package_dep_spec = parse_user_package_dep_spec(package, env, [allow_wildcards], filter_installed)
    # }}}

    # {{{Get CONTENTS
    ids = env[selection(Generator.Matches(package_dep_spec, MatchPackageOptions()) | filter_installed)]

    for package_id in ids:
        if package_id.contents_key() is None:
            Log.instance.message(
                "vdb.no_contents",
                LogLevel.WARNING,
                LogContext.NO_CONTEXT,
                "'%s' does not provide a contents key." % package_id.name,
            )
        else:
            # {{{Match by source repository
            if source_repos:
                if package_id.from_repositories_key() is None:
                    Log.instance.message(
                        "vdb.no_source_origin",
                        LogLevel.WARNING,
                        LogContext.NO_CONTEXT,
                        "'%s' does not provide a from repositories key." % package_id.name,
                    )
                else:
                    repo_origin = package_id.from_repositories_key()

                    if not any(repo in source_repos for repo in repo_origin.parse_value()):
                        continue
            # }}}

            requested_contents = list()
            for content in package_id.contents_key().parse_value():
                if not any([isinstance(content, i) for i in requested_instances]):
                    continue

                content_path = rootjoin(content.location_key().parse_value(), env.root)

                if fnpattern is not None:
                    if ignore_case:
                        if not fnmatch.fnmatchcase(content_path, fnpattern):
                            continue
                    else:
                        if not fnmatch.fnmatch(content_path, fnpattern):
                            continue
                if regexp is not None and pattern.search(content_path) is None:
                    continue
                requested_contents.append(content)

            yield package_id, requested_contents