mirror of
https://github.com/Faraphel/Atlas-Install.git
synced 2025-07-02 02:38:30 +02:00
27 lines
830 B
Python
27 lines
830 B
Python
from io import BytesIO
|
|
from typing import IO
|
|
|
|
from source.mkw.Patch.PatchOperation import AbstractPatchOperation
|
|
from source.wt import bmg
|
|
|
|
Patch: any
|
|
|
|
|
|
class BmgEncoder(AbstractPatchOperation):
|
|
"""
|
|
encode a bmg file to a txt file
|
|
"""
|
|
|
|
type = "bmg-encode"
|
|
|
|
def patch(self, patch: "Patch", file_name: str, file_content: IO) -> (str, IO):
|
|
"""
|
|
Patch a file to encode it to a bmg file
|
|
:param patch: the patch that is applied
|
|
:param file_name: the file_name of the file
|
|
:param file_content: the content of the file
|
|
:return: the new name and new content of the file
|
|
"""
|
|
patched_file_name = file_name.rsplit(".", 1)[0]
|
|
patch_content = BytesIO(bmg.encode_data(file_content.read().decode("utf-8")))
|
|
return patched_file_name, patch_content
|