plumb takes permission

gratipay/models/team/__init__.py

@@ -7,6 +7,7 @@
 from aspen import json, log
 from gratipay.exceptions import InvalidTeamName
 from gratipay.models import add_event
+from gratipay.models.team import mixins
 from postgres.orm import Model
 
 from gratipay.billing.exchanges import MINIMUM_CHARGE
@@ -30,7 +31,7 @@ def slugize(name):
     return slug
 
 
-class Team(Model):
+class Team(Model, mixins.Takes):
     """Represent a Gratipay team.
     """
 

gratipay/models/team/mixins/__init__.py

@@ -0,0 +1,3 @@
+from .takes import TakesMixin as Takes
+
+__all__ = ['Takes']

gratipay/models/team/mixins/takes.py

@@ -0,0 +1,28 @@
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+from gratipay.models import add_event
+
+
+class TakesMixin(object):
+    """This mixin provides management of takes for
+    :py:class:`~gratipay.models.team.Team` objects.
+    """
+
+    def set_takes_enabled(self, to):
+        """Update whether takes are enabled for this team.
+
+        :param bool to: whether the team should have takes enabled
+
+        """
+        if type(to) is not bool:
+            raise TypeError('Boolean required.')
+        with self.db.get_cursor() as cursor:
+            cursor.run("UPDATE teams SET takes_enabled=%s WHERE id=%s", (to, self.id))
+            add_event( cursor
+                     , 'team'
+                     , dict( action='set takes_enabled'
+                           , id=self.id
+                           , old_value=self.takes_enabled
+                           , new_value=to
+                            )
+                      )

sql/branch.sql

@@ -41,3 +41,8 @@ CREATE TRIGGER enforce_email_for_participant_identity
 -- participants.has_verified_identity
 
 ALTER TABLE participants ADD COLUMN has_verified_identity bool NOT NULL DEFAULT false;
+
+
+-- https://github.com/gratipay/gratipay.com/pull/4017
+
+ALTER TABLE teams ADD COLUMN takes_enabled bool NOT NULL DEFAULT false;

tests/py/test_team_takes.py

@@ -0,0 +1,44 @@
+from __future__ import absolute_import, division, print_function, unicode_literals
+
+from gratipay.testing import Harness
+from gratipay.models.team import Team, mixins
+from pytest import raises
+
+
+class Tests(Harness):
+
+    def setUp(self):
+        self.enterprise = self.make_team('The Enterprise')
+
+
+    def test_team_object_subclasses_takes_mixin(self):
+        assert isinstance(self.enterprise, mixins.Takes)
+
+    def test_takes_enabled_defaults_to_false(self):
+        assert self.enterprise.takes_enabled is False
+
+
+    # ste - set_takes_enabled
+
+    def test_ste_sets_takes_enabled(self):
+        self.enterprise.set_takes_enabled(True)
+        assert Team.from_slug('TheEnterprise').takes_enabled is True
+
+    def test_ste_sets_takes_enabled_back_to_false(self):
+        self.enterprise.set_takes_enabled(False)
+        assert Team.from_slug('TheEnterprise').takes_enabled is False
+
+    def test_ste_sets_takes_rejects_non_boolean_values(self):
+        raises(TypeError, self.enterprise.set_takes_enabled, None)
+        raises(TypeError, self.enterprise.set_takes_enabled, 'foo')
+        raises(TypeError, self.enterprise.set_takes_enabled, 123)
+
+    def test_ste_setting_takes_enabled_is_a_logged_event(self):
+        self.enterprise.set_takes_enabled(True)
+        events = self.db.all("SELECT * FROM events")
+        assert len(events) == 1
+        assert events[0].type == 'team'
+        assert events[0].payload['action'] == 'set takes_enabled'
+        assert events[0].payload['old_value'] == False
+        assert events[0].payload['new_value'] == True
+        assert events[0].payload['id'] == self.enterprise.id