add most sorts with defaults

datawhores · Aug 28, 2024 · 7b20ea3 · 7b20ea3
1 parent cdce636
commit 7b20ea3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/ofscraper/commands/managers/db.py b/ofscraper/commands/managers/db.py
@@ -134,7 +134,20 @@ def get_max_post(self):
         self.media=medias
     def sort_media(self):
         medias=self.media
-        medias=sorted(medias,key=lambda x:arrow.get(x["posted_at"]),reverse=True)
+        reversed=not read_args.retriveArgs().db_asc
+        sort=read_args.retriveArgs().db_sort
+        if sort=="posted":
+            medias=sorted(medias,key=lambda x:arrow.get(x["posted_at"] or 0),reverse=reversed)
+        elif sort=="created":
+            medias=sorted(medias,key=lambda x:arrow.get(x["created_at"] or 0),reverse=reversed)
+        elif sort=="filename":
+            medias=sorted(medias,key=lambda x: x["filename"] or "",reverse=reversed)
+        elif sort=="postid":
+            medias=sorted(medias,key=lambda x: x["post_id"],reverse=reversed)
+        elif sort=="mediaid":
+            medias=sorted(medias,key=lambda x: x["media_id"],reverse=reversed)
+        elif sort=="size":
+            medias=sorted(medias,key=lambda x: x["size"] or 0,reverse=reversed)
         self.media=medias
 
 

diff --git a/ofscraper/utils/args/parse/arguments/content.py b/ofscraper/utils/args/parse/arguments/content.py
@@ -296,10 +296,10 @@
 
 db_desc_option = click.option(
     "-bdc",
-    "--db-desc",
+    "--db-asc",
     help=
     """
-    Change the sort order  of table to reverse
+    Change the sort order  of table to ascending
     """,
     is_flag=True,
     default=False,