fix(storage): handle null values in file sorting to prevent crashes

jxxghp · Sep 25, 2024 · 3512e7d · 3512e7d
1 parent 5b1d111
commit 3512e7d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/app/api/endpoints/storage.py b/app/api/endpoints/storage.py
@@ -1,3 +1,4 @@
+from datetime import datetime
 from pathlib import Path
 from typing import Any, List
 
@@ -68,9 +69,9 @@ def list(fileitem: schemas.FileItem,
     file_list = StorageChain().list_files(fileitem)
     if file_list:
         if sort == "name":
-            file_list.sort(key=lambda x: x.name)
+            file_list.sort(key=lambda x: x.name or "")
         else:
-            file_list.sort(key=lambda x: x.modify_time, reverse=True)
+            file_list.sort(key=lambda x: x.modify_time or datetime.min, reverse=True)
     return file_list