Ran python linting, created pyproject

Author: Mindstormman06

.github/workflows/ci.yaml

@@ -0,0 +1,34 @@
+[project]
+name = "totk-vscode-python"
+version = "0.1.0"
+description = "Python scripts for TOTK VSCode extension"
+requires-python = ">=3.9"
+dependencies = [
+    "oead>=1.2.9.post4",
+    "zstandard>=0.22.0",
+    "mmh3>=4.0.0",
+    "Pillow>=10.0.0",
+    "texture2ddecoder>=1.0.4",
+    "PyYAML>=6.0"
+]
+
+[tool.ruff]
+line-length = 100
+target-version = "py39"
+
+[tool.ruff.lint]
+select = [
+    "E",  # pycodestyle errors
+    "W",  # pycodestyle warnings
+    "F",  # pyflakes
+    "I",  # isort
+    "B",  # flake8-bugbear
+    "C4", # flake8-comprehensions
+]
+ignore = [
+    "E501",  # line too long, handled by formatter
+]
+
+[tool.ruff.format]
+quote-style = "double"
+indent-style = "space"

pyproject.toml

@@ -5,7 +5,6 @@
 from pathlib import Path
 
 import oead
-
 from zstd_totk import compress_container, decompress_container
 
 _EXTENSIONS_PATH = Path(__file__).parent.parent / 'config' / 'aamp-extensions.json'

python/aamp_io.py

@@ -4,7 +4,6 @@
 from pathlib import Path
 
 import oead
-
 from bntx_reader import is_bntx, list_textures, read_texture_data
 from zstd_totk import compress_container, decompress_container
 
