summaryrefslogtreecommitdiff
path: root/web-server/opendc/models/scenario.py
blob: 8d53e408023fc1e9caf79b9139a803295a0effc0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from opendc.models.model import Model
from opendc.models.portfolio import Portfolio
from opendc.models.user import User
from opendc.util.exceptions import ClientError
from opendc.util.rest import Response


class Scenario(Model):
    """Model representing a Scenario."""

    collection_name = 'scenarios'

    def check_user_access(self, google_id, edit_access):
        """Raises an error if the user with given [google_id] has insufficient access.

        Checks access on the parent project.

        :param google_id: The Google ID of the user.
        :param edit_access: True when edit access should be checked, otherwise view access.
        """
        portfolio = Portfolio.from_id(self.obj['portfolioId'])
        user = User.from_google_id(google_id)
        authorizations = list(
            filter(lambda x: str(x['projectId']) == str(portfolio.obj['projectId']), user.obj['authorizations']))
        if len(authorizations) == 0 or (edit_access and authorizations[0]['authorizationLevel'] == 'VIEW'):
            raise ClientError(Response(403, 'Forbidden from retrieving/editing scenario.'))