start using webapp on djangoproject, add __eq__

hjwp · hjwp · commit cbbb2b280c4c · 2017-10-16T03:14:29.000+01:00
diff --git a/pythonanywhere/api.py b/pythonanywhere/api.py
@@ -44,6 +44,10 @@ def __init__(self, domain):
         self.domain = domain
 
 
+    def __eq__(self, other):
+        return self.domain == other.domain
+
+
     def sanity_checks(self, nuke):
         print(snakesay('Running API sanity checks'))
         token = os.environ.get('API_TOKEN')
diff --git a/pythonanywhere/django_project.py b/pythonanywhere/django_project.py
@@ -3,6 +3,7 @@
 import subprocess
 from textwrap import dedent
 
+from pythonanywhere.api import Webapp
 from pythonanywhere.snakesay import snakesay
 from pythonanywhere.virtualenvs import create_virtualenv
 
@@ -13,6 +14,7 @@ def __init__(self, domain):
         self.domain = domain
         self.project_path = Path('~/').expanduser() / self.domain
         self.wsgi_file_path = '/var/www/' + domain.replace('.', '_') + '_wsgi.py'
+        self.webapp = Webapp(domain)
 
 
     def create_virtualenv(self, django_version, nuke):
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -33,6 +33,12 @@ def test_init(self):
         assert app.domain == 'www.my-domain.com'
 
 
+    def test_compare_equal(self):
+        assert Webapp('www.my-domain.com') == Webapp('www.my-domain.com')
+
+
+    def test_compare_not_equal(self):
+        assert Webapp('www.my-domain.com') != Webapp('www.other-domain.com')
 
 
 
diff --git a/tests/test_django_project.py b/tests/test_django_project.py
@@ -5,6 +5,7 @@
 
 import pythonanywhere.django_project
 from pythonanywhere.django_project import DjangoProject
+from pythonanywhere.api import Webapp
 
 
 
@@ -19,6 +20,11 @@ def test_wsgi_file_path(self, fake_home):
         project = DjangoProject('mydomain.com')
         assert project.wsgi_file_path == '/var/www/mydomain_com_wsgi.py'
 
+    def test_webapp(self, fake_home):
+        project = DjangoProject('mydomain.com')
+        assert project.webapp == Webapp('mydomain.com')
+
+
 
 
 class TestCreateVirtualenv:
