From 6dfc26e6ba3f8cae1d2c7e34584e08705e32d614 Mon Sep 17 00:00:00 2001
From: SouthFox <master@southfox.me>
Date: Mon, 19 Sep 2022 22:44:22 +0800
Subject: [PATCH] feat/show reblog

---
 BDSM/templates/view.html |  3 +++
 BDSM/views.py            | 17 ++++++++++++++---
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/BDSM/templates/view.html b/BDSM/templates/view.html
index f7d82bb..8478ede 100644
--- a/BDSM/templates/view.html
+++ b/BDSM/templates/view.html
@@ -5,6 +5,9 @@
     <div class="toot">
         <div class="status">
             <div class="meta">
+                {% if toot.is_reblog%}
+                <small><span>转嘟了<i class="fa-solid fa-retweet"></i></span></small>
+                {% endif %}
                 <strong><span class ="username">{{ toot.acct }}</span></strong>
                 <a href="{{ toot.url }}" target="_blank"
                     rel="noopener noreferrer">
diff --git a/BDSM/views.py b/BDSM/views.py
index 41174e9..ce9b145 100644
--- a/BDSM/views.py
+++ b/BDSM/views.py
@@ -3,7 +3,7 @@ import os
 
 from flask import render_template, request, url_for, redirect, flash
 from BDSM import app, db
-from BDSM.models import Media, Settings, Toot, Emoji
+from BDSM.models import Media, Settings, Toot, Emoji, Reblog
 from BDSM.toot import app_register, archive_toot
 from mastodon import Mastodon
 from types import SimpleNamespace
@@ -16,8 +16,19 @@ def index():
 
     for toot_ in toots_.items:
         toot = SimpleNamespace(**toot_.__dict__)
-        if toot.content == None:
-            continue
+        print(toot.reblog_id)
+        if toot.reblog_id != None:
+            if toot.reblog_myself:
+                toot = Toot.query.get(toot.reblog_id)
+                toot = SimpleNamespace(**toot.__dict__)
+                toot.is_reblog = True
+            else:
+                print('---\n\n\n----')
+                toot = Reblog.query.get(toot.reblog_id)
+                toot = SimpleNamespace(**toot.__dict__)
+                toot.is_reblog = True
+
+        print(toot)
 
         if toot.media_list != "":
             toot.medias = []