move the resource-intensive part of "upvotes given" leaderboard to cron

pull/143/head
Aevann 2023-04-25 11:31:02 +02:00
parent 8f7a0bc4e7
commit ac7db0106f
2 changed files with 23 additions and 19 deletions

View File

@ -1,5 +1,4 @@
from typing import Any, Callable, Optional, Tuple, Union
from collections import Counter
from sqlalchemy import Column, func
from sqlalchemy.orm import scoped_session
@ -12,6 +11,7 @@ from .user import User
from .userblock import UserBlock
from .votes import Vote, CommentVote
from files.__main__ import cache
class Leaderboard:
"""
@ -106,21 +106,9 @@ class Leaderboard:
@classmethod
def get_upvotes_lb(cls, lb_criteria, v:User, db:scoped_session, users:Any, limit):
t1 = time.time()
votes1 = db.query(Vote.user_id, func.count(Vote.user_id)).filter(Vote.vote_type==1).group_by(Vote.user_id).order_by(func.count(Vote.user_id).desc()).all()
votes2 = db.query(CommentVote.user_id, func.count(CommentVote.user_id)).filter(CommentVote.vote_type==1).group_by(CommentVote.user_id).order_by(func.count(CommentVote.user_id).desc()).all()
votes3 = Counter(dict(votes1)) + Counter(dict(votes2))
users14 = db.query(User).filter(User.id.in_(votes3.keys())).all()
users13 = []
for user in users14:
users13.append((user.id, votes3[user.id]-user.post_count-user.comment_count))
if not users13: users13 = [(None,None)]
users13 = sorted(users13, key=lambda x: x[1], reverse=True)
users13_1, users13_2 = zip(*users13[:25])
t2 = time.time()
print(f"first block duration: {t2-t1}")
users13 = cache.get("users13") or []
users13_1 = cache.get("users13_1") or []
users13_2 = cache.get("users13_2") or []
users13_accs = db.query(User).filter(User.id.in_(users13_1)).all()
users13_accs = sorted(users13_accs, key=lambda x: users13_1.index(x.id))
@ -130,7 +118,4 @@ class Leaderboard:
pos13 = (pos13+1, users13[pos13][1])
except: pos13 = (len(users13)+1, 0)
t3 = time.time()
print(f"second block duration: {t3-t2}")
return (users13_accs, pos13[0], pos13[1])

View File

@ -4,6 +4,7 @@ import os
from sys import stdout
from shutil import make_archive
from hashlib import md5
from collections import Counter
import click
import requests
@ -40,6 +41,7 @@ def cron(every_5m, every_1h, every_1d, every_1mo):
if every_1h:
awards.award_timers_bots_task()
_generate_emojis_zip()
_leaderboard_task()
if every_1d:
stats.generate_charts_task(SITE)
@ -126,3 +128,20 @@ def _generate_emojis_zip():
size = str(int(os.stat('files/assets/emojis.zip').st_size/1024/1024)) + ' MB'
cache.set('emojis_size', size)
def _leaderboard_task():
votes1 = g.db.query(Vote.user_id, func.count(Vote.user_id)).filter(Vote.vote_type==1).group_by(Vote.user_id).order_by(func.count(Vote.user_id).desc()).all()
votes2 = g.db.query(CommentVote.user_id, func.count(CommentVote.user_id)).filter(CommentVote.vote_type==1).group_by(CommentVote.user_id).order_by(func.count(CommentVote.user_id).desc()).all()
votes3 = Counter(dict(votes1)) + Counter(dict(votes2))
users14 = g.db.query(User).filter(User.id.in_(votes3.keys())).all()
users13 = []
for user in users14:
users13.append((user.id, votes3[user.id]-user.post_count-user.comment_count))
if not users13: users13 = [(None,None)]
users13 = sorted(users13, key=lambda x: x[1], reverse=True)
users13_1, users13_2 = zip(*users13[:25])
cache.set("user13", list(users13))
cache.set("users13_1", list(users13_1))
cache.set("users13_2", list(users13_2))