app/reddit/__init__.py

   1 import re
   2 import sqlite3
   3 from flask_login import current_user, login_required
   4 from flask import Blueprint, render_template, request, redirect, flash, url_for
   5
   6 from ..common.common import *
   7 from .lib import *
   8
   9 frontend = Blueprint('reddit', __name__,
  10     template_folder='templates',
  11     static_folder='static',
  12     static_url_path='/static/rd')
  13
  14 @frontend.route('/feed/subreddits')
  15 @frontend.route('/r/<subreddit>/')
  16 def reddit(subreddit=None):
  17     token = getattr(current_user, 'token', 'guest')
  18     after = request.args.get('after')
  19
  20     sortorder = request.args.get('s', "hot") # TODO: verify!
  21     timerange = request.args.get('t', None) # TODO: verify!
  22
  23     all_subreddits = get_subreddits(token)
  24     subreddits = [subreddit] if subreddit else all_subreddits
  25
  26     if subreddits:
  27         try:
  28             data = fetch_reddit(subreddits,
  29                     sorted_by=sortorder, time=timerange, limit=36, after=after)
  30             videos = parse_reddit_videos(data)
  31             after = data['data']['after']
  32         except RedditException as e:
  33             return f"error retrieving reddit data: <xmp>{e}</xmp>", 502 # TODO: better
  34
  35         # set pin/hide stati of retrieved videos
  36         video_ids = [v['video_id'] for v in videos]
  37         pinned, hidden = fetch_video_flags(token, video_ids)
  38         videos = sorted([
  39             {**v, 'pinned': v['video_id'] in pinned}
  40             for v in videos
  41             if v['video_id'] not in hidden
  42         ], key=lambda v:v['pinned'], reverse=True)
  43     else: # not subscribed to anything
  44         videos = []
  45
  46     title = f"/r/{subreddit}" if subreddit else "my subreddits"
  47     return render_template('reddit.html.j2', title=title, rows=videos,
  48         subreddits=all_subreddits, after=after)
  49
  50 @frontend.route('/manage/subreddits')
  51 # disabled for guest user: @login_required
  52 def subscription_manager():
  53     token = getattr(current_user, 'token', 'guest')
  54     subreddits = get_subreddits(token)
  55     return render_template('subreddit_manager.html.j2', subreddits=subreddits)
  56
  57 @frontend.route('/manage/subreddits', methods=['POST'])
  58 @login_required
  59 def manage_subscriptions():
  60     token = current_user.token
  61     if 'subscribe' in request.form:
  62         subreddit = request.form.get("subscribe")
  63         match = re.search(r"(?:(?:https?://)?(?:old.|www.|\w\w.)?reddit.com)?(?:/?r/)?([-+_0-9A-Za-z]{2,21})", subreddit)
  64         if match:
  65             subreddit = match.group(1)
  66         else:
  67             flash("invalid subreddit", "error")
  68             return redirect(request.url, code=303)
  69         with sqlite3.connect(cf['global']['database']) as conn:
  70             c = conn.cursor()
  71             c.execute("""
  72                         INSERT OR IGNORE INTO subreddits (user, subreddit)
  73                         VALUES (?, ?)
  74             """, (token, subreddit))
  75
  76     elif 'unsubscribe' in request.form:
  77         subreddit = request.form.get("unsubscribe")
  78         with sqlite3.connect(cf['global']['database']) as conn:
  79             c = conn.cursor()
  80             c.execute("""
  81                         DELETE FROM subreddits
  82                         WHERE user = ? AND subreddit = ?
  83             """, (token, subreddit))
  84             # TODO: sql-error-handling, report success
  85
  86     else:
  87         flash("unsupported action", "error")
  88
  89     return redirect(request.url, code=303)
  90
  91 def get_subreddits(token):
  92     with sqlite3.connect(cf['global']['database']) as conn:
  93         c = conn.cursor()
  94         c.execute("""
  95           SELECT subreddit
  96             FROM subreddits
  97            WHERE user = ?
  98         ORDER BY subreddit COLLATE NOCASE ASC
  99         """, (token,))
 100         subreddits = [sr for (sr,) in c.fetchall()]
 101     return subreddits
 102
 103 @frontend.app_template_filter('trim3')
 104 def trim3(n):
 105     if type(n) != int:
 106         return n # not a number
 107     elif round(n, 1) >= 10_000:
 108         return "%.0fk" % (n/1000)
 109     elif n >= 1_000:
 110         return "%.1fk" % (n/1000)
 111     else:
 112         return "%d" % n
 113
 114 @frontend.before_app_request
 115 def inject_reddit_button():
 116     if not 'header_items' in g:
 117         g.header_items = []
 118     g.header_items.append({
 119         'name': 'reddit',
 120         'url': url_for('reddit.reddit'),
 121         'parent': frontend.name,
 122         'priority': 5
 123     })