|
1 | 1 | # -*- coding: utf-8 -*- |
2 | 2 |
|
3 | 3 | from __future__ import unicode_literals |
| 4 | +from __future__ import absolute_import |
4 | 5 |
|
5 | 6 | import collections |
6 | 7 | import copy |
7 | 8 | import six |
8 | 9 |
|
| 10 | +from .base import BaseField |
| 11 | +from .meta import AdapterMetaClass |
| 12 | +from .utils import BindingDict |
| 13 | +from .utils import undefined |
9 | 14 |
|
10 | | -__all__ = ['Adapter', 'Field'] |
11 | 15 |
|
12 | | - |
13 | | -class undefined: |
14 | | - pass |
15 | | - |
16 | | - |
17 | | -class BindingDict(collections.MutableMapping): |
18 | | - def __init__(self, adapter): |
19 | | - self.adapter = adapter |
20 | | - self.fields = collections.OrderedDict() |
21 | | - |
22 | | - def __setitem__(self, key, field): |
23 | | - self.fields[key] = field |
24 | | - field.bind(key, self.adapter) |
25 | | - |
26 | | - def __getitem__(self, key): |
27 | | - return self.fields[key] |
28 | | - |
29 | | - def __delitem__(self, key): |
30 | | - del self.fields[key] |
31 | | - |
32 | | - def __iter__(self): |
33 | | - return iter(self.fields) |
34 | | - |
35 | | - def __len__(self): |
36 | | - return len(self.fields) |
37 | | - |
38 | | - def __repr__(self): |
39 | | - return dict.__repr__(self.fields) |
40 | | - |
41 | | - |
42 | | -def get_attribute(obj, attrs): |
43 | | - for attr in attrs: |
44 | | - if obj is None: |
45 | | - return undefined |
46 | | - |
47 | | - try: |
48 | | - if isinstance(obj, collections.Mapping): |
49 | | - obj = obj[attr] |
50 | | - elif isinstance(obj, collections.Iterable): |
51 | | - obj = obj[int(attr)] |
52 | | - else: |
53 | | - obj = getattr(obj, attr) |
54 | | - except Exception: |
55 | | - return undefined |
56 | | - |
57 | | - if callable(obj): |
58 | | - obj = obj() |
59 | | - |
60 | | - return obj |
61 | | - |
62 | | - |
63 | | -class BaseField(object): |
64 | | - def __init__(self, source=None, default=undefined, required=True): |
65 | | - self.source = source |
66 | | - self.default = default |
67 | | - self.required = required |
68 | | - |
69 | | - def bind(self, field_name, adapter): |
70 | | - if field_name == self.source: |
71 | | - raise ValueError(( |
72 | | - "The `source='{field_name}'` kwarg is redundant on " |
73 | | - "field `{adapter_name}.{field_name}`. " |
74 | | - "Remove the `source` kwarg." |
75 | | - ).format( |
76 | | - field_name=field_name, |
77 | | - adapter_name=adapter.__class__.__name__, |
78 | | - )) |
79 | | - |
80 | | - self.field_name = field_name |
81 | | - self.adapter = adapter |
82 | | - |
83 | | - if self.source is None: |
84 | | - self.source = self.field_name |
85 | | - |
86 | | - if self.source == '*': |
87 | | - self.lookup_attrs = [] |
88 | | - else: |
89 | | - self.lookup_attrs = self.source.split('.') |
90 | | - |
91 | | - def get_attribute(self, obj): |
92 | | - value = get_attribute(obj, self.lookup_attrs) |
93 | | - if value is undefined: |
94 | | - if self.default is not undefined: |
95 | | - return self.default |
96 | | - elif self.required: |
97 | | - raise ValueError(( |
98 | | - "Required value not found for field " |
99 | | - "`{adapter_name}.{field_name}`. Provide a default value." |
100 | | - ).format( |
101 | | - adapter_name=self.adapter.__class__.__name__, |
102 | | - field_name=self.field_name, |
103 | | - )) |
104 | | - else: |
105 | | - return undefined |
106 | | - else: |
107 | | - return value |
108 | | - |
109 | | - def adapt(self, data): |
110 | | - return data |
111 | | - |
112 | | - |
113 | | -class AdapterMetaClass(type): |
114 | | - def __new__(meta, name, bases, attrs): |
115 | | - fields = [ |
116 | | - (key, attrs.pop(key)) |
117 | | - for key in attrs.keys() if isinstance(attrs[key], BaseField) |
118 | | - ] |
119 | | - |
120 | | - for base in reversed(bases): |
121 | | - if hasattr(base, 'declared_fields'): |
122 | | - fields = list(base.declared_fields.items()) + fields |
123 | | - |
124 | | - attrs['declared_fields'] = collections.OrderedDict(fields) |
125 | | - |
126 | | - cls = super(AdapterMetaClass, meta).__new__(meta, name, bases, attrs) |
127 | | - |
128 | | - return cls |
129 | | - |
130 | | - |
131 | | -class Field(BaseField): |
132 | | - pass |
| 16 | +__all__ = ['Adapter'] |
133 | 17 |
|
134 | 18 |
|
135 | 19 | @six.add_metaclass(AdapterMetaClass) |
|
0 commit comments