|
3 | 3 |
|
4 | 4 | from st2common.models.base import BaseAPI
|
5 | 5 | from st2common.models.api.stormbase import StormBaseAPI, StormFoundationAPI
|
6 |
| -from st2common.models.db.reactor import RuleDB, ActionExecutionSpecDB, TriggerDB, AHTriggerDB |
| 6 | +from st2common.models.db.reactor import RuleDB, ActionExecutionSpecDB, TriggerTypeDB, TriggerDB |
7 | 7 | import st2common.validators.api.reactor as validator
|
8 | 8 |
|
9 | 9 |
|
10 |
| -def get_id(identifiable): |
11 |
| - if identifiable is None: |
12 |
| - return '' |
13 |
| - return str(identifiable.id) |
14 |
| - |
15 |
| - |
16 |
| -def get_ref(identifiable): |
17 |
| - if identifiable is None: |
18 |
| - return {} |
19 |
| - return {'id': str(identifiable.id), 'name': identifiable.name} |
20 |
| - |
21 |
| - |
22 |
| -def get_model_from_ref(db_api, ref): |
23 |
| - if ref is None: |
24 |
| - return None |
25 |
| - model_id = ref['id'] if 'id' in ref else None |
26 |
| - if model_id is not None: |
27 |
| - return db_api.get_by_id(model_id) |
28 |
| - model_name = ref['name'] if 'name' in ref else None |
29 |
| - for model in db_api.query(name=model_name): |
30 |
| - return model |
31 |
| - return None |
32 |
| - |
33 |
| - |
34 |
| -class TriggerAPI(BaseAPI): |
| 10 | +class TriggerTypeAPI(BaseAPI): |
35 | 11 | schema = {
|
36 |
| - "type": "object" |
| 12 | + 'type': 'object', |
| 13 | + 'properties': { |
| 14 | + 'id': { |
| 15 | + 'type': 'string', |
| 16 | + 'default': None |
| 17 | + }, |
| 18 | + 'name': { |
| 19 | + 'type': 'string' |
| 20 | + }, |
| 21 | + 'description': { |
| 22 | + 'type': 'string', |
| 23 | + 'default': None |
| 24 | + }, |
| 25 | + 'payload_info': { |
| 26 | + 'type': 'array' |
| 27 | + }, |
| 28 | + 'parameters_schema': { |
| 29 | + 'type': 'object' |
| 30 | + } |
| 31 | + }, |
| 32 | + 'required': ['name'], |
| 33 | + 'additionalProperties': False |
37 | 34 | }
|
38 | 35 |
|
39 | 36 | @classmethod
|
40 | 37 | def from_model(cls, model):
|
41 | 38 | trigger = model.to_mongo()
|
42 |
| - trigger['name'] = str(trigger.pop('_id')) |
| 39 | + trigger['id'] = str(trigger.pop('_id')) |
43 | 40 | return cls(**trigger)
|
44 | 41 |
|
45 | 42 | @classmethod
|
46 |
| - def to_model(cls, trigger): |
47 |
| - model = StormBaseAPI.to_model(TriggerDB, trigger) |
48 |
| - model.payload_info = trigger.payload_info |
| 43 | + def to_model(cls, triggertype): |
| 44 | + model = StormBaseAPI.to_model(TriggerTypeDB, triggertype) |
| 45 | + model.payload_info = triggertype.payload_info |
| 46 | + model.parameter_schema = triggertype.parameter_schema |
49 | 47 | return model
|
50 | 48 |
|
51 | 49 |
|
52 |
| -class AHTriggerAPI(BaseAPI): |
| 50 | +class TriggerAPI(BaseAPI): |
53 | 51 | schema = {
|
54 |
| - "type": "object" |
| 52 | + 'type': 'object', |
| 53 | + 'properties': { |
| 54 | + 'id': { |
| 55 | + 'type': 'string', |
| 56 | + 'default': None |
| 57 | + }, |
| 58 | + 'name': { |
| 59 | + 'type': 'string' |
| 60 | + }, |
| 61 | + 'type': { |
| 62 | + 'type': 'string' |
| 63 | + }, |
| 64 | + 'parameters': { |
| 65 | + 'type': 'object' |
| 66 | + } |
| 67 | + }, |
| 68 | + 'required': ['type'], |
| 69 | + 'additionalProperties': False |
55 | 70 | }
|
56 | 71 |
|
57 | 72 | @classmethod
|
58 | 73 | def from_model(cls, model):
|
59 | 74 | trigger = model.to_mongo()
|
60 | 75 | trigger['name'] = str(trigger.pop('_id'))
|
61 |
| - if trigger.has_key('type'): |
| 76 | + if 'type' in trigger: |
62 | 77 | trigger['type'] = str(trigger.pop('type'))
|
63 | 78 | return cls(**trigger)
|
64 | 79 |
|
65 | 80 | @classmethod
|
66 | 81 | def to_model(cls, trigger):
|
67 |
| - model = StormFoundationAPI.to_model(AHTriggerDB, trigger) |
| 82 | + model = StormFoundationAPI.to_model(TriggerDB, trigger) |
68 | 83 | model.name = trigger.name
|
69 | 84 | model.type = getattr(trigger, 'type', None)
|
70 | 85 | model.parameters = getattr(trigger, 'parameters', None)
|
|
0 commit comments