From e6f2734208e23b938d7eb7fcc9ce92714f6708be Mon Sep 17 00:00:00 2001
From: Jason McBrayer <jmcbray@carcosa.net>
Date: Wed, 25 Apr 2018 20:06:10 -0400
Subject: [PATCH] Add the remaining "easy" timeline views

---
 brutaldon/views.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/brutaldon/views.py b/brutaldon/views.py
index 7263a60..6f52fe1 100644
--- a/brutaldon/views.py
+++ b/brutaldon/views.py
@@ -3,11 +3,9 @@ from django.shortcuts import render, redirect
 from brutaldon.forms import LoginForm, SettingsForm
 from brutaldon.models import Client, Account
 from mastodon import Mastodon
-import datetime
 from urllib import parse
 
-def home(request):
-    now = datetime.datetime.now()
+def timeline(request, timeline='home', timeline_name='Home'):
     if not (request.session.has_key('instance') and
             request.session.has_key('username')):
         return redirect(login)
@@ -25,7 +23,7 @@ def home(request):
         access_token = user.access_token,
         api_base_url = client.api_base_id,
         ratelimit_method="pace")
-    data = mastodon.timeline()
+    data = mastodon.timeline(timeline)
 
     if request.session.has_key('fullbrutalism'):
         fullbrutalism = request.session['fullbrutalism']
@@ -33,9 +31,17 @@ def home(request):
         fullbrutalism = False
 
     return render(request, 'main/timeline.html',
-                  {'toots': data, 'timeline': 'Home',
+                  {'toots': data, 'timeline': timeline_name,
                    'fullbrutalism': fullbrutalism})
 
+def home(request):
+    return timeline(request, 'home', 'Home')
+
+def local(request):
+    return timeline(request, 'local', 'Local')
+
+def fed(request):
+    return timeline(request, 'public', 'Federated')
 
 def login(request):
     if request.method == "GET":
@@ -99,12 +105,6 @@ def error(request):
 def note(request):
     return render(request, 'main/timeline.html', {'timeline': 'Notifications'})
 
-def local(request):
-    return render(request, 'main/timeline.html', {'timeline': 'Local'})
-
-def fed(request):
-    return render(request, 'main/timeline.html', {'timeline': 'Federated'})
-
 
 def settings(request):
     if request.method == 'POST':