ff7dc0c95a
feat(frontend): integrate Prisma and reports API/pages chore(config): add data_sources.yaml; update analysis-config.json docs: add 2025-11-03 dev log; update user guide scripts: enhance dev.sh; add tushare_legacy_client deps: update backend and frontend dependencies
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
import sys
|
|
import os
|
|
import asyncio
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
# Add backend to path to import TushareProvider
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "..", "backend"))
|
|
from app.data_providers.tushare import TushareProvider
|
|
|
|
class TushareLegacyClient:
|
|
"""
|
|
An adapter to mimic the old TushareClient for legacy scripts,
|
|
but uses the new TushareProvider under the hood.
|
|
"""
|
|
def __init__(self, token: str):
|
|
if not token:
|
|
raise ValueError("Token must be provided.")
|
|
self.provider = TushareProvider(token=token)
|
|
|
|
async def query(
|
|
self,
|
|
api_name: str,
|
|
params: Optional[Dict[str, Any]] = None,
|
|
fields: Optional[str] = None, # Note: fields are not used in the new provider's _query
|
|
) -> List[Dict[str, Any]]:
|
|
"""
|
|
Mimics the .query() method by calling the provider's internal _query method.
|
|
"""
|
|
# The new _query method is protected, but we call it here for the script's sake.
|
|
return await self.provider._query(api_name=api_name, params=params, fields=fields)
|
|
|
|
async def aclose(self):
|
|
"""Mimic aclose to allow 'async with' syntax."""
|
|
if hasattr(self.provider, '_client') and self.provider._client:
|
|
await self.provider._client.aclose()
|
|
|
|
async def __aenter__(self):
|
|
return self
|
|
|
|
async def __aexit__(self, exc_type, exc, tb):
|
|
await self.aclose()
|