61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
from typing import Any, AsyncIterator, Dict, Optional
|
|
from uuid import uuid4
|
|
|
|
from fastapi import Depends
|
|
|
|
from freqtrade.enums import RunMode
|
|
from freqtrade.persistence import Trade
|
|
from freqtrade.persistence.models import _request_id_ctx_var
|
|
from freqtrade.rpc.rpc import RPC, RPCException
|
|
|
|
from .webserver import ApiServer
|
|
|
|
|
|
def get_rpc_optional() -> Optional[RPC]:
|
|
if ApiServer._has_rpc:
|
|
return ApiServer._rpc
|
|
return None
|
|
|
|
|
|
async def get_rpc() -> Optional[AsyncIterator[RPC]]:
|
|
|
|
_rpc = get_rpc_optional()
|
|
if _rpc:
|
|
request_id = str(uuid4())
|
|
ctx_token = _request_id_ctx_var.set(request_id)
|
|
Trade.rollback()
|
|
try:
|
|
yield _rpc
|
|
finally:
|
|
Trade.session.remove()
|
|
_request_id_ctx_var.reset(ctx_token)
|
|
|
|
else:
|
|
raise RPCException('Bot is not in the correct state')
|
|
|
|
|
|
def get_config() -> Dict[str, Any]:
|
|
return ApiServer._config
|
|
|
|
|
|
def get_api_config() -> Dict[str, Any]:
|
|
return ApiServer._config['api_server']
|
|
|
|
|
|
def get_exchange(config=Depends(get_config)):
|
|
if not ApiServer._exchange:
|
|
from freqtrade.resolvers import ExchangeResolver
|
|
ApiServer._exchange = ExchangeResolver.load_exchange(
|
|
config['exchange']['name'], config, load_leverage_tiers=False)
|
|
return ApiServer._exchange
|
|
|
|
|
|
def get_message_stream():
|
|
return ApiServer._message_stream
|
|
|
|
|
|
def is_webserver_mode(config=Depends(get_config)):
|
|
if config['runmode'] != RunMode.WEBSERVER:
|
|
raise RPCException('Bot is not in the correct state')
|
|
return None
|