def get_labels()

in bugbug/models/needsdiagnosis.py [0:0]

• 24 lines of code
• 14 McCabe index (conditional complexity)

    def get_labels(self):
        classes = {}

        for issue in self.github.get_issues():
            # Skip issues with empty title or body
            if issue["title"] is None or issue["body"] is None:
                continue

            # Skip issues that are not moderated yet as they don't have a meaningful title or body
            if issue["title"] == "In the moderation queue.":
                continue

            for event in issue["events"]:
                if event["event"] == "milestoned" and (
                    event["milestone"]["title"] == "needsdiagnosis"
                    or event["milestone"]["title"] == "moved"
                ):
                    classes[issue["number"]] = 0

            if issue["number"] not in classes:
                classes[issue["number"]] = 1

        logger.info(
            "%d issues have not been moved to needsdiagnosis",
            sum(label == 1 for label in classes.values()),
        )
        logger.info(
            "%d issues have been moved to needsdiagnosis",
            sum(label == 0 for label in classes.values()),
        )

        return classes, [0, 1]