Github oauth user

Author: Shobhit

app/migrations/0013_alter_monitoredpullrequest_pull_request_status_and_more.py

@@ -0,0 +1,30 @@
+# Generated by Django 4.0.3 on 2022-04-12 06:44
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('app', '0012_alter_githubpullrequest_pr_closed_at_and_more'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='monitoredpullrequest',
+            name='pull_request_status',
+            field=models.CharField(choices=[('NOT_CONNECTED', 'Not Connected'), ('APPROVAL_PENDING', 'Approval Pending'), ('STALE_CODE', 'Stale Code'), ('STALE_APPROVAL', 'Stale Approval'), ('APPROVED', 'Approved'), ('MANUALLY_APPROVED', 'Manual Approval')], default='NOT_CONNECTED', max_length=20),
+        ),
+        migrations.CreateModel(
+            name='GithubAppUser',
+            fields=[
+                ('id', models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('github_user', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='app.githubuser')),
+                ('installation', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to='app.githubappinstallation')),
+            ],
+            options={
+                'unique_together': {('installation', 'github_user')},
+            },
+        ),
+    ]

app/models.py

@@ -161,6 +161,14 @@ def update_token(self):
         )[1]
 
 
+class GithubAppUser(models.Model):
+    installation = models.ForeignKey(GithubAppInstallation, on_delete=models.CASCADE)
+    github_user = models.ForeignKey(GithubUser, on_delete=models.CASCADE)
+
+    class Meta:
+        unique_together = ("installation", "github_user")
+
+
 class GithubInstallationToken(Token):
     github_app_installation = models.ForeignKey(
         GithubAppInstallation, on_delete=models.CASCADE, related_name="tokens"

app/signals.py

@@ -134,10 +134,10 @@ def synchronize_github_check(sender, instance, **kwargs):
                 },
             }
         )
-    elif (
-        instance.ref_pull_request.pull_request_status
-        == app_models.MonitoredPullRequest.PullRequestStatus.APPROVED
-    ):
+    elif instance.ref_pull_request.pull_request_status in [
+        app_models.MonitoredPullRequest.PullRequestStatus.APPROVED,
+        app_models.MonitoredPullRequest.PullRequestStatus.MANUAL_APPROVAL,
+    ]:
         data.update(
             {
                 "conclusion": "success",

app/views.py

@@ -12,7 +12,7 @@
 from django.conf import settings
 from django.contrib.auth import get_user_model, login
 from django.db import transaction
-from django.http import HttpResponseRedirect, JsonResponse
+from django.http import Http404, HttpResponseRedirect, JsonResponse
 from django.shortcuts import render
 from django.utils import timezone
 from django.utils.decorators import method_decorator
@@ -23,6 +23,7 @@
 from django.views.generic import FormView, TemplateView
 from django.contrib.auth import views as auth_views
 from requests.models import PreparedRequest
+from django.shortcuts import get_object_or_404
 
 from . import models as app_models
 from . import forms as app_forms
@@ -144,6 +145,18 @@ def get(self, request, *args, **kwargs):
                         )
                         # installation_instance.save()
                         installation_instance.update_token()
+                for installation in user_instance.get_installations():
+                    if installation.app_id == settings.GITHUB_CREDS["app_id"]:
+                        installation_instance = (
+                            app_models.GithubAppInstallation.objects.get(
+                                account_id=installation.target_id,
+                                installation_id=installation.id,
+                            )
+                        )
+                        app_models.GithubAppUser.objects.update_or_create(
+                            github_user=github_user,
+                            installation=installation_instance,
+                        )
         else:
             logger.error(resp.text)
         return HttpResponseRedirect("/admin/")
@@ -257,6 +270,21 @@ def get(self, request):
 class AllPRView(TemplateView):
     template_name = "index.html"
 
+    def get(self, request, *args: Any, **kwargs: Any):
+        github_user = request.user.github_user
+        context = self.get_context_data(**kwargs)
+
+        if github_user is None:
+            return Http404("User not found")
+        # elif app_models.GithubAppUser.objects.filter():
+        #     pass
+        get_object_or_404(
+            app_models.GithubAppUser,
+            github_user=github_user,
+            installation=context["repo_mapping"].integration,
+        )
+        return self.render_to_response(context)
+
     def get_context_data(self, **kwargs: Any) -> Dict[str, Any]:
         context = super().get_context_data(**kwargs)
         matches = self.request.resolver_match.kwargs