forked from rDrama/rDrama
262 lines
7.6 KiB
Python
262 lines
7.6 KiB
Python
from files.helpers.wrappers import *
|
|
from files.helpers.get import *
|
|
from files.helpers.const import *
|
|
from files.helpers.sorting_and_time import *
|
|
from files.__main__ import app, cache, limiter
|
|
from files.classes.submission import Submission
|
|
from files.helpers.awards import award_timers
|
|
|
|
@app.get("/")
|
|
@app.get("/catalog")
|
|
@app.get("/h/<sub>")
|
|
@app.get("/s/<sub>")
|
|
@app.get("/logged_out")
|
|
@app.get("/logged_out/catalog")
|
|
@app.get("/logged_out/h/<sub>")
|
|
@app.get("/logged_out/s/<sub>")
|
|
@limiter.limit("3/second;30/minute;5000/hour;10000/day")
|
|
@auth_desired_with_logingate
|
|
def front_all(v, sub=None, subdomain=None):
|
|
#### WPD TEMP #### special front logic
|
|
from files.helpers.security import generate_hash, validate_hash
|
|
from datetime import datetime
|
|
now = datetime.utcnow()
|
|
if request.host == 'watchpeopledie.co':
|
|
if v and not v.admin_level: # security: don't auto login admins
|
|
hash = generate_hash(f'{v.id}+{now.year}+{now.month}+{now.day}+{now.hour}+WPDusermigration')
|
|
return redirect(f'https://watchpeopledie.tv/?user={v.id}&code={hash}', 301)
|
|
else:
|
|
return render_template('wpdco.html')
|
|
elif request.host == 'watchpeopledie.tv' and not v: # security: don't try to login people into accounts more than once
|
|
req_user = request.values.get('user')
|
|
req_code = request.values.get('code')
|
|
if req_user and req_code:
|
|
from routes.login import on_login
|
|
user = get_account(req_user)
|
|
if user:
|
|
if user.admin_level:
|
|
abort(401)
|
|
else:
|
|
if validate_hash(req_code, f'{v.id}+{now.year}+{now.month}+{now.day}+{now.hour}+WPDusermigration'):
|
|
on_login(user)
|
|
#### WPD TEMP #### end special front logic
|
|
if sub:
|
|
sub = sub.strip().lower()
|
|
if sub == 'chudrama' and not (v and v.can_see_chudrama): abort(403)
|
|
sub = get_sub_by_name(sub, graceful=True)
|
|
|
|
if (request.path.startswith('/h/') or request.path.startswith('/s/')) and not sub: abort(404)
|
|
|
|
try: page = max(int(request.values.get("page", 1)), 1)
|
|
except: abort(400)
|
|
|
|
if v:
|
|
defaultsorting = v.defaultsorting
|
|
if sub or SITE_NAME != 'rDrama': defaulttime = 'all'
|
|
else: defaulttime = v.defaulttime
|
|
else:
|
|
defaultsorting = "hot"
|
|
if sub or SITE_NAME != 'rDrama': defaulttime = 'all'
|
|
else: defaulttime = DEFAULT_TIME_FILTER
|
|
|
|
sort=request.values.get("sort", defaultsorting)
|
|
t=request.values.get('t', defaulttime)
|
|
ccmode=request.values.get('ccmode', "false").lower()
|
|
|
|
try: gt=int(request.values.get("after", 0))
|
|
except: gt=0
|
|
|
|
try: lt=int(request.values.get("before", 0))
|
|
except: lt=0
|
|
|
|
if sort == 'hot': default = True
|
|
else: default = False
|
|
|
|
pins = session.get(sort, default)
|
|
|
|
ids, next_exists = frontlist(sort=sort,
|
|
page=page,
|
|
t=t,
|
|
v=v,
|
|
ccmode=ccmode,
|
|
filter_words=v.filter_words if v else [],
|
|
gt=gt,
|
|
lt=lt,
|
|
sub=sub,
|
|
site=SITE,
|
|
pins=pins
|
|
)
|
|
|
|
posts = get_posts(ids, v=v)
|
|
|
|
if v:
|
|
if v.hidevotedon: posts = [x for x in posts if not hasattr(x, 'voted') or not x.voted]
|
|
award_timers(v)
|
|
|
|
if v and v.client: return {"data": [x.json for x in posts], "next_exists": next_exists}
|
|
return render_template("home.html", v=v, listing=posts, next_exists=next_exists, sort=sort, t=t, page=page, ccmode=ccmode, sub=sub, home=True, pins=pins)
|
|
|
|
|
|
|
|
@cache.memoize(timeout=86400)
|
|
def frontlist(v=None, sort="hot", page=1, t="all", ids_only=True, ccmode="false", filter_words='', gt=0, lt=0, sub=None, site=None, pins=True):
|
|
|
|
posts = g.db.query(Submission)
|
|
|
|
if v and v.hidevotedon:
|
|
voted = [x[0] for x in g.db.query(Vote.submission_id).filter_by(user_id=v.id).all()]
|
|
posts = posts.filter(Submission.id.notin_(voted))
|
|
|
|
if sub: posts = posts.filter_by(sub=sub.name)
|
|
elif v: posts = posts.filter(or_(Submission.sub == None, Submission.sub.notin_(v.all_blocks)))
|
|
|
|
if gt: posts = posts.filter(Submission.created_utc > gt)
|
|
if lt: posts = posts.filter(Submission.created_utc < lt)
|
|
|
|
if not gt and not lt:
|
|
posts = apply_time_filter(t, posts, Submission)
|
|
|
|
if (ccmode == "true") and FEATURES['COUNTRY_CLUB']:
|
|
posts = posts.filter(Submission.club == True)
|
|
|
|
posts = posts.filter_by(is_banned=False, private=False, deleted_utc = 0)
|
|
|
|
if pins and ccmode == "false" and not gt and not lt:
|
|
if sub: posts = posts.filter_by(hole_pinned=None)
|
|
else: posts = posts.filter_by(stickied=None)
|
|
|
|
if v:
|
|
posts = posts.filter(Submission.author_id.notin_(v.userblocks))
|
|
|
|
if v and filter_words:
|
|
for word in filter_words:
|
|
word = word.replace('\\', '').replace('_', '\_').replace('%', '\%').strip()
|
|
posts=posts.filter(not_(Submission.title.ilike(f'%{word}%')))
|
|
|
|
posts = sort_objects(sort, posts, Submission,
|
|
include_shadowbanned=(v and v.can_see_shadowbanned))
|
|
|
|
if v: size = v.frontsize or 0
|
|
else: size = 25
|
|
|
|
posts = posts.offset(size * (page - 1)).limit(size+1).all()
|
|
|
|
next_exists = (len(posts) > size)
|
|
|
|
posts = posts[:size]
|
|
|
|
if pins and page == 1 and ccmode == "false" and not gt and not lt:
|
|
if sub:
|
|
pins = g.db.query(Submission).filter(Submission.sub == sub.name, Submission.hole_pinned != None)
|
|
else:
|
|
pins = g.db.query(Submission).filter(Submission.stickied != None, Submission.is_banned == False)
|
|
|
|
if v:
|
|
pins = pins.filter(or_(Submission.sub == None, Submission.sub.notin_(v.all_blocks)))
|
|
for pin in pins:
|
|
if pin.stickied_utc and int(time.time()) > pin.stickied_utc:
|
|
pin.stickied = None
|
|
pin.stickied_utc = None
|
|
g.db.add(pin)
|
|
|
|
|
|
if v: pins = pins.filter(Submission.author_id.notin_(v.userblocks))
|
|
|
|
pins = pins.order_by(Submission.created_utc.desc()).all()
|
|
|
|
posts = pins + posts
|
|
|
|
if ids_only: posts = [x.id for x in posts]
|
|
|
|
|
|
return posts, next_exists
|
|
|
|
|
|
@app.get("/random_post")
|
|
@auth_required
|
|
def random_post(v):
|
|
|
|
p = g.db.query(Submission.id).filter(Submission.deleted_utc == 0, Submission.is_banned == False, Submission.private == False).order_by(func.random()).first()
|
|
|
|
if p: p = p[0]
|
|
else: abort(404)
|
|
|
|
return redirect(f"/post/{p}")
|
|
|
|
|
|
@app.get("/random_user")
|
|
@auth_required
|
|
def random_user(v):
|
|
u = g.db.query(User.username).filter(User.song != None, User.shadowbanned == None).order_by(func.random()).first()
|
|
|
|
if u: u = u[0]
|
|
else: return "No users have set a profile anthem so far!"
|
|
|
|
return redirect(f"/@{u}")
|
|
|
|
|
|
@app.get("/comments")
|
|
@auth_required
|
|
def all_comments(v):
|
|
|
|
|
|
try: page = max(int(request.values.get("page", 1)), 1)
|
|
except: page = 1
|
|
|
|
sort=request.values.get("sort", "new")
|
|
t=request.values.get("t", DEFAULT_TIME_FILTER)
|
|
|
|
try: gt=int(request.values.get("after", 0))
|
|
except: gt=0
|
|
|
|
try: lt=int(request.values.get("before", 0))
|
|
except: lt=0
|
|
|
|
idlist = comment_idlist(v=v,
|
|
page=page,
|
|
sort=sort,
|
|
t=t,
|
|
gt=gt,
|
|
lt=lt,
|
|
site=SITE
|
|
)
|
|
|
|
comments = get_comments(idlist, v=v)
|
|
|
|
next_exists = len(idlist) > 25
|
|
|
|
idlist = idlist[:25]
|
|
|
|
if v.client: return {"data": [x.json for x in comments]}
|
|
return render_template("home_comments.html", v=v, sort=sort, t=t, page=page, comments=comments, standalone=True, next_exists=next_exists)
|
|
|
|
|
|
@cache.memoize(timeout=86400)
|
|
def comment_idlist(v=None, page=1, sort="new", t="all", gt=0, lt=0, site=None):
|
|
comments = g.db.query(Comment.id) \
|
|
.join(Comment.post) \
|
|
.filter(Comment.parent_submission != None)
|
|
|
|
if v.admin_level < PERMS['POST_COMMENT_MODERATION']:
|
|
comments = comments.filter(
|
|
Comment.is_banned == False,
|
|
Comment.deleted_utc == 0,
|
|
Submission.private == False,
|
|
Comment.author_id.notin_(v.userblocks),
|
|
)
|
|
|
|
if not v.paid_dues:
|
|
comments = comments.filter(Submission.club == False)
|
|
|
|
if gt: comments = comments.filter(Comment.created_utc > gt)
|
|
if lt: comments = comments.filter(Comment.created_utc < lt)
|
|
|
|
if not gt and not lt:
|
|
comments = apply_time_filter(t, comments, Comment)
|
|
|
|
comments = sort_objects(sort, comments, Comment,
|
|
include_shadowbanned=(v and v.can_see_shadowbanned))
|
|
|
|
comments = comments.offset(25 * (page - 1)).limit(26).all()
|
|
return [x[0] for x in comments]
|