mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
65 lines
1.6 KiB
Python
65 lines
1.6 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import os
|
|
import re
|
|
from shutil import rmtree
|
|
|
|
|
|
def remove(path: str) -> None:
|
|
if os.path.exists(path):
|
|
if os.path.isdir(path):
|
|
rmtree(path)
|
|
else:
|
|
os.remove(path)
|
|
|
|
|
|
# find all indices of a list of strings that match a regex
|
|
def findall_regex(items: list[str], regex: re.Pattern[str]) -> list[int]:
|
|
found = list()
|
|
for i in range(0, len(items)):
|
|
k = regex.match(items[i])
|
|
if k:
|
|
found.append(i)
|
|
k = None
|
|
|
|
return found
|
|
|
|
|
|
def split_by_regex(items: list[str], regex: re.Pattern[str]) -> list[list[str]]:
|
|
splits = list()
|
|
indices = findall_regex(items, regex)
|
|
if not indices:
|
|
splits.append(items)
|
|
return splits
|
|
|
|
# Add first chunk before first match
|
|
splits.append(items[0 : indices[0]])
|
|
|
|
# Add chunks between matches
|
|
for i in range(len(indices) - 1):
|
|
splits.append(items[indices[i] : indices[i + 1]])
|
|
|
|
# Add final chunk after last match
|
|
splits.append(items[indices[-1] :])
|
|
|
|
return splits
|
|
|
|
|
|
# http://stackoverflow.com/questions/377017/test-if-executable-exists-in-python
|
|
def which(program: str) -> str | None:
|
|
def is_exe(fpath: str) -> bool:
|
|
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
|
|
|
|
fpath, fname = os.path.split(program)
|
|
if fpath:
|
|
if is_exe(program):
|
|
return program
|
|
else:
|
|
for path in os.environ['PATH'].split(os.pathsep):
|
|
path = path.strip('"')
|
|
exe_file = os.path.join(path, program)
|
|
if is_exe(exe_file):
|
|
return exe_file
|
|
|
|
return None
|