30 lines
631 B
Python
30 lines
631 B
Python
import datetime
|
|
import logging
|
|
|
|
from src.db import AnalyzeTask, TaskStatus, DBSession
|
|
from src.settings import MESSAGE_TTL, init_logging
|
|
|
|
|
|
REMOVE_MULTIPLIER = 3
|
|
|
|
|
|
def main():
|
|
session = DBSession()
|
|
count = (
|
|
session.query(AnalyzeTask)
|
|
.filter(
|
|
AnalyzeTask.status == TaskStatus.InQueue,
|
|
AnalyzeTask.created_at + REMOVE_MULTIPLIER * MESSAGE_TTL
|
|
< datetime.datetime.now().timestamp(),
|
|
)
|
|
.delete()
|
|
)
|
|
session.commit()
|
|
session.close()
|
|
logging.info(f"deleted {count} old in_queue tasks")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
init_logging()
|
|
main()
|