| import marshmallow as ma |
| |
| from app.utility.base_object import BaseObject |
| |
| |
| class ResultSchema(ma.Schema): |
| id = ma.fields.String() |
| output = ma.fields.String() |
| stderr = ma.fields.String() |
| exit_code = ma.fields.String() |
| pid = ma.fields.String() |
| status = ma.fields.String() |
| agent_reported_time = ma.fields.DateTime(format=BaseObject.TIME_FORMAT, load_default=None) |
| |
| @ma.post_load |
| def build_result(self, data, **_): |
| return Result(**data) |
| |
| @ma.post_dump() |
| def prepare_dump(self, data, **_): |
| if data.get('agent_reported_time', None) is None: |
| data.pop('agent_reported_time', None) |
| return data |
| |
| |
| class Result(BaseObject): |
| |
| schema = ResultSchema() |
| |
| def __init__(self, id, output, stderr='', exit_code='', pid=0, status=0, agent_reported_time=None): |
| super().__init__() |
| self.id = id |
| self.output = output |
| self.stderr = stderr |
| self.exit_code = exit_code |
| self.pid = pid |
| self.status = status |
| self.agent_reported_time = agent_reported_time |
