1# SPDX-License-Identifier: Apache-2.0 

2 

3# Copyright 2020 Contributors to OpenLEADR 

4 

5# Licensed under the Apache License, Version 2.0 (the "License"); 

6# you may not use this file except in compliance with the License. 

7# You may obtain a copy of the License at 

8 

9# http://www.apache.org/licenses/LICENSE-2.0 

10 

11# Unless required by applicable law or agreed to in writing, software 

12# distributed under the License is distributed on an "AS IS" BASIS, 

13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 

14# See the License for the specific language governing permissions and 

15# limitations under the License. 

16 

17def handler(message_type): 

18 """ 

19 Decorator to mark a method as the handler for a specific message type. 

20 """ 

21 def _actual_decorator(decorated_function): 

22 decorated_function.__message_type__ = message_type 

23 return decorated_function 

24 return _actual_decorator 

25 

26 

27def service(service_name): 

28 """ 

29 Decorator to mark a class as an OpenADR Service for a specific endpoint. 

30 """ 

31 def _actual_decorator(decorated_function): 

32 decorated_function.__service_name__ = service_name 

33 return decorated_function 

34 return _actual_decorator