1# 

2# Copyright (C) 2019-2020 Leo P. Singer <leo.singer@ligo.org> 

3# 

4# This program is free software: you can redistribute it and/or modify 

5# it under the terms of the GNU General Public License as published by 

6# the Free Software Foundation, either version 3 of the License, or 

7# (at your option) any later version. 

8# 

9# This program is distributed in the hope that it will be useful, 

10# but WITHOUT ANY WARRANTY; without even the implied warranty of 

11# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 

12# GNU General Public License for more details. 

13# 

14# You should have received a copy of the GNU General Public License 

15# along with this program. If not, see <https://www.gnu.org/licenses/>. 

16# 

17from .base import Deletable, Mutable, Resource 

18 

19 

20# FIXME: GraceDB expects different HTTP methods to write labels for events vs. 

21# superevents! Replace BaseLabels, EventLabels, SupereventLabels with a single 

22# Labels class once this is fixed. 

23class BaseLabels(Deletable, Mutable, Resource): 

24 

25 path = 'labels/' 

26 

27 def get(self, **kwargs): 

28 return super().get(**kwargs)['labels'] 

29 

30 

31class EventLabels(BaseLabels): 

32 

33 path = 'labels/' 

34 

35 def create(self, label): 

36 return super().create_or_update(label) 

37 

38 

39class SupereventLabels(BaseLabels): 

40 

41 path = 'labels/' 

42 

43 def create(self, label): 

44 return super().create_or_update(None, data={'name': label})