From 16832c9bf8ab14b2d712125bba28a5003995a06b Mon Sep 17 00:00:00 2001
From: anima <p.schreiner@4nima.de>
Date: Sun, 22 Sep 2024 00:42:08 +0200
Subject: [PATCH] add download game

---
 SatiAPI.py | 36 +++++++++++++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 5 deletions(-)

diff --git a/SatiAPI.py b/SatiAPI.py
index cb86679..247f963 100644
--- a/SatiAPI.py
+++ b/SatiAPI.py
@@ -9,6 +9,8 @@ import logging
 import json
 import yaml
 from os.path import exists
+from os import makedirs
+
 
 import urllib3
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
@@ -16,7 +18,7 @@ urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 class SatiAPI:
     """A API wrapper for Satisfactory dedicated server"""
     __AUTHOR__      = 'anima'
-    __VERSION__     = '0.15.0'
+    __VERSION__     = '0.16.1'
 
     def __init__(self, host: str = None, port: int = 7777, token: str = None, conffile: str = 'conf.yml', logfile: str = 'SatiAPI.log', loglevel: int = 20) -> None:
         """create a wrapper for satisfactory dedicated server
@@ -188,7 +190,11 @@ class SatiAPI:
 
         response = requests.post(f'https://{self.host}:{self.port}/api/v1', headers=headers, json=payload, verify=False)
         if response.status_code == 200:
-            json_response = json.loads(response.text)
+            if response.headers['Content-Type'] == 'application/json':
+                json_response = json.loads(response.text)
+            else:
+                return response.content
+
             if 'data' in json_response:
                 return json_response
             else:
@@ -455,6 +461,27 @@ class SatiAPI:
 
         return self.__query('SaveGame', data, True)
 
+    def download_game(self, savename: str = None, savepath: str = 'saves/') -> bool:
+        """Downloads save game with the given name from the Dedicated Server. Requires Admin privileges. This function responds with the file attachment containing the save game file on success, and with normal error response in case of error.
+
+        :param savename: Name of the save game file to download from the Dedicated Server. Default: Sessionname_APISave (will be create before)
+        :type savename: string
+        :param savepath: path to save the savefile
+        :type savepath: sting
+        :return: true if save successfull downloaded
+        :rtype: bool
+        """
+        if savename is None: 
+            pass
+
+        data = dict()
+        data['SaveName'] = savename
+        response = self.__query('DownloadSaveGame', data, True)
+        if not exists(savepath):
+            makedirs(savepath)
+        with open(f'{savepath}/{savename}.sav', 'wb') as f:
+            f.write(response)
+
     def load_game(self, savename: str, enableAdvancedSettings: bool = True) -> bool:
         """Loads the save game file by name, optionally with Advanced Game Settings enabled. Requires Admin privileges. Dedicated Server HTTPS API will become temporarily unavailable when save game is being loaded. Function does not return any data on succcess.
         
@@ -473,7 +500,6 @@ class SatiAPI:
 
         return False
 
-
     def get_server_options(self) -> dict:
         """Retrieves currently applied server options and server options that are still pending application (because of needing session or server restart) Does not require input parameters.
         
@@ -499,7 +525,6 @@ class SatiAPI:
     #ApplyServerOptions
     #ApplyAdvancedGameSettings
     
-    #DownloadSaveGame
     #UploadSaveGame
     
     #DeleteSaveFile
@@ -513,4 +538,5 @@ if __name__ == "__main__":
     print()
     print(sati.get_sessions())
     print()
-    print(sati.load_game('MyWorld_APISave'))
\ No newline at end of file
+    print(sati.download_game('MyWorld_APISave'))
+    
\ No newline at end of file