diff --git a/src/pdns/models.py b/src/pdns/models.py index b7ac0c6..560fc79 100644 --- a/src/pdns/models.py +++ b/src/pdns/models.py @@ -49,6 +49,9 @@ class PDNSContext(models.Model): 'online': self.online_servers(), 'offline': self.offline_servers()} + def available(self): + return self.online_servers() > 0 + class PDNSServer(models.Model): name = models.CharField(verbose_name=_('Name'), diff --git a/src/pdns/test_models.py b/src/pdns/test_models.py index fe40be6..0ddce5d 100644 --- a/src/pdns/test_models.py +++ b/src/pdns/test_models.py @@ -55,6 +55,7 @@ class PDNSModelTestCase(TestCase): self.assertEqual(test_context.total_servers(),1) self.assertEqual(test_context.online_servers(),1) self.assertEqual(test_context.offline_servers(),0) + self.assertTrue(test_context.available()) def test_context_offline(self): test_context = PDNSContext.objects.get(name='test context') @@ -64,8 +65,7 @@ class PDNSModelTestCase(TestCase): self.assertEqual(test_context.total_servers(),1) self.assertEqual(test_context.online_servers(),0) self.assertEqual(test_context.offline_servers(),1) - - + self.assertFalse(test_context.available()) diff --git a/src/requirements.txt b/src/requirements.txt index 5ca0079..a7da4bd 100644 --- a/src/requirements.txt +++ b/src/requirements.txt @@ -3,7 +3,6 @@ Django==5.2.5 django-allauth==65.12.1 django-environ==0.12.0 dotenv==0.9.9 -environ==1.0 psycopg==3.2.9 python-dotenv==1.1.1 sqlparse==0.5.3