From 1afb0af8bcd3b072fd9d9862f3480773edb92ab0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C3=A1rmenas=20Haniel?=
 <42049025+mitsuaky@users.noreply.github.com>
Date: Wed, 9 Nov 2022 15:14:53 -0300
Subject: [PATCH] Add __all__ to export classes properly

---
 InquirerPy/__init__.py            |  2 ++
 InquirerPy/base/__init__.py       |  9 +++++++++
 InquirerPy/containers/__init__.py |  2 ++
 InquirerPy/inquirer.py            | 14 ++++++++++++++
 InquirerPy/prompts/__init__.py    | 14 ++++++++++++++
 5 files changed, 41 insertions(+)

diff --git a/InquirerPy/__init__.py b/InquirerPy/__init__.py
index 4d3f794..0755827 100644
--- a/InquirerPy/__init__.py
+++ b/InquirerPy/__init__.py
@@ -1,2 +1,4 @@
+__all__ = ["prompt", "prompt_async", "get_style"]
+
 from InquirerPy.resolver import prompt, prompt_async
 from InquirerPy.utils import get_style
diff --git a/InquirerPy/base/__init__.py b/InquirerPy/base/__init__.py
index fb6d176..4f7b1d2 100644
--- a/InquirerPy/base/__init__.py
+++ b/InquirerPy/base/__init__.py
@@ -9,6 +9,15 @@
 ListPrompt ← ExpandPrompt ...
 """
 
+__all__ = [
+    "BaseComplexPrompt",
+    "FakeDocument",
+    "Choice",
+    "InquirerPyUIListControl",
+    "BaseSimplePrompt",
+    "BaseListPrompt",
+]
+
 from .complex import BaseComplexPrompt, FakeDocument
 from .control import Choice, InquirerPyUIListControl
 from .list import BaseListPrompt
diff --git a/InquirerPy/containers/__init__.py b/InquirerPy/containers/__init__.py
index 4877158..e04aa8a 100644
--- a/InquirerPy/containers/__init__.py
+++ b/InquirerPy/containers/__init__.py
@@ -1 +1,3 @@
+__all__ = ["SpinnerWindow"]
+
 from .spinner import SpinnerWindow
diff --git a/InquirerPy/inquirer.py b/InquirerPy/inquirer.py
index 3553a3f..6e31d61 100644
--- a/InquirerPy/inquirer.py
+++ b/InquirerPy/inquirer.py
@@ -5,6 +5,20 @@
 
 `inquirer` directly interact with individual prompt classes. It’s more flexible, easier to customise and also provides IDE type hintings/completions.
 """
+
+__all__ = [
+    "checkbox",
+    "confirm",
+    "expand",
+    "filepath",
+    "fuzzy",
+    "text",
+    "select",
+    "number",
+    "rawlist",
+    "secret",
+]
+
 from InquirerPy.prompts import CheckboxPrompt as checkbox
 from InquirerPy.prompts import ConfirmPrompt as confirm
 from InquirerPy.prompts import ExpandPrompt as expand
diff --git a/InquirerPy/prompts/__init__.py b/InquirerPy/prompts/__init__.py
index a04cf64..d8c2d64 100644
--- a/InquirerPy/prompts/__init__.py
+++ b/InquirerPy/prompts/__init__.py
@@ -1,4 +1,18 @@
 """Module contains import of all prompts classes."""
+
+__all__ = [
+    "CheckboxPrompt",
+    "ConfirmPrompt",
+    "ExpandPrompt",
+    "FilePathPrompt",
+    "FuzzyPrompt",
+    "InputPrompt",
+    "ListPrompt",
+    "NumberPrompt",
+    "RawlistPrompt",
+    "SecretPrompt",
+]
+
 from InquirerPy.prompts.checkbox import CheckboxPrompt
 from InquirerPy.prompts.confirm import ConfirmPrompt
 from InquirerPy.prompts.expand import ExpandPrompt