diff options
Diffstat (limited to 'pydis_site/apps/home')
| -rw-r--r-- | pydis_site/apps/home/tests/test_repodata_helpers.py | 6 | ||||
| -rw-r--r-- | pydis_site/apps/home/views/home.py | 8 | 
2 files changed, 6 insertions, 8 deletions
diff --git a/pydis_site/apps/home/tests/test_repodata_helpers.py b/pydis_site/apps/home/tests/test_repodata_helpers.py index 4007eded..a963f733 100644 --- a/pydis_site/apps/home/tests/test_repodata_helpers.py +++ b/pydis_site/apps/home/tests/test_repodata_helpers.py @@ -42,7 +42,7 @@ class TestRepositoryMetadataHelpers(TestCase):          metadata = self.home_view._get_repo_data()          self.assertIsInstance(metadata[0], RepositoryMetadata) -        self.assertEquals(len(metadata), len(self.home_view.repos)) +        self.assertEqual(len(metadata), len(self.home_view.repos))      def test_returns_cached_metadata(self):          """Test if the _get_repo_data helper returns cached data when available.""" @@ -82,7 +82,7 @@ class TestRepositoryMetadataHelpers(TestCase):          repo = self.home_view.repos[0]          self.assertIsInstance(api_data, dict) -        self.assertEquals(len(api_data), len(self.home_view.repos)) +        self.assertEqual(len(api_data), len(self.home_view.repos))          self.assertIn(repo, api_data.keys())          self.assertIn("stargazers_count", api_data[repo]) @@ -126,7 +126,7 @@ class TestRepositoryMetadataHelpers(TestCase):          with self.assertLogs():              metadata = self.home_view._get_repo_data() -        self.assertEquals(len(metadata), 0) +        self.assertEqual(len(metadata), 0)      def test_cleans_up_stale_metadata(self):          """Tests that we clean up stale metadata when we start the HomeView.""" diff --git a/pydis_site/apps/home/views/home.py b/pydis_site/apps/home/views/home.py index 9bb1f8fd..8a165682 100644 --- a/pydis_site/apps/home/views/home.py +++ b/pydis_site/apps/home/views/home.py @@ -32,9 +32,7 @@ class HomeView(View):      def __init__(self):          """Clean up stale RepositoryMetadata.""" -        self._static_build = settings.env("STATIC_BUILD") - -        if not self._static_build: +        if not settings.STATIC_BUILD:              RepositoryMetadata.objects.exclude(repo_name__in=self.repos).delete()          # If no token is defined (for example in local development), then @@ -94,7 +92,7 @@ class HomeView(View):      def _get_repo_data(self) -> List[RepositoryMetadata]:          """Build a list of RepositoryMetadata objects that we can use to populate the front page."""          # First off, load the timestamp of the least recently updated entry. -        if self._static_build: +        if settings.STATIC_BUILD:              last_update = None          else:              last_update = ( @@ -121,7 +119,7 @@ class HomeView(View):                  for api_data in api_repositories.values()              ] -            if settings.env("STATIC_BUILD"): +            if settings.STATIC_BUILD:                  return data              else:                  return RepositoryMetadata.objects.bulk_create(data)  |