@@ -307,7 +306,7 @@ def read_archive_file_bytes(disk_archive_path: str, file_path: str, romfs_path:
 def _reject_bntx_mutation(disk_archive_path: str, operation: str, target_path: str = '') -> None:
     if _is_bntx_name(disk_archive_path):
         raise PermissionError(f'Cannot {operation} inside a BNTX texture container (read-only)')
-        
+
     if target_path:
         segments = [s for s in target_path.split('/') if s]
         if any(_is_bntx_name(seg) for seg in segments[:-1]):

python/archive_resolve.py

@@ -8,7 +8,7 @@
 import os
 import tempfile
 
-from bntx_reader import _parse_textures, _log
+from bntx_reader import _log, _parse_textures
 
 _FORMAT_TABLE: dict[int, tuple[str, int, int, int, str]] = {
     0x01: ('R4G4_UNORM',    1, 1, 1, 'r8'),
@@ -344,10 +344,10 @@ def _apply_channel_swizzle(
         from PIL import Image
         img = Image.frombytes('RGBA', (width, height), pixels, 'raw', raw_mode)
         bands = img.split()
-        
+
         zero_band = None
         one_band = None
-        
+
         def get_band(ch_val):
             nonlocal zero_band, one_band
             if ch_val == 0:

python/bntx_renderer.py

@@ -2,7 +2,6 @@
 
 import re
 
-
 _U64_DECIMAL_PATTERN = re.compile(
     r'(?P<prefix>:\s*|,\s*|\{\s*)(?P<num>\d{10,})(?=\s*(?:$|[,}\]]))',
     re.MULTILINE,

python/byml_yaml_utils.py

@@ -8,8 +8,8 @@
 if str(_VENDOR_PYMSBT) not in sys.path:
     sys.path.insert(0, str(_VENDOR_PYMSBT))
 
-from pymsbt.classes import TextCommand, TextComponent
 from msbt_tag_formatter import command_to_tag, tag_to_command
+from pymsbt.classes import TextCommand, TextComponent
 
 _CMD_PATTERN = re.compile(
     r'\{cmd:([^:}]+):(\d+):(\d+):([^}]*)\}|\{\{([^}]+)\}\}',
@@ -70,7 +70,7 @@ def display_to_components(text: str) -> list:
                 magic, group, cmd_type, hexdata = 14, 0, 0, ''
         else:
             magic, group, cmd_type, hexdata = match.group(1), match.group(2), match.group(3), match.group(4)
-        
+
         hex_clean = hexdata.replace('0x', '').replace(' ', '') if hexdata else ''
         data_size = (len(hex_clean) + 1) // 2
         command = TextCommand.__new__(TextCommand)
@@ -172,4 +172,4 @@ def from_editor_text(text: str) -> dict[str, list]:
         label = line[:colon].strip()
         value = _parse_value(line[colon + 1 :])
         labels[label] = display_to_components(value)
-    return labels
+    return labels

python/msbt_editor_format.py

@@ -1,7 +1,9 @@
-import struct
 import shlex
+import struct
+
 from msbt_tags import MSBT_TAGS_BY_ID, MSBT_TAGS_BY_NAME
 
+
 def _decode_str_arg(raw: str):
     """Strip the leading Nintendo ctrl char from a decoded str argument.
 
@@ -85,7 +87,7 @@ def command_to_tag(magic, group, type_, hexdata):
             if str_run_start is None:
                 str_run_start = i
         else:
-            str_run_start = None 
+            str_run_start = None
 
     for i, arg in enumerate(args_def):
         dtype = arg.get('dataType')
@@ -215,4 +217,4 @@ def tag_to_command(tag_content):
                 b.extend(val_str.encode('utf-16-le'))
             b.extend(b'\x00\x00')
 
-    return str(magic), group, type_, b.hex()
+    return str(magic), group, type_, b.hex()

python/msbt_tag_formatter.py

@@ -8,7 +8,6 @@
 
 from archive_resolve import list_archive_files
 
-
 _ARCHIVE_EXTENSIONS = (
     '.pack',
     '.sarc',

python/romfs_index.py

@@ -1,17 +1,19 @@
-import yaml
 import json
-import oead
 import os
 
+import oead
+import yaml
+
+
 def to_editor_text(byml_doc: oead.byml.Hash) -> str:
     path_list = byml_doc.get("PathList", [])
     tag_list = [str(x) for x in byml_doc.get("TagList", [])]
     bit_table = bytes(byml_doc.get("BitTable", b""))
-    
+
     actor_tag_data = {}
     path_list_count = len(path_list)
     tag_list_count = len(tag_list)
-    
+
     def get_bit(data: bytes, index: int) -> bool:
         byte_idx = index // 8
         bit_idx = index % 8
@@ -26,12 +28,12 @@ def get_bit(data: bytes, index: int) -> bool:
             if get_bit(bit_table, i * tag_list_count + k):
                 actor_tag_list.append(tag_list[k])
         actor_tag_data[actor_path] = actor_tag_list
-        
+
     data = {
         "PathList": actor_tag_data,
         "TagList": tag_list
     }
-    
+
     fmt = os.environ.get('TOTK_TAG_PRODUCT_FORMAT', 'json')
     if fmt == 'yaml':
         return yaml.safe_dump(data, sort_keys=False, allow_unicode=True, width=float('inf'))
@@ -42,41 +44,41 @@ def from_editor_text(editor_text: str, big_endian: bool, version: int) -> bytes:
     json_data = yaml.safe_load(editor_text)
     cached_tag_list = json_data.get("TagList", [])
     path_list_map = json_data.get("PathList", {})
-    
+
     path_vec = list(path_list_map.items())
     def sort_key(item):
         key = item[0]
         parts = key.split('|')
         extract = parts[1] if len(parts) > 1 else key
         return (extract, key)
-        
+
     path_vec.sort(key=sort_key)
-    
+
     path_list_out = []
     bit_table_bits = []
-    
+
     for path, tag_entries in path_vec:
         if '|' in path:
             slices = path.split('|')
             path_list_out.extend(slices)
-            
+
         for tag in cached_tag_list:
             bit_table_bits.append(tag in tag_entries)
-            
+
     # Pack bits to bytes (LSB0)
     bit_table_bytes = bytearray((len(bit_table_bits) + 7) // 8)
     for i, bit in enumerate(bit_table_bits):
         if bit:
             byte_idx = i // 8
             bit_idx = i % 8
             bit_table_bytes[byte_idx] |= (1 << bit_idx)
-            
+
     byml_dict = {
         "PathList": path_list_out,
         "BitTable": bytes(bit_table_bytes),
         "RankTable": "",
         "TagList": cached_tag_list
     }
-    
+
     byml_doc = oead.byml.Hash(byml_dict)
     return oead.byml.to_binary(byml_doc, big_endian=big_endian, version=version)