Skip to content

Commit

Permalink
feat: add validate decorator to client api
Browse files Browse the repository at this point in the history
  • Loading branch information
dlbrittain committed Aug 5, 2022
1 parent f406453 commit 402497e
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion materializationengine/blueprints/client/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
from materializationengine.info_client import get_aligned_volumes, get_datastack_info, get_relevant_datastack_info
from materializationengine.schemas import AnalysisTableSchema, AnalysisVersionSchema
from middle_auth_client import auth_requires_permission
from materializationengine.blueprints.client.datastack import validate_version
from materializationengine.blueprints.client.datastack import validate_datastack
__version__ = "4.0.20"


Expand Down Expand Up @@ -203,6 +203,7 @@ def get_flat_model(datastack_name: str, table_name: str, version: int, Session):
class DatastackVersions(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("datastack_versions", security="apikey")
def get(self, datastack_name: str):
"""get available versions
Expand Down Expand Up @@ -233,6 +234,7 @@ def get(self, datastack_name: str):
class DatastackVersion(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("version metadata", security="apikey")
def get(self, datastack_name: str, version: int):
"""get version metadata
Expand Down Expand Up @@ -265,6 +267,7 @@ def get(self, datastack_name: str, version: int):
class DatastackMetadata(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("all valid version metadata", security="apikey")
def get(self, datastack_name: str):
"""get materialized metadata for all valid versions
Expand Down Expand Up @@ -293,6 +296,7 @@ def get(self, datastack_name: str):
class FrozenTableVersions(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("get_frozen_tables", security="apikey")
def get(self, datastack_name: str, version: int):
"""get frozen tables
Expand Down Expand Up @@ -335,6 +339,7 @@ def get(self, datastack_name: str, version: int):
class FrozenTableMetadata(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("get_frozen_table_metadata", security="apikey")
def get(self, datastack_name: str, version: int, table_name: str):
"""get frozen table metadata
Expand Down Expand Up @@ -372,6 +377,7 @@ def get(self, datastack_name: str, version: int, table_name: str):
class FrozenTableCount(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("simple_query", security="apikey")
def get(self, datastack_name: str, version: int, table_name: str):
"""get annotation count in table
Expand Down Expand Up @@ -400,6 +406,7 @@ def get(self, datastack_name: str, version: int, table_name: str):
class LiveTableQuery(Resource):
@reset_auth
@auth_requires_permission("admin_view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("live_simple_query", security="apikey")
@accepts("SimpleQuerySchema", schema=SimpleQuerySchema, api=client_bp)
def post(self, datastack_name: str, table_name: str):
Expand Down Expand Up @@ -545,6 +552,7 @@ def _format_filter(filter, table_in, seg_table):
class FrozenTableQuery(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("simple_query", security="apikey")
@accepts("SimpleQuerySchema", schema=SimpleQuerySchema, api=client_bp)
def post(self, datastack_name: str, version: int, table_name: str):
Expand Down Expand Up @@ -669,6 +677,7 @@ def post(self, datastack_name: str, version: int, table_name: str):
class FrozenQuery(Resource):
@reset_auth
@auth_requires_permission("view", table_arg="datastack_name")
@validate_datastack
@client_bp.doc("complex_query", security="apikey")
@accepts("ComplexQuerySchema", schema=ComplexQuerySchema, api=client_bp)
def post(self, datastack_name: str, version: int):
Expand Down

0 comments on commit 402497e

Please sign in to comment.