From 3ccba169eeab541f3d834952703c58002dd620cb Mon Sep 17 00:00:00 2001
From: nick <nick@bisonai.com>
Date: Thu, 10 Oct 2024 17:42:31 +0900
Subject: [PATCH] fix: add missing imports & update docs

---
 datamaxi/datamaxi/__init__.py | 4 ++++
 docs/orderbook.md             | 6 ++++++
 docs/wallet-status.md         | 6 ++++++
 mkdocs.yml                    | 2 ++
 4 files changed, 18 insertions(+)
 create mode 100644 docs/orderbook.md
 create mode 100644 docs/wallet-status.md

diff --git a/datamaxi/datamaxi/__init__.py b/datamaxi/datamaxi/__init__.py
index f3d91a5..e20a3af 100644
--- a/datamaxi/datamaxi/__init__.py
+++ b/datamaxi/datamaxi/__init__.py
@@ -9,6 +9,8 @@
 from datamaxi.datamaxi.cex_candle import CexCandle  # used in documentation # noqa:F401
 from datamaxi.datamaxi.dex_candle import DexCandle  # used in documentation # noqa:F401
 from datamaxi.datamaxi.dex_trade import DexTrade  # used in documentation # noqa:F401
+from datamaxi.datamaxi.orderbook import Orderbook
+from datamaxi.datamaxi.wallet_status import WalletStatus
 
 
 class Datamaxi:
@@ -30,3 +32,5 @@ def __init__(self, api_key=None, **kwargs: Any):
         self.forex = Forex(api_key, **kwargs)
         self.ticker = Ticker(api_key, **kwargs)
         self.premium = Premium(api_key, **kwargs)
+        self.orderbook = Orderbook(api_key, **kwargs)
+        self.wallet_status = WalletStatus(api_key, **kwargs)
diff --git a/docs/orderbook.md b/docs/orderbook.md
new file mode 100644
index 0000000..5e4bcf0
--- /dev/null
+++ b/docs/orderbook.md
@@ -0,0 +1,6 @@
+# Orderbook
+
+::: datamaxi.datamaxi.Orderbook
+    options:
+      show_submodules: true
+      show_source: false
diff --git a/docs/wallet-status.md b/docs/wallet-status.md
new file mode 100644
index 0000000..4953548
--- /dev/null
+++ b/docs/wallet-status.md
@@ -0,0 +1,6 @@
+# Wallet Status
+
+::: datamaxi.datamaxi.WalletStatus
+    options:
+      show_submodules: true
+      show_source: false
diff --git a/mkdocs.yml b/mkdocs.yml
index db64a9f..e622ec0 100644
--- a/mkdocs.yml
+++ b/mkdocs.yml
@@ -40,6 +40,8 @@ nav:
       - Forex: forex.md
       - Ticker: ticker.md
       - Premium: premium.md
+      - Orderbook: orderbook.md
+      - Wallet Status: wallet-status.md
   - Defillama: defillama.md
   - Trend:
       - Naver Trend: naver-trend.md