Refactor model schema package

rbw · Nov 7, 2020 · 7b39724 · 7b39724
1 parent 5e79e0a
commit 7b39724
Show file tree

Hide file tree

Showing 17 changed files with 6 additions and 6 deletions.
diff --git a/aiosnow/models/__init__.py b/aiosnow/models/__init__.py
@@ -1,3 +1,3 @@
 from .attachment import AttachmentModel
 from .table import TableModel
-from ._base.schema import fields, ModelSchema, ModelSchemaMeta, Pluck, BaseField
+from ._schema import fields, ModelSchema, ModelSchemaMeta, Pluck, BaseField
diff --git a/aiosnow/models/_base/__init__.py b/aiosnow/models/_base/__init__.py
@@ -1,4 +1,3 @@
-from .schema import fields, ModelSchema, ModelSchemaMeta, Pluck, Nested, BaseField
 from .model import BaseModel, BaseModelMeta
 from .table import BaseTableModel
 
diff --git a/aiosnow/models/_base/model.py b/aiosnow/models/_base/model.py
@@ -18,7 +18,7 @@
     methods,
 )
 
-from .._base.schema import ModelSchema, ModelSchemaMeta, Nested, BaseField
+from .._schema import ModelSchema, ModelSchemaMeta, BaseField, Nested
 
 req_cls_map = {
     methods.GET: GetRequest,

diff --git a/aiosnow/models/_base/schema/__init__.py → aiosnow/models/_schema/__init__.py b/aiosnow/models/_base/schema/__init__.py → aiosnow/models/_schema/__init__.py
diff --git a/...ow/models/_base/schema/fields/__init__.py → aiosnow/models/_schema/fields/__init__.py b/...ow/models/_base/schema/fields/__init__.py → aiosnow/models/_schema/fields/__init__.py
diff --git a/aiosnow/models/_base/schema/fields/base.py → aiosnow/models/_schema/fields/base.py b/aiosnow/models/_base/schema/fields/base.py → aiosnow/models/_schema/fields/base.py
@@ -2,7 +2,7 @@
 
 import marshmallow
 
-from aiosnow.models._base.schema.helpers import Pluck
+from aiosnow.models._schema.helpers.pluck import Pluck
 
 
 class BaseField(marshmallow.fields.Field):

diff --git a/...now/models/_base/schema/fields/boolean.py → aiosnow/models/_schema/fields/boolean.py b/...now/models/_base/schema/fields/boolean.py → aiosnow/models/_schema/fields/boolean.py
diff --git a/...ow/models/_base/schema/fields/datetime.py → aiosnow/models/_schema/fields/datetime.py b/...ow/models/_base/schema/fields/datetime.py → aiosnow/models/_schema/fields/datetime.py
diff --git a/...now/models/_base/schema/fields/integer.py → aiosnow/models/_schema/fields/integer.py b/...now/models/_base/schema/fields/integer.py → aiosnow/models/_schema/fields/integer.py
diff --git a/aiosnow/models/_base/schema/fields/mapped.py → aiosnow/models/_schema/fields/mapped.py b/aiosnow/models/_base/schema/fields/mapped.py → aiosnow/models/_schema/fields/mapped.py
diff --git a/aiosnow/models/_base/schema/fields/string.py → aiosnow/models/_schema/fields/string.py b/aiosnow/models/_base/schema/fields/string.py → aiosnow/models/_schema/fields/string.py
diff --git a/...w/models/_base/schema/helpers/__init__.py → aiosnow/models/_schema/helpers/__init__.py b/...w/models/_base/schema/helpers/__init__.py → aiosnow/models/_schema/helpers/__init__.py
diff --git a/aiosnow/models/_base/schema/helpers/pluck.py → aiosnow/models/_schema/helpers/pluck.py b/aiosnow/models/_base/schema/helpers/pluck.py → aiosnow/models/_schema/helpers/pluck.py
diff --git a/aiosnow/models/_base/schema/nested.py → aiosnow/models/_schema/nested.py b/aiosnow/models/_base/schema/nested.py → aiosnow/models/_schema/nested.py
diff --git a/aiosnow/models/_base/schema/schema.py → aiosnow/models/_schema/schema.py b/aiosnow/models/_base/schema/schema.py → aiosnow/models/_schema/schema.py
diff --git a/aiosnow/models/attachment/model.py b/aiosnow/models/attachment/model.py
@@ -9,7 +9,8 @@
 
 from .file import FileHandler, FileReader, FileWriter
 
-from .._base import fields, BaseTableModel
+from .._base import BaseTableModel
+from .._schema import fields
 
 
 class AttachmentModel(BaseTableModel):

diff --git a/tests/conftest.py b/tests/conftest.py
@@ -3,7 +3,7 @@
 import pytest
 from aiohttp import web
 
-from aiosnow import fields, Client
+from aiosnow import Client, fields
 from aiosnow.request.response import Response