from flask_restful import abort from marshmallow import Schema, fields from webargs.flaskparser import use_args from flasgger import swag_from from flask import g, request, redirect from app.api.blueprint import api import app.devices.api as devices from app.api.auth_protection import ProtectedResource from app.api.schemas import (BaseResourceSchema, BaseTimestampedSchema, BaseTimestampedResourceSchema) from flask import current_app as app class BasicDeviceTypeSchema(BaseTimestampedSchema): id = fields.Integer(dump_only=True) name = fields.Str(required=True) class DeviceTypeSchema(BaseResourceSchema, BasicDeviceTypeSchema): pass class DeviceSchema(BaseTimestampedResourceSchema): id = fields.Integer(dump_only=True) name = fields.Str(required=True) device_type = fields.Nested(BasicDeviceTypeSchema, dump_only=True) device_type_id = fields.Integer(load_only=True, missing=1) class DeviceWithConfigurationSchema(DeviceSchema): configuration = fields.Raw(dump_only=True) class RecordingsSchema(BaseResourceSchema): recorded_at = fields.DateTime() record_type = fields.Integer() record_value = fields.Float() class RecordingsQuerySchema(Schema): selections = fields.Raw() filters = fields.Raw() groups = fields.Raw() orders = fields.Raw() class DeviceDocumentationSchema(BaseTimestampedResourceSchema): device_id = fields.Integer(dump_only=True) text = fields.String(required=True) class DeviceSecretSchema(BaseResourceSchema): device_secret = fields.String(dump_only=True) secret_algorithm = fields.String(required=True) class DeviceShareSchema(BaseResourceSchema): access_level_id = fields.Integer() account_id = fields.Integer(required=False) class DeviceShareTokenSchema(BaseResourceSchema): token = fields.String() activation_url = fields.String() def validate_device_ownership(device_id): if not devices.can_user_access_device(g.current_account.id, device_id): abort(403, message='You are not allowed to access this device', status='error') class DeviceResource(ProtectedResource): @swag_from('swagger/get_device_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) return DeviceWithConfigurationSchema().dump( devices.get_device(device_id)), 200 @swag_from('swagger/delete_device_spec.yaml') def delete(self, device_id): validate_device_ownership(device_id) devices.delete_device(device_id) return '', 204 class DeviceTypeResource(ProtectedResource): @swag_from('swagger/get_device_type_spec.yaml') def get(self, device_type_id): return DeviceTypeSchema().dump( devices.get_device_type(device_type_id)), 200 class DeviceTypeListResource(ProtectedResource): @use_args(DeviceTypeSchema(), locations=('json',)) @swag_from('swagger/create_device_type_spec.yaml') def post(self, args): if g.current_account.role_id != 1: abort(403, message='Only admin may create device types', status='error') created_device_type = devices.create_device_type( args['name']) return DeviceTypeSchema().dump(created_device_type), 201 @swag_from('swagger/get_device_types_spec.yaml') def get(self): return DeviceTypeSchema().dump(devices.get_device_types(), many=True), 200 class DeviceRecordingResource(ProtectedResource): @swag_from('swagger/get_device_recordings_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) request_args = request.args return RecordingsSchema().dump( devices.get_device_recordings_filtered( device_id, request_args.get('record_type'), request_args.get('start_date'), request_args.get('end_date')), many=True), 200 @swag_from('swagger/create_device_recording_spec.yaml') def post(self, device_id): validate_device_ownership(device_id) created_recording = devices.create_recording_and_return( device_id, request.json, True) return RecordingsSchema().dump(created_recording), 201 class DeviceLatestRecordingResource(ProtectedResource): @swag_from('swagger/get_latest_device_recording_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) return RecordingsSchema().dump( devices.get_latest_device_recording(device_id)), 200 class DeviceRecordingQueryResource(ProtectedResource): @use_args(RecordingsQuerySchema(), locations=('json',)) @swag_from('swagger/create_device_recording_query_spec.yaml') def post(self, args, device_id): validate_device_ownership(device_id) try: return {'content': devices.run_custom_query(device_id, args)}, 200 except ValueError as e: abort(400, message=str(e), status='error') class DeviceListResource(ProtectedResource): @use_args(DeviceSchema(), locations=('json',)) @swag_from('swagger/create_device_spec.yaml') def post(self, args): created_device = devices.create_device( args['name'], g.current_account.id, args['device_type_id']) return DeviceSchema().dump(created_device), 201 @swag_from('swagger/get_devices_spec.yaml') def get(self): return DeviceSchema().dump( devices.get_devices(g.current_account.id), many=True), 200 class DeviceConfigurationResource(ProtectedResource): @swag_from('swagger/update_device_configuration_spec.yaml') def put(self, device_id): validate_device_ownership(device_id) updated_device = devices.set_device_configuration( device_id, request.json) return DeviceWithConfigurationSchema().dump( updated_device), 200 @swag_from('swagger/get_device_configuration_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) return devices.get_device_configuration(device_id), 200 class DeviceDocumentationResource(ProtectedResource): @use_args(DeviceDocumentationSchema(), locations=('json',)) @swag_from('swagger/update_device_documentation_spec.yaml') def put(self, args, device_id): validate_device_ownership(device_id) update_device_documentation = devices.update_device_documentation( device_id, args['text']) return DeviceDocumentationSchema().dump( update_device_documentation), 200 @swag_from('swagger/get_device_documentation_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) return DeviceDocumentationSchema().dump( devices.get_device_documentation(device_id)), 200 class DeviceSecretResource(ProtectedResource): @swag_from('swagger/get_device_secret_spec.yaml') def get(self, device_id): validate_device_ownership(device_id) return DeviceSecretSchema().dump(devices.get_device(device_id)), 200 @use_args(DeviceSecretSchema(), locations=('json',)) @swag_from('swagger/update_device_secret_spec.yaml') def put(self, args, device_id): validate_device_ownership(device_id) return DeviceSecretSchema().dump( devices.update_algorithm( device_id, args['secret_algorithm'] ) ), 200 class DeviceSecretResetResource(ProtectedResource): @swag_from('swagger/reset_device_secret_spec.yaml') def post(self, device_id): validate_device_ownership(device_id) return DeviceSecretSchema().dump( devices.reset_device_secret(device_id)), 200 class DeviceShareResource(ProtectedResource): @use_args(DeviceShareSchema(), locations=('json',)) @swag_from('swagger/create_device_share_token_spec.yaml') def post(self, args, device_id): validate_device_ownership(device_id) created_token = devices.create_targeted_device_sharing_token( device_id, args['access_level_id'], args.get('account_id')) activation_url = api.url_for( DeviceShareActivationResource, device_id=device_id, token=created_token, _external=True) return DeviceShareTokenSchema().dump( { 'token': created_token, 'activation_url': activation_url } ), 201 class DeviceShareActivationResource(ProtectedResource): def get(self, device_id, token): try: success = devices.activate_device_sharing_token( g.current_account.id, token) if not success: abort(403, message='You may not get access to this device', status='error') return redirect(app.config['FRONTEND_URL']) except ValueError as e: abort(400, message=str(e), status='error')