mirror of
https://github.com/denoland/deno.git
synced 2024-11-27 16:10:57 -05:00
89 lines
2.2 KiB
Python
Executable file
89 lines
2.2 KiB
Python
Executable file
#!/usr/bin/env python
|
|
# Copyright 2018-2020 the Deno authors. All rights reserved. MIT license.
|
|
import os
|
|
import sys
|
|
import argparse
|
|
from third_party import python_env, get_prebuilt_tool_path
|
|
from util import git_ls_files, git_staged, third_party_path, root_path
|
|
from util import print_command, run
|
|
|
|
cmd_args = None
|
|
|
|
|
|
def get_cmd_args():
|
|
global cmd_args
|
|
|
|
if cmd_args:
|
|
return cmd_args
|
|
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("--js", help="run dprint", action="store_true")
|
|
parser.add_argument("--py", help="run yapf", action="store_true")
|
|
parser.add_argument("--rs", help="run rustfmt", action="store_true")
|
|
parser.add_argument(
|
|
"--staged", help="run only on staged files", action="store_true")
|
|
cmd_args = parser.parse_args()
|
|
return cmd_args
|
|
|
|
|
|
def get_sources(*args):
|
|
getter = git_staged if get_cmd_args().staged else git_ls_files
|
|
return getter(*args)
|
|
|
|
|
|
def main():
|
|
os.chdir(root_path)
|
|
|
|
args = get_cmd_args()
|
|
|
|
did_fmt = False
|
|
if args.js:
|
|
dprint()
|
|
did_fmt = True
|
|
if args.py:
|
|
yapf()
|
|
did_fmt = True
|
|
if args.rs:
|
|
rustfmt()
|
|
did_fmt = True
|
|
|
|
if not did_fmt:
|
|
dprint()
|
|
yapf()
|
|
rustfmt()
|
|
|
|
|
|
def dprint():
|
|
executable_path = get_prebuilt_tool_path("dprint")
|
|
command = [executable_path, "fmt"]
|
|
run(command, shell=False, quiet=True)
|
|
|
|
|
|
def yapf():
|
|
script = os.path.join(third_party_path, "python_packages", "bin", "yapf")
|
|
source_files = get_sources(root_path, ["*.py"])
|
|
if source_files:
|
|
print_command("yapf", source_files)
|
|
run([sys.executable, script, "-i", "--style=pep8", "--"] +
|
|
source_files,
|
|
env=python_env(),
|
|
shell=False,
|
|
quiet=True)
|
|
|
|
|
|
def rustfmt():
|
|
config_file = os.path.join(root_path, ".rustfmt.toml")
|
|
source_files = get_sources(root_path, ["*.rs"])
|
|
if source_files:
|
|
print_command("rustfmt", source_files)
|
|
run([
|
|
"rustfmt",
|
|
"--config-path=" + config_file,
|
|
"--",
|
|
] + source_files,
|
|
shell=False,
|
|
quiet=True)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|