]>
git.gir.st - subscriptionfeed.git/blob - app/browse/__init__.py
2 from flask
import Blueprint
, render_template
, request
, flash
, g
, url_for
, redirect
3 from flask_login
import current_user
4 from werkzeug
.exceptions
import BadRequest
, NotFound
6 from ..common
.common
import *
7 from ..common
.innertube
import *
9 from .protobuf
import make_sp
, make_channel_params
, make_playlist_params
, Filters
11 frontend
= Blueprint('browse', __name__
,
12 template_folder
='templates',
13 static_folder
='static',
14 static_url_path
='/static/ys')
16 @frontend.route('/results')
17 @frontend.route('/search')
19 #token = getattr(current_user, 'token', 'guest')
20 q
= request
.args
.get('q') or request
.args
.get('search_query')
21 page
= int(request
.args
.get('page') or 1)
23 sp
= make_sp(page
, **{
24 k
:v
for k
,v
in request
.args
.items()
25 if k
in ['sort','date','type','len']
27 f
for f
in request
.args
.getlist('feature')
28 if f
in Filters
.__dataclass
_fields
__.keys()
30 e
for e
in request
.args
.getlist('feature')
35 yt_results
= fetch_searchresults(q
, sp
)
37 results
, extras
= prepare_searchresults(yt_results
)
44 return render_template('search.html.j2', rows
=results
, query
=q
, page
=page
)
46 @frontend.route('/channel/<channel_id>/')
47 @frontend.route('/channel/<channel_id>/<subpage>')
48 def channel(channel_id
, subpage
="videos"):
49 token
= getattr(current_user
, 'token', 'guest')
50 if subpage
== "videos":
51 page
= int(request
.args
.get('page') or 1)
52 sort_by
= request
.args
.get('sort') or "newest"
54 elif subpage
== "playlists":
55 page
= int(request
.args
.get('page') or 1)
56 sort_by
= request
.args
.get('sort', "modified")
58 elif subpage
== "search":
59 query
= request
.args
.get('q')
60 page
= int(request
.args
.get('page') or 1)
62 else: # we don't support /home, /about, ..., so redirect to /videos.
63 return redirect(url_for('.channel', channel_id
=channel_id
))
65 # best effort; if it fails, it fails in the redirect.
66 if not re
.match(r
"(UC[A-Za-z0-9_-]{22})", channel_id
):
67 return redirect(url_for('.channel_redirect', user
=channel_id
))
69 result
= fetch_ajax(make_channel_params(channel_id
, subpage
, page
, sort_by
, query
, v3
=(subpage
!= "search")))
70 error
= find_and_parse_error(result
)
72 if result
is None: # if fetching from innertube failed, fall back to xmlfeed:
73 flash("unable to fetch results from ajax; displaying fallback results (15 newest)", "error")
74 return fallback_route(channel_id
, subpage
)
77 return error
, 400 # todo: ugly
79 title
, descr
, thumb
, rows
, more
= prepare_channel(result
, channel_id
)
80 if not rows
: # overran end of list, or is special channel (e.g. music topic)
81 flash("ajax returned nothing; displaying fallback results (15 newest)", "error")
82 return fallback_route(channel_id
, subpage
)
84 # set pin/hide stati of retrieved videos:
85 video_ids
= [card
['content']['video_id'] for card
in rows
]
86 pinned
, hidden
= fetch_video_flags(token
, video_ids
)
88 {'type':v
['type'], 'content':{**v
['content'], 'pinned': v
['content']['video_id'] in pinned
}}
90 if v
['content']['video_id'] not in hidden
91 ], key
=lambda v
:v
['content']['pinned'], reverse
=True)
93 with sqlite3
.connect(cf
['global']['database']) as conn
:
98 WHERE channel_id = ? AND user = ?
99 """, (channel_id
, token
))
100 (is_subscribed
,) = c
.fetchone()
102 return render_template('channel.html.j2',
106 channel_id
=channel_id
,
109 is_subscribed
=is_subscribed
,
113 @frontend.route('/user/<user>/')
114 @frontend.route('/user/<user>/<subpage>')
115 @frontend.route('/c/<user>/')
116 @frontend.route('/c/<user>/<subpage>')
117 def channel_redirect(user
, subpage
=None):
119 The browse_ajax 'API' needs the UCID.
122 # inverse of the test in /channel/:
123 if re
.match(r
"(UC[A-Za-z0-9_-]{22})", user
):
124 return redirect(url_for('.channel', channel_id
=user
))
126 channel_id
= canonicalize_channel(user
)
128 raise NotFound("channel appears to not exist")
130 url_for('.channel', channel_id
=channel_id
, subpage
=subpage
), 308
133 @frontend.route('/playlist')
135 playlist_id
= request
.args
.get('list')
137 raise BadRequest("No playlist ID")
138 page
= int(request
.args
.get('page', 1))
140 xmlfeed
= fetch_xml("playlist_id", playlist_id
)
142 raise NotFound("Unable to fetch playlist")
143 title
, author
, _
, channel_id
, _
= parse_xml(xmlfeed
)
145 offset
= (page
-1)*100 # each call returns 100 items
146 result
= fetch_ajax(make_playlist_params(playlist_id
, offset
))
147 error
= find_and_parse_error(result
)
150 flash(f
"1 {error}. Loading fallback.", 'error')
151 return fallback_route()
153 if not 'continuationContents' in result
:
154 flash(f
"2 {error}. Loading fallback.", 'error')
155 return fallback_route()
157 rows
, more
= prepare_playlist(result
)
159 return render_template('playlist.html.j2',
162 channel_id
=channel_id
,
167 @frontend.route('/<something>', strict_slashes
=False)
168 def plain_user_or_video(something
):
169 # this is a near-copy of the same route in app/youtube, but using a
170 # different, more reliable endpoint to determine whether a channel exists.
172 # prevent a lot of false-positives (and reduce youtube api calls)
175 channel_id
= canonicalize_channel(something
)
177 return redirect(url_for('.channel', channel_id
=channel_id
))
178 elif re
.match(r
"^[-_0-9A-Za-z]{11}$", something
): # looks like a video id
179 return redirect(url_for('youtube.watch', v
=something
, t
=request
.args
.get('t')))
181 raise NotFound("Note: some usernames not recognized; try searching it")
183 @frontend.before_app_request
185 if not 'header_items' in g
:
187 g
.header_items
.append({
189 'url': url_for('browse.search'),
190 'parent': frontend
.name
,