import re import time import sqlite3 import requests #from flask_login import current_user, login_required from flask_login import LoginManager, UserMixin, current_user, login_user, logout_user, login_required from flask import Blueprint, render_template, request, redirect, flash, url_for, jsonify, g from ..common.common import * frontend = Blueprint('reddit', __name__, template_folder='templates', static_folder='static', static_url_path='/static/rd') @frontend.route('/feed/subreddits') # disabled for guest user: @login_required def feed(): return "reddit feed: TODO" if current_user.is_anonymous: token = 'guest' else: token = current_user.token page = int(request.args.get('page', 0)) with sqlite3.connect(cf['global']['database']) as conn: c = conn.cursor() c.execute(""" SELECT videos.id, channel_id, name, title, published, flags.display FROM videos JOIN channels ON videos.channel_id = channels.id LEFT JOIN flags ON (videos.id = flags.video_id) AND (flags.user = ?) WHERE channel_id IN (SELECT channel_id FROM subscriptions WHERE user = ?) AND flags.display IS NOT 'hidden' ORDER BY (display = 'pinned') DESC, crawled DESC LIMIT 36 OFFSET 36*?""", (token, token, page)) rows = [{ 'video_id': video_id, 'channel_id': channel_id, 'author': author, 'title': title, 'published': published, 'pinned': display == 'pinned', } for (video_id, channel_id, author, title, published, display) in c.fetchall()] return render_template('index.html.j2', rows=rows, page=page) @frontend.route('/manage/subreddits') # disabled for guest user: @login_required def subscription_manager(): return "reddit feed: TODO" if current_user.is_anonymous: token = 'guest' else: token = current_user.token with sqlite3.connect(cf['global']['database']) as conn: #with conn.cursor() as c: c = conn.cursor() c.execute(""" SELECT subscriptions.channel_id, name, (subscribed_until < datetime('now')) AS obsolete FROM subscriptions left JOIN channels ON channels.id = subscriptions.channel_id left JOIN websub ON channels.id = websub.channel_id WHERE user = ? AND subscriptions.type IN ('channel', 'playlist') ORDER BY obsolete=0, name COLLATE NOCASE ASC""", (token,)) rows = [{ 'channel_id': channel_id, 'author': author or channel_id, 'subscribed_until': subscribed_until } for (channel_id, author, subscribed_until) in c.fetchall()] return render_template('subscription_manager.html.j2', rows=rows) @frontend.route('/feed/subreddits', methods=['POST']) @login_required def feed_post(): return "reddit feed: TODO" token = current_user.token action = next(request.form.keys(), None) if action in ['pin', 'unpin', 'hide']: video_id = request.form.get(action) display = { 'pin': 'pinned', 'unpin': None, 'hide': 'hidden', }[action] with sqlite3.connect(cf['global']['database']) as conn: #with conn.cursor() as c: c = conn.cursor() c.execute(""" INSERT OR REPLACE INTO flags (user, video_id, display) VALUES (?, ?, ?) """, (token, video_id, display)) else: flash("unsupported action", "error") return redirect(request.url, code=303) @frontend.route('/manage/subreddits', methods=['POST']) @login_required def manage_subscriptions(): return "reddit feed: TODO" token = current_user.token if 'subscribe' in request.form: channel_id = request.form.get("subscribe") match = re.search(r"(UC[A-Za-z0-9_-]{22})", channel_id) if match: channel_id = match.group(1) else: match = re.search(r"((?:PL|LL|EC|UU|FL|UL|OL)[A-Za-z0-9_-]{10,})", channel_id) if match: # NOTE: PL-playlists are 32chars, others differ in length. flash("playlists not (yet?) supported.", "error") return redirect(request.url, code=303) # TODO: dedup redirection else: flash("not a valid/subscribable URI", "error") return redirect(request.url, code=303) # TODO: dedup redirection with sqlite3.connect(cf['global']['database']) as conn: #with conn.cursor() as c: c = conn.cursor() c.execute(""" INSERT OR IGNORE INTO subscriptions (user, channel_id) VALUES (?, ?) """, (token, channel_id)) # TODO: sql-error-handling, asynchronically calling update-subs.pl elif 'unsubscribe' in request.form: channel_id = request.form.get("unsubscribe") with sqlite3.connect(cf['global']['database']) as conn: #with conn.cursor() as c: c = conn.cursor() c.execute(""" DELETE FROM subscriptions WHERE user = ? AND channel_id = ? """, (token, channel_id)) # TODO: sql-error-handling, report success else: flash("unsupported action", "error") return redirect(request.url, code=303) @frontend.route('/r/') def reddit_index(): return "" @frontend.route('/r/') def reddit(subreddit="videos"): count = int(request.args.get('count', 0)) before = request.args.get('before') after = request.args.get('after') query = '&'.join([f"{k}={v}" for k,v in [('count',count), ('before',before), ('after',after)] if v]) r = requests.get(f"https://old.reddit.com/r/{subreddit}.json?{query}", headers={'User-Agent':'Mozilla/5.0'}) if not r.ok or not 'data' in r.json(): return r.text+"error retrieving reddit data", 502 good = [e for e in r.json()['data']['children'] if e['data']['score'] > 1] bad = [e for e in r.json()['data']['children'] if e['data']['score'] <=1] videos = [] for entry in (good+bad): e = entry['data'] if e['domain'] not in ['youtube.com', 'youtu.be', 'invidio.us']: continue video_id = re.match(r'^https?://(?:www.|m.)?(?:youtube.com/watch\?(?:.*&)?v=|youtu.be/|youtube.com/embed/)([-_0-9A-Za-z]+)', e['url']).group(1) if not video_id: continue videos.append({ 'video_id': video_id, 'title': e['title'], 'url': e['permalink'], 'n_comments': e['num_comments'], 'n_karma': e['score'], }) before = r.json()['data']['before'] after = r.json()['data']['after'] return render_template('reddit.html.j2', subreddit=subreddit, rows=videos, before=before, after=after, count=count)