mirror of
https://github.com/jfmcbrayer/brutaldon
synced 2024-12-23 21:51:43 +01:00
Refactor some more common stuff for views
This commit is contained in:
parent
e6f2734208
commit
51cb1c42fe
@ -5,17 +5,20 @@ from brutaldon.models import Client, Account
|
|||||||
from mastodon import Mastodon
|
from mastodon import Mastodon
|
||||||
from urllib import parse
|
from urllib import parse
|
||||||
|
|
||||||
def timeline(request, timeline='home', timeline_name='Home'):
|
class NotLoggedInException(Exception):
|
||||||
|
pass
|
||||||
|
|
||||||
|
def get_mastodon(request):
|
||||||
if not (request.session.has_key('instance') and
|
if not (request.session.has_key('instance') and
|
||||||
request.session.has_key('username')):
|
request.session.has_key('username')):
|
||||||
return redirect(login)
|
raise NotLoggedInException()
|
||||||
|
|
||||||
try:
|
try:
|
||||||
client = Client.objects.get(api_base_id=request.session['instance'])
|
client = Client.objects.get(api_base_id=request.session['instance'])
|
||||||
user = Account.objects.get(username=request.session['username'])
|
user = Account.objects.get(username=request.session['username'])
|
||||||
except (Client.DoesNotExist, Client.MultipleObjectsReturned,
|
except (Client.DoesNotExist, Client.MultipleObjectsReturned,
|
||||||
Account.DoesNotExist, Account.MultipleObjectsReturned):
|
Account.DoesNotExist, Account.MultipleObjectsReturned):
|
||||||
return redirect(login)
|
raise NotLoggedInException()
|
||||||
|
|
||||||
mastodon = Mastodon(
|
mastodon = Mastodon(
|
||||||
client_id = client.client_id,
|
client_id = client.client_id,
|
||||||
@ -23,6 +26,13 @@ def timeline(request, timeline='home', timeline_name='Home'):
|
|||||||
access_token = user.access_token,
|
access_token = user.access_token,
|
||||||
api_base_url = client.api_base_id,
|
api_base_url = client.api_base_id,
|
||||||
ratelimit_method="pace")
|
ratelimit_method="pace")
|
||||||
|
return mastodon
|
||||||
|
|
||||||
|
def timeline(request, timeline='home', timeline_name='Home'):
|
||||||
|
try:
|
||||||
|
mastodon = get_mastodon(request)
|
||||||
|
except NotLoggedInException:
|
||||||
|
return redirect(login)
|
||||||
data = mastodon.timeline(timeline)
|
data = mastodon.timeline(timeline)
|
||||||
|
|
||||||
if request.session.has_key('fullbrutalism'):
|
if request.session.has_key('fullbrutalism'):
|
||||||
|
Loading…
Reference in New Issue
Block a user