Class util

APKDeepLens.py:24–123

A static class for which contain some useful variables and methods

class util(util):
    """
    A static class for which contain some useful variables and methods
    """

    @staticmethod
    def mod_print(text_output, color):
        """
        Better mod print. It gives the line number, file name in which error occured.
        """
        stack = traceback.extract_stack()
        filename, line_no, func_name, text = stack[-2]
        formatted_message = f"{filename}:{line_no}: {text_output}"
        print(color + formatted_message + util.ENDC)

    @staticmethod
    def print_summary(results_dict):
        """Print a colour-coded summary table after scanning completes."""
        SEVS      = ["CRITICAL", "HIGH", "MEDIUM", "LOW", "INFO"]
        SEV_COLOR = {
            "CRITICAL": util.FAIL,
            "HIGH":     util.WARNING,
            "MEDIUM":   util.OKCYAN,
            "LOW":      "",
            "INFO":     "",
        }

        def count_by_sev(findings):
            counts = {s: 0 for s in SEVS}
            for f in (findings or []):
                sev = f.get("severity", "INFO")
                if sev in counts:
                    counts[sev] += 1
            return counts

        manifest_counts = count_by_sev(results_dict.get("manifest_security", []))
        code_counts     = count_by_sev(results_dict.get("code_findings", []))
        n_secrets  = len(results_dict.get("hardcoded_secrets") or [])
        n_insecure = len(results_dict.get("insecure_requests") or [])
        n_danger   = len(results_dict.get("dangerous_permission") or [])

        cat_w    = 30
        col_keys = ["CRITICAL", "HIGH", "MEDIUM", "LOW", "INFO", "TOTAL"]
        col_w    = {"CRITICAL": 10, "HIGH": 6, "MEDIUM": 8, "LOW": 5, "INFO": 6, "TOTAL": 7}

        sep = (
            "+" + "-" * (cat_w + 2) + "+"
            + "+".join("-" * (col_w[k] + 2) for k in col_keys)
            + "+"
        )

        def fmt(val, w, color=""):
            s = str(val).center(w)
            return (" " + color + s + util.ENDC + " ") if color else (" " + s + " ")

        def header_row():
            cells = [fmt(k, col_w[k]) for k in col_keys]
            return "| " + "Category".ljust(cat_w) + " |" + "|".join(cells) + "|"