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


class Simulation(Model):
    collection_name = 'simulations'

    def check_user_access(self, google_id, edit_access):
        user = User.from_google_id(google_id)
        authorizations = list(
            filter(lambda x: str(x['simulationId']) == str(self.obj['_id']), user.obj['authorizations']))
        if len(authorizations) == 0 or (edit_access and authorizations[0]['authorizationLevel'] == 'VIEW'):
            raise ClientError(Response(403, "Forbidden from retrieving simulation."))

    def get_all_authorizations(self):
        return [
            user['_id'] for user in DB.fetch_all({'authorizations': {
                'simulationId': self.obj['_id']
            }}, User.collection_name)
        ]