From b615a5ed23b4d75ee9638462f0054043daa115d7 Mon Sep 17 00:00:00 2001
From: SouthFox <master@southfox.me>
Date: Tue, 8 Nov 2022 18:59:46 +0800
Subject: [PATCH] feat/archive bookmarks and favourites

---
 BDSM/toot.py | 45 +++++++++++++++++++++++++--------------------
 1 file changed, 25 insertions(+), 20 deletions(-)

diff --git a/BDSM/toot.py b/BDSM/toot.py
index 3a6dab9..e1a7f19 100644
--- a/BDSM/toot.py
+++ b/BDSM/toot.py
@@ -222,27 +222,32 @@ def toot_process(statuses, my_acct, duplicates_counter=0):
 def archive_toot(url):
     mastodon, user = app_login(url)
     acct = mastodon.me().acct
+
+    def archive(statuses, skip_duplicates=True):
+        happy_counter = 20
+        duplicates_counter = 0
+        while(True):
+            duplicates_counter = toot_process(statuses, acct)
+            db.session.commit()
+            print(str(happy_counter) + ' / '  + statuses_count)
+            happy_counter += 20
+
+            if duplicates_counter >= 10 and skip_duplicates:
+                print("检测到重复嘟文达到十次，取消存档……")
+                break
+
+            statuses = mastodon.fetch_next(statuses)
+
+            if statuses == None:
+                break
+
     statuses_count = str(mastodon.me().statuses_count)
-
     statuses = mastodon.account_statuses(user["id"], limit=20)
-    # xx = statuses['created_at'].astimezone(tz_cn)
-    # pprint(xx.strftime("%m/%d/%Y, %H:%M:%S"))
-    # pprint(statuses)
+    archive(statuses)
 
-    happy_counter = 20
-    duplicates_counter = 0
+    statuses_count = '???'
+    statuses = mastodon.favourites()
+    archive(statuses)
 
-    while(True):
-        duplicates_counter = toot_process(statuses, acct)
-        db.session.commit()
-        print(str(happy_counter) + ' / '  + statuses_count)
-        happy_counter += 20
-
-        if duplicates_counter >= 10:
-            print("检测到重复嘟文达到十次，取消存档……")
-            break
-
-        statuses = mastodon.fetch_next(statuses)
-        # statuses = None
-        if statuses == None:
-            break
+    statuses = mastodon.bookmarks()
+    archive(statuses, skip_duplicates=False)