Skip to content

Commit c7336cb

Browse files
committed
fixes #16
1 parent 20e2073 commit c7336cb

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

fastlite/core.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ def _get_flds(tbl):
8181
def _dataclass(self:Table, store=True, suf='')->type:
8282
"Create a `dataclass` with the types and defaults of this table"
8383
res = make_dataclass(self.name.title()+suf, _get_flds(self))
84+
flexiclass(res)
8485
if store: self.cls = res
8586
return res
8687

@@ -154,7 +155,7 @@ def create(
154155
strict=False, # Apply STRICT mode to table
155156
):
156157
"Create table from `cls`, default name to snake-case version of class name"
157-
mk_dataclass(cls)
158+
flexiclass(cls)
158159
if name is None: name = camel2snake(cls.__name__)
159160
typs = {o.name: o.type for o in fields(cls)}
160161
res = self.create_table(

nbs/00_core.ipynb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -371,6 +371,7 @@
371371
"def _dataclass(self:Table, store=True, suf='')->type:\n",
372372
" \"Create a `dataclass` with the types and defaults of this table\"\n",
373373
" res = make_dataclass(self.name.title()+suf, _get_flds(self))\n",
374+
" flexiclass(res)\n",
374375
" if store: self.cls = res\n",
375376
" return res\n",
376377
"\n",
@@ -770,7 +771,7 @@
770771
" strict=False, # Apply STRICT mode to table\n",
771772
"):\n",
772773
" \"Create table from `cls`, default name to snake-case version of class name\"\n",
773-
" mk_dataclass(cls)\n",
774+
" flexiclass(cls)\n",
774775
" if name is None: name = camel2snake(cls.__name__)\n",
775776
" typs = {o.name: o.type for o in fields(cls)}\n",
776777
" res = self.create_table(\n",
@@ -797,7 +798,7 @@
797798
{
798799
"data": {
799800
"text/plain": [
800-
"Cat(id=1, name=None, age=None, city='Unknown')"
801+
"Cat(id=1, name=UNSET, age=UNSET, city='Unknown')"
801802
]
802803
},
803804
"execution_count": null,

0 commit comments

Comments
 (0)