Refactored some stuff

extensions/saves.py

@@ -1,64 +1,17 @@
 from datetime import datetime
-from io import BytesIO
 from urllib.parse import quote_plus
-from os import makedirs, path, remove
-from typing import Dict, List, Tuple, Union
-from uuid import uuid4
-from zipfile import ZipFile, ZIP_DEFLATED
+from os import remove
+from typing import Dict, List, Union
 from xmltodict import parse
 from models.saves import StardewSave
 from modules.app import app, get_api_key, user_by_key
-from modules.utils import configGet, saveFile
 from modules.database import col_devices, col_saves
 from fastapi import HTTPException, Depends, UploadFile
 from fastapi.responses import UJSONResponse, FileResponse, Response
 from fastapi.openapi.models import APIKey
 from starlette.status import HTTP_204_NO_CONTENT, HTTP_404_NOT_FOUND, HTTP_406_NOT_ACCEPTABLE
 
-def zip_saves(save_filename: str, save_bytes: bytes, saveinfo_bytes: bytes) -> Tuple[str, str]:
-
-    save_uuid = str(uuid4())
-
-    makedirs(path.join(configGet("data", "locations"), "files", save_uuid))
-
-    #saveFile(save_bytes, filename=save_filename, dirname=zipname)
-    #saveFile(saveinfo_bytes, filename="SaveGameInfo", dirname=zipname)
-
-    with ZipFile(path.join(configGet("data", "locations"), "files", save_uuid, save_filename+".svsave"), 'w', ZIP_DEFLATED, compresslevel=configGet("compression")) as ziph:
-        ziph.writestr("SaveGameInfo", saveinfo_bytes)
-        ziph.writestr(save_filename, save_bytes)
-    
-    return save_uuid, path.join(configGet("data", "locations"), "files", save_uuid, save_filename+".svsave")
-
-
-def zipfiles(filenames, save_name: str) -> Response:
-
-    zip_filename = save_name+".svsave"
-
-    s = BytesIO()
-    zf = ZipFile(s, "w")
-
-    for fpath in filenames:
-
-        # Calculate path for file in zip
-        fdir, fname = path.split(fpath)
-
-        # Add file, at correct path
-        for entry in (list(col_saves.find({"files.save.uuid": fname})) + list(col_saves.find({"files.saveinfo.uuid": fname}))):
-            filename = entry["file"]["save"]["name"] if (entry["file"]["save"]["uuid"] == fname) else entry["file"]["saveinfo"]["name"]
-        zf.write(fpath, filename)
-
-    # Must close zip for all contents to be written
-    zf.close()
-
-    # Grab ZIP file from in-memory, make response with correct MIME-type
-    return Response(
-        s.getvalue(),
-        media_type="application/x-zip-compressed",
-        headers={
-            'Content-Disposition': f'attachment;filename={quote_plus(zip_filename)}'
-        }
-    )
+from modules.utils import zip_saves
 
 
 @app.get("/saves", response_class=UJSONResponse, response_model=Dict[str, StardewSave], description="Get all available game saves")
@@ -149,7 +102,6 @@ async def saves_download(id: int, save_date: int, device: Union[str, None] = Non
             }
         )
         return response
-        # return zipfiles([saves_entry["file"]["save"]["path"], saves_entry["file"]["saveinfo"]["path"]], save_name=f'{saves_entry["data"]["farmer"]}_{saves_entry["id"]}')
     else:
         raise HTTPException(HTTP_404_NOT_FOUND, detail="Could not find save with such id.")
 
@@ -204,7 +156,8 @@ async def saves_post(device: str, files: List[UploadFile], apikey: APIKey = Depe
             "save_time": int(save_info["Farmer"]["saveTime"]),
             "year": int(save_info["Farmer"]["yearForSaveGame"]),
             "season": int(save_info["Farmer"]["seasonForSaveGame"]),
-            "day": int(save_info["Farmer"]["dayOfMonthForSaveGame"])
+            "day": int(save_info["Farmer"]["dayOfMonthForSaveGame"]),
+            "game_version": save_info["Farmer"]["gameVersion"]
         },
         "file": {
             "name": save_data_filename,