summaryrefslogtreecommitdiffstats
path: root/utils/test/testapi/opnfv_testapi/resources/models.py
blob: 0ea482fd2e3033b11e51bf4438b228be294a9c5d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
##############################################################################
# Copyright (c) 2015 Orange
# guyrodrigue.koffi@orange.com / koffirodrigue@gmail.com
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
# feng.xiaowei@zte.com.cn  mv Pod to pod_models.py                 5-18-2016
# feng.xiaowei@zte.com.cn  add MetaCreateResponse/MetaGetResponse  5-18-2016
# feng.xiaowei@zte.com.cn  mv TestProject to project_models.py     5-19-2016
# feng.xiaowei@zte.com.cn  delete meta class                       5-19-2016
# feng.xiaowei@zte.com.cn  add CreateResponse                      5-19-2016
# feng.xiaowei@zte.com.cn  mv TestCase to testcase_models.py       5-20-2016
# feng.xiaowei@zte.com.cn  mv TestResut to result_models.py        5-23-2016
# feng.xiaowei@zte.com.cn  add ModelBase                           12-20-2016
##############################################################################
import copy
import ast


from opnfv_testapi.tornado_swagger import swagger


class ModelBase(object):

    def format(self):
        return self._format(['_id'])

    def format_http(self):
        return self._format([])

    @classmethod
    def from_dict(cls, a_dict):
        if a_dict is None:
            return None

        attr_parser = cls.attr_parser()
        t = cls()
        for k, v in a_dict.iteritems():
            value = v
            if isinstance(v, dict) and k in attr_parser:
                value = attr_parser[k].from_dict(v)
            elif isinstance(v, list) and k in attr_parser:
                value = []
                for item in v:
                    value.append(attr_parser[k].from_dict(item))

            t.__setattr__(k, value)

        return t

    @staticmethod
    def attr_parser():
        return {}

    def _format(self, excludes):
        new_obj = copy.deepcopy(self)
        dicts = new_obj.__dict__
        for k in dicts.keys():
            if k in excludes:
                del dicts[k]
            elif dicts[k]:
                dicts[k] = self._obj_format(dicts[k])
        return dicts

    def _obj_format(self, obj):
        if self._has_format(obj):
            obj = obj.format()
        elif isinstance(obj, unicode):
            try:
                obj = self._obj_format(ast.literal_eval(obj))
            except:
                try:
                    obj = str(obj)
                except:
                    obj = obj
        elif isinstance(obj, list):
            hs = list()
            for h in obj:
                hs.append(self._obj_format(h))
            obj = hs
        elif not isinstance(obj, (str, int, float, dict)):
            obj = str(obj)
        return obj

    @staticmethod
    def _has_format(obj):
        return not isinstance(obj, (str, unicode)) and hasattr(obj, 'format')


@swagger.model()
class CreateResponse(ModelBase):
    def __init__(self, href=''):
        self.href = href


@swagger.model()
class Versions(ModelBase):
    """
        @property versions:
        @ptype versions: C{list} of L{Version}
    """

    def __init__(self):
        self.versions = list()

    @staticmethod
    def attr_parser():
        return {'versions': Version}


@swagger.model()
class Version(ModelBase):
    def __init__(self, version=None, description=None):
        self.version = version
        self.description = description