|
1 | | -from __future__ import annotations |
2 | | - |
3 | | -import json |
4 | | -from typing import Any, Dict, Optional |
5 | | - |
6 | | -import attrs |
7 | | - |
8 | | -from ..schema import Field |
9 | | - |
10 | | - |
11 | | -@attrs.define(kw_only=True, repr=False) |
12 | | -class ArrayField(Field): |
13 | | - type = "array" |
14 | | - builtin = True |
15 | | - supported_constraints = [ |
16 | | - "required", |
17 | | - "minLength", |
18 | | - "maxLength", |
19 | | - "enum", |
20 | | - ] |
21 | | - |
22 | | - array_item: Optional[Dict[str, Any]] = attrs.field(factory=dict) |
23 | | - """ |
24 | | - A dictionary that specifies the type and other constraints for the |
25 | | - data that will be read in this data type field. |
26 | | - """ |
27 | | - |
28 | | - # Read |
29 | | - |
30 | | - def create_cell_reader(self): |
31 | | - default_reader = super().create_cell_reader() |
32 | | - |
33 | | - # Create field |
34 | | - field_reader = None |
35 | | - if self.array_item: |
36 | | - descriptor = self.array_item.copy() |
37 | | - descriptor.pop("arrayItem", None) |
38 | | - descriptor.setdefault("name", self.name) |
39 | | - descriptor.setdefault("type", "any") |
40 | | - field = Field.from_descriptor(descriptor) |
41 | | - field_reader = field.create_cell_reader() |
42 | | - |
43 | | - # Create reader |
44 | | - def cell_reader(cell: Any): |
45 | | - cell, notes = default_reader(cell) |
46 | | - if cell is not None and not notes and field_reader: |
47 | | - for index, item in enumerate(cell): |
48 | | - item_cell, item_notes = field_reader(item) |
49 | | - if item_notes: |
50 | | - notes = notes or {} |
51 | | - for name, note in item_notes.items(): |
52 | | - notes[name] = f"array item {note}" |
53 | | - cell[index] = item_cell |
54 | | - return cell, notes |
55 | | - |
56 | | - return cell_reader |
57 | | - |
58 | | - def create_value_reader(self): |
59 | | - # Create reader |
60 | | - def value_reader(cell: Any): # type: ignore |
61 | | - if not isinstance(cell, list): |
62 | | - if isinstance(cell, str): |
63 | | - try: |
64 | | - cell = json.loads(cell) |
65 | | - except Exception: |
66 | | - return None |
67 | | - if not isinstance(cell, list): |
68 | | - return None |
69 | | - elif isinstance(cell, tuple): |
70 | | - cell = list(cell) # type: ignore |
71 | | - else: |
72 | | - return None |
73 | | - return cell # type: ignore |
74 | | - |
75 | | - return value_reader |
76 | | - |
77 | | - # Write |
78 | | - |
79 | | - def create_value_writer(self): |
80 | | - # Create writer |
81 | | - def value_writer(cell: Any): |
82 | | - return json.dumps(cell) |
83 | | - |
84 | | - return value_writer |
85 | | - |
86 | | - # Metadata |
87 | | - |
88 | | - metadata_profile_patch = { |
89 | | - "properties": { |
90 | | - "arrayItem": {"type": "object"}, |
91 | | - } |
92 | | - } |
| 1 | +# from __future__ import annotations |
| 2 | + |
| 3 | +# import json |
| 4 | +# from dataclasses import dataclass |
| 5 | +# from typing import Any |
| 6 | + |
| 7 | +# from ..schema.field_descriptor import ArrayFieldDescriptor, FieldDescriptor |
| 8 | +from . import AnyReadWriter |
| 9 | + |
| 10 | +ArrayReadWriter = AnyReadWriter |
| 11 | +# @dataclass |
| 12 | +# class ArrayReadWriter: |
| 13 | +# _descriptor: ArrayFieldDescriptor |
| 14 | + |
| 15 | +# def create_cell_reader(self): |
| 16 | +# # Create field |
| 17 | +# field_reader = None |
| 18 | + |
| 19 | +# if self._descriptor.arrayItem: |
| 20 | +# item_descriptor = self._descriptor.arrayItem |
| 21 | +# field = Fielditem_descriptor.from_descriptor(item_descriptor) |
| 22 | +# field_reader = field.create_cell_reader() |
| 23 | + |
| 24 | +# # Create reader |
| 25 | +# def cell_reader(cell: Any): |
| 26 | +# cell, notes = default_reader(cell) |
| 27 | +# if cell is not None and not notes and field_reader: |
| 28 | +# for index, item in enumerate(cell): |
| 29 | +# item_cell, item_notes = field_reader(item) |
| 30 | +# if item_notes: |
| 31 | +# notes = notes or {} |
| 32 | +# for name, note in item_notes.items(): |
| 33 | +# notes[name] = f"array item {note}" |
| 34 | +# cell[index] = item_cell |
| 35 | +# return cell, notes |
| 36 | + |
| 37 | +# return cell_reader |
| 38 | + |
| 39 | +# def create_value_reader(self): |
| 40 | +# # Create reader |
| 41 | +# def value_reader(cell: Any): # type: ignore |
| 42 | +# if not isinstance(cell, list): |
| 43 | +# if isinstance(cell, str): |
| 44 | +# try: |
| 45 | +# cell = json.loads(cell) |
| 46 | +# except Exception: |
| 47 | +# return None |
| 48 | +# if not isinstance(cell, list): |
| 49 | +# return None |
| 50 | +# elif isinstance(cell, tuple): |
| 51 | +# cell = list(cell) # type: ignore |
| 52 | +# else: |
| 53 | +# return None |
| 54 | +# return cell # type: ignore |
| 55 | + |
| 56 | +# return value_reader |
| 57 | + |
| 58 | +# # Write |
| 59 | + |
| 60 | +# def create_value_writer(self): |
| 61 | +# # Create writer |
| 62 | +# def value_writer(cell: Any): |
| 63 | +# return json.dumps(cell) |
| 64 | + |
| 65 | +# return value_writer |
0 commit comments