MarseyWorld/files/helpers/sorting_and_time.py

import time
from typing import Optional

from sqlalchemy.sql import func

from files.helpers.config.const import *

def apply_time_filter(t, objects, cls):
	now = int(time.time())
	if t == 'hour':
		cutoff = now - 3600
	elif t == 'day':
		cutoff = now - 86400
	elif t == 'week':
		cutoff = now - 604800
	elif t == 'month':
		cutoff = now - 2592000
	elif t == 'year':
		cutoff = now - 31536000
	else:
		cutoff = 0

	return objects.filter(cls.created_utc >= cutoff)

def sort_objects(sort, objects, cls):
	if sort == 'hot':
		ti = int(time.time()) + 3600
		metric = cls.realupvotes
		if cls.__name__ == "Submission": metric += cls.comment_count/5
		return objects.order_by(-1000000*(metric + 1)/(func.power(((ti - cls.created_utc)/1000), 1.23)), cls.created_utc.desc())
	elif sort == "bump" and cls.__name__ == "Submission":
		return objects.filter(cls.comment_count > 1).order_by(cls.bump_utc.desc(), cls.created_utc.desc())
	elif sort == "comments" and cls.__name__ == "Submission":
		return objects.order_by(cls.comment_count.desc(), cls.created_utc.desc())
	elif sort == "new":
		return objects.order_by(cls.created_utc.desc())
	elif sort == "old":
		return objects.order_by(cls.created_utc)
	elif sort == "controversial":
		return objects.order_by((cls.upvotes+1)/(cls.downvotes+1) + (cls.downvotes+1)/(cls.upvotes+1), cls.downvotes.desc(), cls.created_utc.desc())
	elif sort == "bottom":
		return objects.order_by(cls.upvotes - cls.downvotes, cls.created_utc.desc())
	else:
		return objects.order_by(cls.downvotes - cls.upvotes, cls.created_utc.desc())

def make_age_string(compare:Optional[int]) -> str:
	if not compare or compare < 1577865600: return ""
	age = int(time.time()) - compare

	if age < 60:
		return "just now"
	elif age < 3600:
		minutes = int(age / 60)
		return f"{minutes}m ago"
	elif age < 86400:
		hours = int(age / 3600)
		return f"{hours}hr ago"
	elif age < 2678400:
		days = int(age / 86400)
		return f"{days}d ago"

	now = time.gmtime()
	ctd = time.gmtime(compare)
	months = now.tm_mon - ctd.tm_mon + 12 * (now.tm_year - ctd.tm_year)
	if now.tm_mday < ctd.tm_mday:
		months -= 1
	if months < 12:
		return f"{months}mo ago"
	else:
		years = int(months / 12)
		return f"{years}yr ago"