Files
nuitka-mirror/bin/autoformat-nuitka-source-stdio
Kay Hayen 0ff19a9a4c Zed: Autoformat files on save with our script
* Interestingly that's not possible with Visual Code,
  but zed allows it. We need a tool that works off
  stdin though for it.
2026-06-22 11:45:00 +02:00

68 lines
2.0 KiB
Python
Executable File

#!/usr/bin/env python3
# Copyright 2026, Kay Hayen, mailto:kay.hayen@gmail.com find license text at end of file
"""Wrapper that formats a file and outputs the result to stdout.
For editor/formatter integration:
- Content is read from stdin.
- The first argument is the filename, used for type detection.
"""
import os
import sys
import tempfile
sys.path.insert(0, os.path.normpath(os.path.join(os.path.dirname(__file__), "..")))
# isort:start
from nuitka.tools.quality.auto_format.AutoFormat import autoFormatFile
def main():
if len(sys.argv) < 2:
sys.exit("Error, filename argument required.")
src = sys.argv[1]
ext = os.path.splitext(src)[1]
fd, tmp_name = tempfile.mkstemp(suffix=ext)
try:
os.write(fd, sys.stdin.buffer.read())
os.close(fd)
autoFormatFile(
tmp_name,
git_stage=False,
effective_filename=src,
trace=False,
)
with open(tmp_name, "rb") as f:
sys.stdout.buffer.write(f.read())
finally:
os.unlink(tmp_name)
if __name__ == "__main__":
main()
# Part of "Nuitka", an optimizing Python compiler that is compatible and
# integrates with CPython, but also works on its own.
#
# Licensed under the GNU Affero General Public License, Version 3 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.gnu.org/licenses/agpl-3.0.txt
#
# See also: "Nuitka Runtime Library Exception, Version 1.0" in file
# "LICENSE-RUNTIME.txt" for additional permissions granted under Section 7.
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.