diff --git a/apps/wallet/src/app/__main__.py b/apps/wallet/src/app/__main__.py index f7ebccf6..a1fc6d3b 100755 --- a/apps/wallet/src/app/__main__.py +++ b/apps/wallet/src/app/__main__.py @@ -7,19 +7,13 @@ with multi-chain support. """ import uvicorn -import logging from pathlib import Path +from aitbc import get_logger from app.main import app from app.settings import settings -# Configure logging -logging.basicConfig( - level=logging.INFO, - format="%(asctime)s - %(name)s - %(levelname)s - %(message)s" -) - -logger = logging.getLogger(__name__) +logger = get_logger(__name__) def main(): """Main entry point for the wallet daemon""" diff --git a/apps/wallet/src/app/chain/chain_aware_wallet_service.py b/apps/wallet/src/app/chain/chain_aware_wallet_service.py index 56e0df8d..33f0d41d 100755 --- a/apps/wallet/src/app/chain/chain_aware_wallet_service.py +++ b/apps/wallet/src/app/chain/chain_aware_wallet_service.py @@ -7,15 +7,15 @@ isolation, and management across different blockchain networks. from typing import Dict, List, Optional, Any from pathlib import Path -import logging from datetime import datetime +from aitbc import get_logger from .manager import ChainManager, ChainConfig, ChainStatus from .multichain_ledger import MultiChainLedgerAdapter, ChainWalletMetadata from ..keystore.persistent_service import PersistentKeystoreService from ..security import wipe_buffer -logger = logging.getLogger(__name__) +logger = get_logger(__name__) class ChainAwareWalletService: diff --git a/apps/wallet/src/app/chain/manager.py b/apps/wallet/src/app/chain/manager.py index afbbf55e..087c74aa 100755 --- a/apps/wallet/src/app/chain/manager.py +++ b/apps/wallet/src/app/chain/manager.py @@ -10,10 +10,11 @@ from pathlib import Path from dataclasses import dataclass, field from enum import Enum import json -import logging from datetime import datetime -logger = logging.getLogger(__name__) +from aitbc import get_logger + +logger = get_logger(__name__) class ChainStatus(Enum): diff --git a/apps/wallet/src/app/chain/multichain_ledger.py b/apps/wallet/src/app/chain/multichain_ledger.py index 6285395a..f8f5e511 100755 --- a/apps/wallet/src/app/chain/multichain_ledger.py +++ b/apps/wallet/src/app/chain/multichain_ledger.py @@ -12,11 +12,11 @@ import threading import json from datetime import datetime from dataclasses import dataclass, asdict -import logging +from aitbc import get_logger from .manager import ChainManager, ChainConfig -logger = logging.getLogger(__name__) +logger = get_logger(__name__) @dataclass