edit : 원격 저장으로 변경

custom_apps/bingart/bingart.py

@@ -6,14 +6,16 @@ from bingart import BingArt
 from custom_apps.utils import cookie_manager
 from main_rest.app.utils.parsing_utils import prompt_to_filenames
 from main_rest.app.utils.date_utils import D
-from const import OUTPUT_FOLDER
+from const import REMOTE_FOLDER, TEMP_FOLDER
+from utils.custom_sftp import sftp_client
 
 
 class BingArtGenerator:
     
     model = 'dalle3'
     detail = 'art'
-    output_folder = os.path.join(OUTPUT_FOLDER,"dalle","art")
+    output_folder = os.path.join(TEMP_FOLDER,"dalle","art")
+    remote_folder = os.path.join(REMOTE_FOLDER,"dalle","art")
 
     def __init__(self):
         self.bing_art = BingArt(auth_cookie_U=cookie_manager.get_cookie())
@@ -67,6 +69,7 @@ class BingArtGenerator:
             path= f"{self.model}_{self.detail}_{self.prompt}_{jpeg_index}_{self.datetime}.png"
 
             urllib.request.urlretrieve(i, os.path.join(self.output_folder,path))
+            sftp_client.remote_copy_data(os.path.join(self.output_folder,path), os.path.join(self.remote_folder,path))
             jpeg_index += 1
             
     def get_images(self, prompt, image_len):