75 lines
2.1 KiB
Python
Executable File
75 lines
2.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import re, subprocess, sys
|
|
|
|
|
|
DEBUG = False
|
|
DRY_RUN = False
|
|
NUM_KEEP_PER_BOARD = 4
|
|
|
|
|
|
def main():
|
|
ssh_machine = sys.argv[1]
|
|
ssh_firmware_dir = sys.argv[2]
|
|
|
|
# SSH to get list of existing files.
|
|
p = subprocess.run(
|
|
["ssh", ssh_machine, "find", ssh_firmware_dir, "-name", "\\*-preview.\\*"],
|
|
capture_output=True,
|
|
)
|
|
if p.returncode != 0:
|
|
print(p.stderr)
|
|
return
|
|
all_files = p.stdout.split(b"\n")
|
|
|
|
# Parse all files to organise into boards/date/version.
|
|
boards = {}
|
|
for file in all_files:
|
|
m = re.match(
|
|
rb"([a-z/.]+)/([A-Za-z0-9_-]+)-(20[0-9]{6})-(v[0-9.]+)-preview.([0-9]+).g[0-9a-f]+.",
|
|
file,
|
|
)
|
|
if not m:
|
|
continue
|
|
dir, board, date, version, ncommits = m.groups()
|
|
if board not in boards:
|
|
boards[board] = {}
|
|
if (date, version, ncommits) not in boards[board]:
|
|
boards[board][(date, version, ncommits)] = []
|
|
boards[board][(date, version, ncommits)].append(file)
|
|
|
|
# Collect files to remove based on date and version.
|
|
remove = []
|
|
for board in boards.values():
|
|
filelist = [
|
|
(date, version, ncommits, files) for (date, version, ncommits), files in board.items()
|
|
]
|
|
filelist.sort(reverse=True)
|
|
keep = []
|
|
for date, version, ncommits, files in filelist:
|
|
if keep and (version, ncommits) == keep[-1]:
|
|
remove.extend(files)
|
|
elif len(keep) >= NUM_KEEP_PER_BOARD:
|
|
remove.extend(files)
|
|
else:
|
|
keep.append((version, ncommits))
|
|
|
|
if DEBUG:
|
|
all_files.sort(reverse=True)
|
|
for file in all_files:
|
|
print(file, file in remove)
|
|
print(len(remove), "/", len(all_files))
|
|
|
|
# Do removal of files.
|
|
for file in remove:
|
|
file = str(file, "ascii")
|
|
print("remove:", file)
|
|
if not DRY_RUN:
|
|
p = subprocess.run(["ssh", ssh_machine, "/bin/rm", file], capture_output=True)
|
|
if p.returncode != 0:
|
|
print(p.stderr)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|