优化日志推送逻辑，简化代码并修复日期判断

logging_helper.py

@@ -75,35 +75,20 @@ class LoggingHelper:
 
     def push_yesterday(self) -> bool:
         try:
+            print("[push_yesterday] Pushing yesterday's log files to remote")
             year = self.today.year.__str__()
             month = self.today.month.__str__()
             day = self.today.day.__str__()
             remotedir = "/".join([year, month, day])
-            files_to_push = []
-            for filename in self.buffer.keys():
-                if not filename.startswith(remotedir):
-                    continue
-                files_to_push.append(filename)
-            if len(files_to_push) == 0:
-                return True
-            print("[push_yesterday] Writing datasets to json files")
-            for filename in files_to_push:
-                localpath = os.sep.join([self.local_dir, filename])
-                self.buffer[filename].to_json(localpath)
-                files_to_push.append(filename)
-                print(f"[push_yesterday] {filename} finished")
-            print("[push_yesterday] Done")
-            print("[push_yesterday] Pushing log files to remote")
-            if len(files_to_push):
-                localdir = os.sep.join([self.local_dir, remotedir])
-                res = hf.upload_folder(
-                    repo_id=self.repo_id,
-                    folder_path=localdir,
-                    path_in_repo=remotedir,
-                    repo_type=self.repo_type,
-                    commit_message=f"Updated at {beijing()}",
-                )
-                print(f"[push_yesterday] Log files pushed to {res}")
+            localdir = os.sep.join([self.local_dir, remotedir])
+            res = hf.upload_folder(
+                repo_id=self.repo_id,
+                folder_path=localdir,
+                path_in_repo=remotedir,
+                repo_type=self.repo_type,
+                commit_message=f"Updated at {beijing()}",
+            )
+            print(f"[push_yesterday] Log files pushed to {res}")
             print("[push_yesterday] Done")
             return True
         except Exception as e:
@@ -112,22 +97,23 @@ class LoggingHelper:
 
     def push(self):
         try:
-            now = beijing().date()
-            if now != self.today:  # new day comes
-                if not self.push_yesterday():
-                    print("[push] Failed to upload yesterday's log files")
-                self.today = now
             files_to_push = [
                 filename for filename in self.need_push if self.need_push[filename]
             ]
-            if len(files_to_push) == 0:
-                return True
             print("[push] Writing datasets to json files")
             for filename in files_to_push:
                 localpath = os.path.join(self.local_dir, filename)
                 self.buffer[filename].to_json(localpath)
                 print(f"[push] {filename} finished")
-            print("[push] Done")
+                self.need_push[filename] = False
+            now = beijing().date()
+            if len(files_to_push) == 0:
+                print("[push] Done")
+                return True
+            if now != self.today:  # new day comes
+                if not self.push_yesterday():
+                    print("[push] Failed to upload yesterday's log files")
+                self.today = now
             print("[push] Pushing log files to remote")
             remotedir = self.remotedir()
             localdir = "/".join([self.local_dir, remotedir])
@@ -138,8 +124,6 @@ class LoggingHelper:
                 repo_type=self.repo_type,
                 commit_message=f"Updated at {beijing()}",
             )
-            for filename in files_to_push:
-                self.need_push[filename] = False
             print(f"[push] Log files pushed to {res}")
             print("[push] Done")
             return True