From 5d9d6eec3159a9c87ca45dc5b294534daf8495fa Mon Sep 17 00:00:00 2001
From: Leon Sandøy <leon.sandoy@conmodo.com>
Date: Fri, 19 Apr 2019 15:29:27 +0200
Subject: Addressing all comments in volcyy's second review. The tests now mock
 the API calls so we don't have to actually call the API every time we run
 tests.

---
 pydis_site/apps/home/views/home.py | 110 +++++++++++++++++++++++++++++++++++++
 1 file changed, 110 insertions(+)
 create mode 100644 pydis_site/apps/home/views/home.py

(limited to 'pydis_site/apps/home/views/home.py')

diff --git a/pydis_site/apps/home/views/home.py b/pydis_site/apps/home/views/home.py
new file mode 100644
index 00000000..c91d13e2
--- /dev/null
+++ b/pydis_site/apps/home/views/home.py
@@ -0,0 +1,110 @@
+import requests
+from django.shortcuts import render
+from django.utils import timezone
+from django.views import View
+
+from pydis_site.apps.home.models import RepositoryMetadata
+
+
+class HomeView(View):
+    """The view"""
+
+    github_api = "https://api.github.com/users/python-discord/repos"
+
+    # Which of our GitHub repos should be displayed on the front page, and in which order?
+    repos = [
+        "python-discord/site",
+        "python-discord/bot",
+        "python-discord/snekbox",
+        "python-discord/seasonalbot",
+        "python-discord/django-simple-bulma",
+        "python-discord/django-crispy-bulma",
+    ]
+
+    def _get_api_data(self):
+        """Call the GitHub API and get information about our repos."""
+
+        repo_dict = {repo_name: {} for repo_name in self.repos}
+
+        # Fetch the data from the GitHub API
+        api_data = requests.get(self.github_api)
+        api_data = api_data.json()
+
+        # Process the API data into our dict
+        for repo in api_data:
+            full_name = repo["full_name"]
+
+            if full_name in self.repos:
+                repo_dict[full_name] = {
+                    "full_name": repo["full_name"],
+                    "description": repo["description"],
+                    "language": repo["language"],
+                    "forks_count": repo["forks_count"],
+                    "stargazers_count": repo["stargazers_count"],
+                }
+        return repo_dict
+
+    def _get_repo_data(self):
+        """Build a list of RepositoryMetadata objects that we can use to populate the front page."""
+
+        # Try to get site data from the cache
+        try:
+            repo_data = RepositoryMetadata.objects.get(repo_name="python-discord/site")
+
+            # If the data is older than 2 minutes, we should refresh it.
+            if (timezone.now() - repo_data.last_updated).seconds > 120:
+
+                # Get new data from API
+                api_repositories = self._get_api_data()
+                database_repositories = []
+
+                # Update or create all RepoData objects in self.repos
+                for repo_name, api_data in api_repositories.items():
+                    try:
+                        repo_data = RepositoryMetadata.objects.get(repo_name=repo_name)
+                        repo_data.description = api_data["description"]
+                        repo_data.language = api_data["language"]
+                        repo_data.forks = api_data["forks_count"]
+                        repo_data.stargazers = api_data["stargazers_count"]
+                    except RepositoryMetadata.DoesNotExist:
+                        repo_data = RepositoryMetadata(
+                            repo_name=api_data["full_name"],
+                            description=api_data["description"],
+                            forks=api_data["forks_count"],
+                            stargazers=api_data["stargazers_count"],
+                            language=api_data["language"],
+                        )
+                    repo_data.save()
+                    database_repositories.append(repo_data)
+                return database_repositories
+
+            # Otherwise, if the data is fresher than 2 minutes old, we should just return it.
+            else:
+                return RepositoryMetadata.objects.all()
+
+        # If this is raised, the database has no repodata at all, we will create them all.
+        except RepositoryMetadata.DoesNotExist:
+
+            # Get new data from API
+            api_repositories = self._get_api_data()
+            database_repositories = []
+
+            # Create all the repodata records in the database.
+            for api_data in api_repositories.values():
+                repo_data = RepositoryMetadata(
+                    repo_name=api_data["full_name"],
+                    description=api_data["description"],
+                    forks=api_data["forks_count"],
+                    stargazers=api_data["stargazers_count"],
+                    language=api_data["language"],
+                )
+                repo_data.save()
+                database_repositories.append(repo_data)
+
+            return database_repositories
+
+    def get(self, request):
+        """Collect repo data and render the homepage view"""
+
+        repo_data = self._get_repo_data()
+        return render(request, "home/index.html", {"repo_data": repo_data})
-- 
cgit v1.2.3