49 lines
1.4 KiB
Python
49 lines
1.4 KiB
Python
import asyncio
|
|
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from asgiref.sync import sync_to_async
|
|
from django.core.management.base import BaseCommand
|
|
|
|
from core.models import Strategy
|
|
from core.trading import active_management
|
|
from core.util import logs
|
|
|
|
log = logs.get_logger("scheduling")
|
|
|
|
INTERVAL = 5
|
|
|
|
|
|
async def job():
|
|
"""
|
|
Run all schedules matching the given interval.
|
|
:param interval_seconds: The interval to run.
|
|
"""
|
|
strategies = await sync_to_async(list)(
|
|
Strategy.objects.filter(enabled=True, active_management_enabled=True)
|
|
)
|
|
log.debug(f"Found {len(strategies)} strategies")
|
|
for strategy in strategies:
|
|
log.debug(f"Running strategy {strategy.name}")
|
|
ams = active_management.ActiveManagement(strategy) # noqa
|
|
ams.run_checks()
|
|
ams.execute_actions()
|
|
|
|
|
|
class Command(BaseCommand):
|
|
def handle(self, *args, **options):
|
|
"""
|
|
Start the scheduling process.
|
|
"""
|
|
scheduler = AsyncIOScheduler()
|
|
|
|
log.debug(f"Scheduling checking process job every {INTERVAL} seconds")
|
|
scheduler.add_job(job, "interval", seconds=INTERVAL)
|
|
scheduler.start()
|
|
loop = asyncio.get_event_loop()
|
|
try:
|
|
loop.run_forever()
|
|
except (KeyboardInterrupt, SystemExit):
|
|
log.info("Process terminating")
|
|
finally:
|
|
loop.close()
|