Commit 08812a66 authored by 陈涛's avatar 陈涛

修改scheduler获取

parent e10b6aea
...@@ -6,8 +6,6 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler ...@@ -6,8 +6,6 @@ from apscheduler.schedulers.asyncio import AsyncIOScheduler
from fastapi import APIRouter, Depends, Query, Request from fastapi import APIRouter, Depends, Query, Request
from motor.core import AgnosticCollection from motor.core import AgnosticCollection
from pymongo import ReturnDocument from pymongo import ReturnDocument
from starlette.background import BackgroundTasks
from exception.db import NotFundError from exception.db import NotFundError
from model import Response, PageResponse, Page from model import Response, PageResponse, Page
from model.fund import FundType, StakingFund, NormalFund, FundStatus from model.fund import FundType, StakingFund, NormalFund, FundStatus
...@@ -28,19 +26,17 @@ fund_type_map = { ...@@ -28,19 +26,17 @@ fund_type_map = {
summary='创建基金', summary='创建基金',
description='创建基金') description='创建基金')
async def create( async def create(
request: Request,
create_fund: CreateFund, create_fund: CreateFund,
user: User = Depends(get_current_user), user: User = Depends(get_current_user),
fund_collect: AgnosticCollection = Depends(get_fund_collect) fund_collect: AgnosticCollection = Depends(get_fund_collect),
scheduler: AsyncIOScheduler = Depends(get_scheduler)
): ):
create_model = fund_type_map[create_fund.fund_type](**create_fund.dict(), nodes=[], **user.db_save()) create_model = fund_type_map[create_fund.fund_type](**create_fund.dict(), nodes=[], **user.db_save())
create_model.nav = create_model.base_nav create_model.nav = create_model.base_nav
data = create_model.dict() data = create_model.dict()
response_model = fund_type_map[data['fund_type']] response_model = fund_type_map[data['fund_type']]
await fund_collect.insert_one(data) await fund_collect.insert_one(data)
scheduler = request.app.state.scheduler
job_id = f"calculate_nav_{data['id']}" job_id = f"calculate_nav_{data['id']}"
time_obj = datetime.datetime.strptime(data["settlement_time"], "%H:%M") time_obj = datetime.datetime.strptime(data["settlement_time"], "%H:%M")
scheduler.add_job( scheduler.add_job(
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment