feat: better module instantiation

nachatz · nachatz · commit 07d752a71370 · 2023-11-30T18:35:04.000-08:00
diff --git a/lib/engine/__init__.py b/lib/engine/__init__.py
@@ -0,0 +1,3 @@
+from lib.engine.serializer import make_request, update_post, pretty_print
+
+__all__ = ["make_request", "update_post", "pretty_print"]
diff --git a/lib/engine/serializer.py b/lib/engine/serializer.py
@@ -7,7 +7,7 @@
 
 import json
 import requests
-from lib.models.post import PostModel
+from lib.models import PostModel
 
 API_URL = "https://jsonplaceholder.typicode.com/posts/1"
 
diff --git a/lib/models/__init__.py b/lib/models/__init__.py
@@ -0,0 +1,3 @@
+from lib.models.post import PostModel
+
+__all__ = ["PostModel"]
diff --git a/main.py b/main.py
@@ -8,7 +8,7 @@
 This showcases working with a Python built library
 """
 
-from lib.engine.serializer import make_request, update_post, pretty_print
+from lib.engine import make_request, update_post, pretty_print
 
 
 def main():
diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -1,5 +1,5 @@
-from lib.engine.serializer import update_post
-from lib.models.post import PostModel
+from lib.engine import update_post
+from lib.models import PostModel
 
 
 def mocked_body():

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+from lib.engine.serializer import make_request, update_post, pretty_print`
	`2`	`+`
	`3`	`+__all__ = ["make_request", "update_post", "pretty_print"]`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,3 @@`
	`1`	`+from lib.models.post import PostModel`
	`2`	`+`
	`3`	`+__all__ = ["PostModel"]`