Upload main.py

main.py

@@ -104,8 +104,9 @@ class CivitAICrawler:
 
     def encrypt_with_rclone(self, local_path: str):
         """
-        指定ファイル or ディレクトリを cryptLocal: にコピー。
-        フォルダ構造やファイル名を rclone の filename_encryption 設定に応じて暗号化する。
+        local_path がファイルかフォルダかで処理を分け、
+        - フォルダ → 従来通り rclone copy
+        - ファイル → rclone copyto を使って暗号化
         """
         if not os.path.exists(local_path):
             raise FileNotFoundError(f"[ERROR] Local path not found: {local_path}")
@@ -114,18 +115,36 @@ class CivitAICrawler:
         if os.path.isdir(self.config.ENCRYPTED_DIR):
             shutil.rmtree(self.config.ENCRYPTED_DIR, ignore_errors=True)
 
-        top_level_name = os.path.basename(local_path.rstrip("/"))
-        if not top_level_name:
-            top_level_name = "unnamed"
-
-        cmd = ["rclone", "copy", local_path, f"cryptLocal:{top_level_name}", "-v"]
-        logger.info(f"[INFO] Running: {' '.join(cmd)}")
-        subprocess.run(cmd, check=True)
-        logger.info(f"[OK] rclone copy => cryptLocal:{top_level_name}")
-
+        # ========= 1) local_path がフォルダの場合 =========
+        if os.path.isdir(local_path):
+            top_level_name = os.path.basename(local_path.rstrip("/")) or "unnamed"
+            cmd = [
+                "rclone", "copy",
+                local_path,
+                f"cryptLocal:{top_level_name}",
+                "-v"
+            ]
+            logger.info(f"[INFO] Running: {' '.join(cmd)}")
+            subprocess.run(cmd, check=True)
+
+        # ========= 2) local_path がファイルの場合 =============
+        else:
+            filename = os.path.basename(local_path)
+            # "rclone copyto <file> cryptLocal:<same_filename>" にすると
+            # cryptLocal 側では1ファイルとして扱われる
+            cmd = [
+                "rclone", "copyto",
+                local_path,
+                f"cryptLocal:{filename}",
+                "-v"
+            ]
+            logger.info(f"[INFO] Running: {' '.join(cmd)}")
+            subprocess.run(cmd, check=True)
+
+        logger.info(f"[OK] rclone => cryptLocal:")
         if not os.path.isdir(self.config.ENCRYPTED_DIR):
             raise FileNotFoundError(
-                f"[ERROR] {self.config.ENCRYPTED_DIR} not found. Check your rclone config."
+                f"[ERROR] {self.config.ENCRYPTED_DIR} not found. Check rclone config."
             )
 
     # 例: upload_encrypted_files の中の再試行処理