counterweight
4cb561d54f
- Add create() method to PriceRepository - Update PriceService to use repository.create() instead of direct db operations
73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
"""Price repository for database queries."""
|
|
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import desc, select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from models import PriceHistory
|
|
from price_fetcher import PAIR_BTC_EUR, SOURCE_BITFINEX
|
|
|
|
|
|
class PriceRepository:
|
|
"""Repository for price-related database queries."""
|
|
|
|
def __init__(self, db: AsyncSession):
|
|
self.db = db
|
|
|
|
async def get_latest(
|
|
self, source: str = SOURCE_BITFINEX, pair: str = PAIR_BTC_EUR
|
|
) -> PriceHistory | None:
|
|
"""Get the most recent price from the database."""
|
|
query = (
|
|
select(PriceHistory)
|
|
.where(PriceHistory.source == source, PriceHistory.pair == pair)
|
|
.order_by(desc(PriceHistory.timestamp))
|
|
.limit(1)
|
|
)
|
|
result = await self.db.execute(query)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def get_recent(self, limit: int = 20) -> list[PriceHistory]:
|
|
"""Get the most recent price history records."""
|
|
query = select(PriceHistory).order_by(desc(PriceHistory.timestamp)).limit(limit)
|
|
result = await self.db.execute(query)
|
|
return list(result.scalars().all())
|
|
|
|
async def get_by_timestamp(
|
|
self,
|
|
timestamp: str | datetime,
|
|
source: str = SOURCE_BITFINEX,
|
|
pair: str = PAIR_BTC_EUR,
|
|
) -> PriceHistory | None:
|
|
"""Get a price record by timestamp."""
|
|
# Convert string timestamp to datetime if needed
|
|
timestamp_dt: datetime
|
|
if isinstance(timestamp, str):
|
|
timestamp_dt = datetime.fromisoformat(timestamp.replace("Z", "+00:00"))
|
|
else:
|
|
timestamp_dt = timestamp
|
|
|
|
result = await self.db.execute(
|
|
select(PriceHistory).where(
|
|
PriceHistory.source == source,
|
|
PriceHistory.pair == pair,
|
|
PriceHistory.timestamp == timestamp_dt,
|
|
)
|
|
)
|
|
return result.scalar_one_or_none()
|
|
|
|
async def create(self, record: PriceHistory) -> PriceHistory:
|
|
"""
|
|
Create a new price history record.
|
|
|
|
Args:
|
|
record: PriceHistory instance to persist
|
|
|
|
Returns:
|
|
Created PriceHistory record (refreshed from database)
|
|
"""
|
|
self.db.add(record)
|
|
await self.db.commit()
|
|
await self.db.refresh(record)
|
|
return record
|