rDrama/files/classes/leaderboard.py

from sqlalchemy import Column, func
from flask import g

from files.helpers.config.const import *

from .badges import Badge
from .emoji import *
from .user import User
from .userblock import UserBlock
from .votes import Vote, CommentVote
from .casino_game import CasinoGame
from .post import Post

from files.__main__ import cache

class Leaderboard:
	"""
	Represents an request-context leaderboard. None of this is persisted yet,
	although this is probably a good idea to do at some point.
	"""
	all_users = None
	v_position = 0
	v_value = None
	v_appears_in_ranking = False
	user_func = None
	value_func = None

	def __init__(self, header_name, table_header_name, html_id, table_column_name,
				user_relative_url, query_function, criteria, v, value_func, users, limit=LEADERBOARD_LIMIT, desc=True):
		self.header_name = header_name
		self.table_header_name = table_header_name
		self.html_id = html_id
		self.table_column_name = table_column_name
		self.user_relative_url = user_relative_url
		self.limit = limit
		lb = query_function(criteria, v, users, limit, desc)
		self.all_users = lb[0]
		self.v_position = lb[1]
		self.v_value = lb[2]
		self.v_appears_in_ranking = self.v_position and self.v_position <= len(self.all_users)
		if value_func:
			self.user_func = lambda u:u
			self.value_func = value_func
			self.v_value = value_func(v)
		else:
			self.user_func = lambda u:u[0]
			self.value_func = lambda u: u[1] or 0
		self.desc = desc

	@classmethod
	def get_simple_lb(cls, order_by, v, users, limit, desc):
		leaderboard = users.order_by(order_by.desc()).limit(limit).all()
		position = None
		if v not in leaderboard:
			sq = g.db.query(User.id, func.rank().over(order_by=order_by.desc()).label("rank")).subquery()
			position = g.db.query(sq.c.id, sq.c.rank).filter(sq.c.id == v.id).limit(1).one()[1]
		return (leaderboard, position, None)


	@classmethod
	def count_and_label(cls, criteria):
		return func.count(criteria).label("count")

	@classmethod
	def sum_and_label(cls, criteria):
		return func.sum(criteria).label("sum")

	@classmethod
	def avg_and_label(cls, criteria1, criteria2):
		return (func.sum(criteria1)/func.count(criteria2)).label("avg")


	@classmethod
	def rank_filtered_rank_label_by_desc(cls, criteria):
		return func.rank().over(order_by=func.count(criteria).desc()).label("rank")

	@classmethod
	def rank_filtered_rank_label_by_desc_sum(cls, criteria):
		return func.rank().over(order_by=func.sum(criteria).desc()).label("rank")

	@classmethod
	def rank_filtered_rank_label_by_asc_sum(cls, criteria):
		return func.rank().over(order_by=func.sum(criteria)).label("rank")

	@classmethod
	def rank_filtered_rank_label_by_desc_avg(cls, criteria1, criteria2):
		return func.rank().over(order_by=-func.sum(criteria1)/func.count(criteria2)).label("rank")


	@classmethod
	def get_badge_emoji_lb(cls, lb_criteria, v, users, limit, desc):
		sq = g.db.query(lb_criteria, cls.count_and_label(lb_criteria), cls.rank_filtered_rank_label_by_desc(lb_criteria))
		if lb_criteria == Emoji.author_id:
			sq = sq.filter(Emoji.author_id != 2, Emoji.kind.in_(["Marsey", "Platy", "Wolf", "Capy", "Carp", "Marsey Flags", "Marsey Alphabet"]))
		sq = sq.group_by(lb_criteria).subquery()

		sq_criteria = None
		if lb_criteria == Badge.user_id:
			sq_criteria = User.id == sq.c.user_id
		elif lb_criteria == Emoji.author_id:
			sq_criteria = User.id == sq.c.author_id
		else:
			raise ValueError("This leaderboard function only supports Badge.user_id and Emoji.author_id")

		leaderboard = g.db.query(User, sq.c.count).join(sq, sq_criteria).order_by(sq.c.count.desc())
		position = g.db.query(User.id, sq.c.rank, sq.c.count).join(sq, sq_criteria).filter(User.id == v.id).one_or_none()
		if position: position = (position[1], position[2])
		else: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])

	@classmethod
	def get_winnings_lb(cls, lb_criteria, v, users, limit, desc):
		if lb_criteria != CasinoGame.winnings:
			raise ValueError("This leaderboard function only supports CasinoGame.user_id")

		if desc: fn = cls.rank_filtered_rank_label_by_desc_sum
		else: fn = cls.rank_filtered_rank_label_by_asc_sum
		sq = g.db.query(CasinoGame.user_id, cls.sum_and_label(lb_criteria), fn(lb_criteria))
		sq = sq.group_by(CasinoGame.user_id).subquery()

		sq_criteria = User.id == sq.c.user_id

		if desc: order = sq.c.sum.desc()
		else: order = sq.c.sum
		leaderboard = g.db.query(User, sq.c.sum).join(sq, sq_criteria).order_by(order)
		position = g.db.query(User.id, sq.c.rank, sq.c.sum).join(sq, sq_criteria).filter(User.id == v.id).one_or_none()
		if position: position = (position[1], position[2])
		else: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])

	@classmethod
	def get_blockers_lb(cls, lb_criteria, v, users, limit, desc):
		if lb_criteria != UserBlock.target_id:
			raise ValueError("This leaderboard function only supports UserBlock.target_id")
		sq = g.db.query(lb_criteria, cls.count_and_label(lb_criteria)).group_by(lb_criteria).subquery()
		leaderboard = g.db.query(User, sq.c.count).join(User, User.id == sq.c.target_id).order_by(sq.c.count.desc())

		sq = g.db.query(lb_criteria, cls.count_and_label(lb_criteria), cls.rank_filtered_rank_label_by_desc(lb_criteria)).group_by(lb_criteria).subquery()
		position = g.db.query(sq.c.rank, sq.c.count).join(User, User.id == sq.c.target_id).filter(sq.c.target_id == v.id).limit(1).one_or_none()
		if not position: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])

	@classmethod
	def get_hat_lb(cls, lb_criteria, v, users, limit, desc):
		leaderboard = g.db.query(User, func.count(lb_criteria)).join(lb_criteria).group_by(User).order_by(func.count(lb_criteria).desc())
		sq = g.db.query(User.id, cls.count_and_label(lb_criteria), cls.rank_filtered_rank_label_by_desc(lb_criteria)).join(lb_criteria).group_by(User).subquery()
		position = g.db.query(sq.c.rank, sq.c.count).filter(sq.c.id == v.id).limit(1).one_or_none()
		if not position: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])

	@classmethod
	def get_upvotes_lb(cls, lb_criteria, v, users, limit, desc):
		users13 = cache.get("users13") or []
		users13_1 = cache.get("users13_1") or []
		users13_2 = cache.get("users13_2") or []

		users13_accs = g.db.query(User).filter(User.id.in_(users13_1)).all()
		users13_accs = sorted(users13_accs, key=lambda x: users13_1.index(x.id))
		users13_accs = tuple(zip(users13_accs, users13_2))
		try:
			pos13 = [x[0] for x in users13].index(v.id)
			pos13 = (pos13+1, users13[pos13][1])
		except: pos13 = (len(users13)+1, 0)

		return (users13_accs, pos13[0], pos13[1])

	@classmethod
	def get_downvotes_lb(cls, lb_criteria, v, users, limit, desc):
		users9 = cache.get("users9") or []
		users9_1 = cache.get("users9_1") or []
		users9_2 = cache.get("users9_2") or []

		users9_accs = g.db.query(User).filter(User.id.in_(users9_1)).all()
		users9_accs = sorted(users9_accs, key=lambda x: users9_1.index(x.id))
		users9_accs = tuple(zip(users9_accs, users9_2))
		try:
			pos9 = [x[0] for x in users9].index(v.id)
			pos9 = (pos9+1, users9[pos9][1])
		except: pos9 = (len(users9)+1, 0)

		return (users9_accs, pos9[0], pos9[1])

	@classmethod
	def get_avg_upvotes_lb(cls, lb_criteria, v, users, limit, desc):
		if lb_criteria == Post:
			minimum = 10
		else:
			minimum = 100

		sq = g.db.query(lb_criteria.author_id, cls.avg_and_label(lb_criteria.upvotes, lb_criteria.author_id)).filter_by(deleted_utc=0)
		if cls == Post:
			sq = sq.filter_by(draft=False)
		sq = sq.group_by(lb_criteria.author_id).having(func.count(lb_criteria.author_id) >= minimum).subquery()
		leaderboard = g.db.query(User, sq.c.avg).join(User, User.id == sq.c.author_id).order_by(sq.c.avg.desc())

		sq = g.db.query(lb_criteria.author_id, cls.avg_and_label(lb_criteria.upvotes, lb_criteria.author_id), cls.rank_filtered_rank_label_by_desc_avg(lb_criteria.upvotes, lb_criteria.author_id)).filter_by(deleted_utc=0)
		if cls == Post:
			sq = sq.filter_by(draft=False)
		sq = sq.group_by(lb_criteria.author_id).having(func.count(lb_criteria.author_id) >= minimum).subquery()
		position = g.db.query(sq.c.rank, sq.c.avg).join(User, User.id == sq.c.author_id).filter(sq.c.author_id == v.id).limit(1).one_or_none()

		if not position: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])

	@classmethod
	def get_effortposts_lb(cls, lb_criteria, v, users, limit, desc):
		sq = g.db.query(lb_criteria, cls.count_and_label(lb_criteria), cls.rank_filtered_rank_label_by_desc(lb_criteria))
		sq = sq.filter(Post.effortpost == True)
		sq = sq.group_by(lb_criteria).subquery()
		sq_criteria = User.id == sq.c.author_id

		leaderboard = g.db.query(User, sq.c.count).join(sq, sq_criteria).order_by(sq.c.count.desc())
		position = g.db.query(User.id, sq.c.rank, sq.c.count).join(sq, sq_criteria).filter(User.id == v.id).one_or_none()
		if position: position = (position[1], position[2])
		else: position = (leaderboard.count() + 1, 0)
		leaderboard = leaderboard.limit(limit).all()
		return (leaderboard, position[0], position[1])