mirror of
https://github.com/wpilibsuite/allwpilib
synced 2026-06-30 02:31:44 +00:00
[upstream_utils] Use pathlib instead of os.path (#7983)
A noteworthy change is the replacement of the `dp.startswith(os.path.join(".", "subdir"))` pattern. pathlib doesn't offer something with similar semantics besides `match` and `full_match`, so there's now a helper function that replicates the behavior.
Other notable changes include the addition of type annotations to ensure code correctness, using == to check file names instead of `endswith` for clarity (`endswith` is still used to check extensions), manual walking and copying being refactored in googletest, json, memory, nanopb, protobuf, and sleipnir to use `walk_cwd_and_copy_if`, and matching functions being shortened to the point where they can just be inlined into the lambda.
Co-authored-by: Tyler Veness <calcmogul@gmail.com>
Co-authored-by: David Vo <auscompgeek@users.noreply.github.com>
This commit is contained in:
@@ -1,58 +1,35 @@
|
||||
#!/usr/bin/env python3
|
||||
|
||||
import os
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
|
||||
from upstream_utils import Lib, copy_to
|
||||
from upstream_utils import Lib, has_prefix, walk_cwd_and_copy_if
|
||||
|
||||
|
||||
def copy_upstream_src(wpilib_root):
|
||||
wpimath = os.path.join(wpilib_root, "wpimath")
|
||||
def copy_upstream_src(wpilib_root: Path):
|
||||
wpimath = wpilib_root / "wpimath"
|
||||
|
||||
# Delete old install
|
||||
for d in [
|
||||
"src/main/native/thirdparty/sleipnir/src",
|
||||
"src/main/native/thirdparty/sleipnir/include",
|
||||
]:
|
||||
shutil.rmtree(os.path.join(wpimath, d), ignore_errors=True)
|
||||
shutil.rmtree(wpimath / d, ignore_errors=True)
|
||||
|
||||
# Copy Sleipnir source files into allwpilib
|
||||
src_files = [os.path.join(dp, f) for dp, dn, fn in os.walk("src") for f in fn]
|
||||
src_files = copy_to(
|
||||
src_files, os.path.join(wpimath, "src/main/native/thirdparty/sleipnir")
|
||||
# Copy Sleipnir files into allwpilib
|
||||
walk_cwd_and_copy_if(
|
||||
lambda dp, f: (has_prefix(dp, Path("include")) or has_prefix(dp, Path("src")))
|
||||
or f == ".clang-format"
|
||||
or f == ".clang-tidy"
|
||||
or f == ".styleguide"
|
||||
or f == ".styleguide-license",
|
||||
wpimath / "src/main/native/thirdparty/sleipnir",
|
||||
)
|
||||
|
||||
# Copy Sleipnir header files into allwpilib
|
||||
include_files = [
|
||||
os.path.join(dp, f) for dp, dn, fn in os.walk("include") for f in fn
|
||||
]
|
||||
include_files = copy_to(
|
||||
include_files, os.path.join(wpimath, "src/main/native/thirdparty/sleipnir")
|
||||
)
|
||||
|
||||
for filename in [
|
||||
".clang-format",
|
||||
".clang-tidy",
|
||||
".styleguide",
|
||||
".styleguide-license",
|
||||
]:
|
||||
shutil.copyfile(
|
||||
filename,
|
||||
os.path.join(wpimath, "src/main/native/thirdparty/sleipnir", filename),
|
||||
)
|
||||
|
||||
# Write shim for wpi::SmallVector
|
||||
try:
|
||||
os.mkdir(
|
||||
os.path.join(wpimath, "src/main/native/thirdparty/sleipnir/include/gch")
|
||||
)
|
||||
except:
|
||||
pass
|
||||
(wpimath / "src/main/native/thirdparty/sleipnir/include/gch").mkdir()
|
||||
with open(
|
||||
os.path.join(
|
||||
wpimath,
|
||||
"src/main/native/thirdparty/sleipnir/include/gch/small_vector.hpp",
|
||||
),
|
||||
wpimath / "src/main/native/thirdparty/sleipnir/include/gch/small_vector.hpp",
|
||||
"w",
|
||||
) as f:
|
||||
f.write(
|
||||
|
||||
Reference in New Issue
Block a user