Update payment redirect and scheduler job for deleting document id

This commit is contained in:
jetli 2025-04-20 00:05:00 -07:00
parent 21b49df49c
commit b95a818e30
3 changed files with 16 additions and 15 deletions

View File

@ -1,6 +1,7 @@
from common.config.app_settings import app_settings
from backend.content.models import DocumentDoc
from backend.document.models import BasicProfileDoc
from datetime import datetime, timezone
import httpx
import base64
@ -53,9 +54,12 @@ class DocumentManager:
print(f"Failed to queue deletion: {response.text}")
async def cleanup_document(self):
# Get today's date at midnight (UTC)
today_start = datetime.now(timezone.utc).replace(hour=0, minute=0, second=0, microsecond=0)
# Corrected query with regex
documents = await DocumentDoc.find(
{"created_by": {"$regex": "^content-service-"}}
{"created_by": {"$regex": "^content-service-"},
"create_time": {"$lt": today_start}}
).to_list()
if documents:

View File

@ -1,5 +1,6 @@
from apscheduler.schedulers.asyncio import AsyncIOScheduler
from apscheduler.triggers.date import DateTrigger
from apscheduler.triggers.cron import CronTrigger
from datetime import datetime, timedelta
from scheduler.refresh_sharepoint_content_job import (
refresh_sharepoint_content_job,
@ -28,15 +29,11 @@ async def register_job(scheduler: AsyncIOScheduler):
"date",
run_date=datetime(2025, 2, 7, 20, 0, 0),
)
# Register cleanup_document_job as a one-time job
# This job is just one-time job for removing many unused documents
# Run already, now comment it out
# await init_lock(ScheduleJobLocker.CLEANUP_DOCUMENT_JOB_LOCKER)
# execution_time = datetime.now() + timedelta(
# seconds=60
# ) # Schedule to run 60 seconds from now
# scheduler.add_job(
# cleanup_document_job, # Job function
# trigger=DateTrigger(run_date=execution_time), # One-time trigger
# id="cleanup_document_one_time", # Optional: Give the job an ID
# )
await init_lock(ScheduleJobLocker.CLEANUP_DOCUMENT_JOB_LOCKER)
scheduler.add_job(
cleanup_document_job,
trigger=CronTrigger(hour=2, minute=0), # Runs every day at 2:00 AM
id="cleanup_document_daily",
)

View File

@ -292,10 +292,10 @@ class StripeManager:
},
},
mode="payment",
success_url="{}/work-space".format(
success_url="{}/projects".format(
self.site_url_root
), # needs to be set, local: http://localhost/
cancel_url="{}/work-space".format(self.site_url_root),
cancel_url="{}/projects".format(self.site_url_root),
)
if session: