fix: resolve lint issues in main.py and test_main.py
Some checks failed
CI / test (pull_request) Has been cancelled
Some checks failed
CI / test (pull_request) Has been cancelled
Remove unused imports (sys, ScenarioMatch, asyncio, StockPlaybook), fix import ordering, and split long lines for ruff compliance. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
10
src/main.py
10
src/main.py
@@ -10,13 +10,14 @@ import argparse
|
|||||||
import asyncio
|
import asyncio
|
||||||
import logging
|
import logging
|
||||||
import signal
|
import signal
|
||||||
import sys
|
|
||||||
from datetime import UTC, date, datetime
|
from datetime import UTC, date, datetime
|
||||||
from typing import Any
|
from typing import Any
|
||||||
|
|
||||||
from src.analysis.scanner import MarketScanner
|
from src.analysis.scanner import MarketScanner
|
||||||
from src.analysis.smart_scanner import ScanCandidate, SmartVolatilityScanner
|
from src.analysis.smart_scanner import ScanCandidate, SmartVolatilityScanner
|
||||||
from src.analysis.volatility import VolatilityAnalyzer
|
from src.analysis.volatility import VolatilityAnalyzer
|
||||||
|
from src.brain.context_selector import ContextSelector
|
||||||
|
from src.brain.gemini_client import GeminiClient, TradeDecision
|
||||||
from src.broker.kis_api import KISBroker
|
from src.broker.kis_api import KISBroker
|
||||||
from src.broker.overseas import OverseasBroker
|
from src.broker.overseas import OverseasBroker
|
||||||
from src.config import Settings
|
from src.config import Settings
|
||||||
@@ -29,13 +30,11 @@ from src.db import init_db, log_trade
|
|||||||
from src.logging.decision_logger import DecisionLogger
|
from src.logging.decision_logger import DecisionLogger
|
||||||
from src.logging_config import setup_logging
|
from src.logging_config import setup_logging
|
||||||
from src.markets.schedule import MarketInfo, get_next_market_open, get_open_markets
|
from src.markets.schedule import MarketInfo, get_next_market_open, get_open_markets
|
||||||
from src.brain.context_selector import ContextSelector
|
|
||||||
from src.brain.gemini_client import GeminiClient, TradeDecision
|
|
||||||
from src.notifications.telegram_client import TelegramClient, TelegramCommandHandler
|
from src.notifications.telegram_client import TelegramClient, TelegramCommandHandler
|
||||||
from src.strategy.models import DayPlaybook
|
from src.strategy.models import DayPlaybook
|
||||||
from src.strategy.playbook_store import PlaybookStore
|
from src.strategy.playbook_store import PlaybookStore
|
||||||
from src.strategy.pre_market_planner import PreMarketPlanner
|
from src.strategy.pre_market_planner import PreMarketPlanner
|
||||||
from src.strategy.scenario_engine import ScenarioEngine, ScenarioMatch
|
from src.strategy.scenario_engine import ScenarioEngine
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
@@ -1156,7 +1155,8 @@ async def run(settings: Settings) -> None:
|
|||||||
metrics = await priority_queue.get_metrics()
|
metrics = await priority_queue.get_metrics()
|
||||||
if metrics.total_enqueued > 0:
|
if metrics.total_enqueued > 0:
|
||||||
logger.info(
|
logger.info(
|
||||||
"Priority queue metrics: enqueued=%d, dequeued=%d, size=%d, timeouts=%d, errors=%d",
|
"Priority queue metrics: enqueued=%d, dequeued=%d,"
|
||||||
|
" size=%d, timeouts=%d, errors=%d",
|
||||||
metrics.total_enqueued,
|
metrics.total_enqueued,
|
||||||
metrics.total_dequeued,
|
metrics.total_dequeued,
|
||||||
metrics.current_size,
|
metrics.current_size,
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
"""Tests for main trading loop integration."""
|
"""Tests for main trading loop integration."""
|
||||||
|
|
||||||
import asyncio
|
|
||||||
from datetime import date
|
from datetime import date
|
||||||
from unittest.mock import AsyncMock, MagicMock, patch
|
from unittest.mock import AsyncMock, MagicMock, patch
|
||||||
|
|
||||||
@@ -8,7 +7,12 @@ import pytest
|
|||||||
|
|
||||||
from src.core.risk_manager import CircuitBreakerTripped, FatFingerRejected
|
from src.core.risk_manager import CircuitBreakerTripped, FatFingerRejected
|
||||||
from src.main import safe_float, trading_cycle
|
from src.main import safe_float, trading_cycle
|
||||||
from src.strategy.models import DayPlaybook, ScenarioAction, StockCondition, StockPlaybook, StockScenario
|
from src.strategy.models import (
|
||||||
|
DayPlaybook,
|
||||||
|
ScenarioAction,
|
||||||
|
StockCondition,
|
||||||
|
StockScenario,
|
||||||
|
)
|
||||||
from src.strategy.scenario_engine import ScenarioEngine, ScenarioMatch
|
from src.strategy.scenario_engine import ScenarioEngine, ScenarioMatch
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user