update code

Author: mrkeshav-05

backend/Makefile

@@ -156,6 +156,7 @@ update-data: \
 	github-add-related-repositories \
 	github-update-related-organizations \
 	github-update-users \
+	github-aggregate-contributions \
 	owasp-aggregate-projects \
 	owasp-aggregate-contributions \
 	owasp-update-events \

backend/apps/github/Makefile

@@ -21,3 +21,7 @@ github-update-related-organizations:
 github-update-users:
 	@echo "Updating GitHub users"
 	@CMD="python manage.py github_update_users" $(MAKE) exec-backend-command
+
+github-aggregate-contributions:
+	@echo "Aggregating GitHub user contributions"
+	@CMD="python manage.py github_aggregate_contributions" $(MAKE) exec-backend-command

backend/apps/github/api/internal/nodes/user.py

@@ -13,6 +13,7 @@
         "avatar_url",
         "bio",
         "company",
+        "contribution_data",
         "contributions_count",
         "email",
         "followers_count",

backend/apps/github/management/commands/github_aggregate_contributions.py

@@ -0,0 +1,141 @@
+"""Management command to aggregate user contribution data."""
+
+from datetime import datetime, timedelta
+from typing import Any
+
+from django.core.management.base import BaseCommand
+from django.db.models import Count
+from django.utils import timezone
+
+from apps.github.models.commit import Commit
+from apps.github.models.issue import Issue
+from apps.github.models.pull_request import PullRequest
+from apps.github.models.user import User
+
+
+class Command(BaseCommand):
+    """Aggregate contribution data for users."""
+
+    help = "Aggregate contribution data (commits, PRs, issues) for users"
+
+    def add_arguments(self, parser):
+        """Add command arguments."""
+        parser.add_argument(
+            "--user",
+            type=str,
+            help="Specific user login to process",
+        )
+        parser.add_argument(
+            "--days",
+            type=int,
+            default=365,
+            help="Number of days to look back (default: 365)",
+        )
+        parser.add_argument(
+            "--batch-size",
+            type=int,
+            default=100,
+            help="Batch size for processing users (default: 100)",
+        )
+
+    def handle(self, *args: Any, **options: Any) -> None:
+        """Handle the command execution."""
+        user_login = options.get("user")
+        days = options.get("days", 365)
+        batch_size = options.get("batch_size", 1000)
+
+        start_date = timezone.now() - timedelta(days=days)
+
+        self.stdout.write(
+            self.style.SUCCESS(
+                f"Aggregating contributions since {start_date.date()} ({days} days back)"
+            )
+        )
+
+        if user_login:
+            users = User.objects.filter(login=user_login)
+            if not users.exists():
+                self.stdout.write(self.style.ERROR(f"User '{user_login}' not found"))
+                return
+        else:
+            users = User.objects.filter(contributions_count__gt=0)
+
+        total_users = users.count()
+        self.stdout.write(f"Processing {total_users} users...")
+
+        processed = 0
+        for user in users.iterator(chunk_size=batch_size):
+            contribution_data = self._aggregate_user_contributions(user, start_date)
+
+            if contribution_data:
+                user.contribution_data = contribution_data
+                user.save(update_fields=["contribution_data"])
+                processed += 1
+
+                if processed % 100 == 0:
+                    self.stdout.write(f"Processed {processed}/{total_users} users...")
+
+        self.stdout.write(
+            self.style.SUCCESS(f"Successfully aggregated contributions for {processed} users")
+        )
+
+    def _aggregate_user_contributions(self, user: User, start_date: datetime) -> dict[str, int]:
+        """Aggregate contributions for a user.
+
+        Args:
+            user: User instance
+            start_date: Start datetime for aggregation
+
+        Returns:
+            Dictionary mapping YYYY-MM-DD to contribution counts
+
+        """
+        contribution_data = {}
+        current_date = start_date.date()
+        end_date = timezone.now().date()
+
+        while current_date <= end_date:
+            date_str = current_date.strftime("%Y-%m-%d")
+            contribution_data[date_str] = 0
+            current_date += timedelta(days=1)
+
+        commits = (
+            Commit.objects.filter(
+                author=user,
+                created_at__gte=start_date,
+            )
+            .values("created_at")
+            .annotate(count=Count("id"))
+        )
+
+        for commit in commits:
+            date_str = commit["created_at"].strftime("%Y-%m-%d")
+            contribution_data[date_str] = contribution_data.get(date_str, 0) + commit["count"]
+
+        prs = (
+            PullRequest.objects.filter(
+                author=user,
+                created_at__gte=start_date,
+            )
+            .values("created_at")
+            .annotate(count=Count("id"))
+        )
+
+        for pr in prs:
+            date_str = pr["created_at"].strftime("%Y-%m-%d")
+            contribution_data[date_str] = contribution_data.get(date_str, 0) + pr["count"]
+
+        issues = (
+            Issue.objects.filter(
+                author=user,
+                created_at__gte=start_date,
+            )
+            .values("created_at")
+            .annotate(count=Count("id"))
+        )
+
+        for issue in issues:
+            date_str = issue["created_at"].strftime("%Y-%m-%d")
+            contribution_data[date_str] = contribution_data.get(date_str, 0) + issue["count"]
+
+        return contribution_data

backend/apps/github/migrations/0041_user_contribution_data.py

@@ -0,0 +1,22 @@
+# Generated by Django 6.0.1 on 2026-01-12 19:20
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("github", "0040_merge_20251117_0136"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="user",
+            name="contribution_data",
+            field=models.JSONField(
+                blank=True,
+                default=dict,
+                help_text="Aggregated contribution data as date -> count mapping",
+                verbose_name="Contribution heatmap data",
+            ),
+        ),
+    ]

backend/apps/github/models/user.py

@@ -54,6 +54,13 @@ class Meta:
         verbose_name="Contributions count", default=0
     )
 
+    contribution_data = models.JSONField(
+        verbose_name="Contribution heatmap data",
+        default=dict,
+        blank=True,
+        help_text="Aggregated contribution data as date -> count mapping",
+    )
+
     def __str__(self) -> str:
         """Return a human-readable representation of the user.
 

backend/tests/apps/github/api/internal/nodes/user_test.py

@@ -23,6 +23,7 @@ def test_meta_configuration(self):
             "badges",
             "bio",
             "company",
+            "contribution_data",
             "contributions_count",
             "created_at",
             "email",
@@ -308,3 +309,30 @@ def test_linkedin_page_id_without_profile(self):
         result = UserNode.linkedin_page_id(mock_user)
 
         assert result == ""
+
+    def test_contribution_data_field(self):
+        """Test contribution_data field returns stored data."""
+        mock_user = Mock()
+        mock_user.contribution_data = {
+            "2025-01-01": 5,
+            "2025-01-02": 3,
+            "2025-01-03": 0,
+        }
+
+        # Access the field directly since it's just a passthrough
+        result = mock_user.contribution_data
+
+        assert result == {
+            "2025-01-01": 5,
+            "2025-01-02": 3,
+            "2025-01-03": 0,
+        }
+
+    def test_contribution_data_field_empty(self):
+        """Test contribution_data field when empty."""
+        mock_user = Mock()
+        mock_user.contribution_data = {}
+
+        result = mock_user.contribution_data
+
+        assert result == {}