Имя файла без учета регистра Python?
Мне нужно загрузить файл с именем, но имя, которое я получаю, нечувствительно к регистру. "A.txt" может быть "a.txt". Как это сделать быстро (не генерировать все возможные имена и попробовать каждый)?
4 ответов
можно список каталогов файле в (os.listdir
), и посмотреть, есть ли совпадения для вашего имени файла. Сопоставление может быть сделано с помощью нижнего корпуса как имен файлов, так и сравнения.
вы не можете сделать это, не взяв список каталогов и принимая как элемент, который вы ищете, так и каждый в каталоге в общий случай для сравнения. Файловая система чувствительна к регистру, и это все, что нужно.
вот функция (ну, две), которую я написал, чтобы сделать это полностью, сопоставляя имя файла нечувствительным образом, рекурсивно: http://portableapps.hg.sourceforge.net/hgweb/portableapps/development-toolkit/file/775197d56e86/utils.py#l78.
def path_insensitive(path):
"""
Get a case-insensitive path for use on a case sensitive system.
>>> path_insensitive('/Home')
'/home'
>>> path_insensitive('/Home/chris')
'/home/chris'
>>> path_insensitive('/HoME/CHris/')
'/home/chris/'
>>> path_insensitive('/home/CHRIS')
'/home/chris'
>>> path_insensitive('/Home/CHRIS/.gtk-bookmarks')
'/home/chris/.gtk-bookmarks'
>>> path_insensitive('/home/chris/.GTK-bookmarks')
'/home/chris/.gtk-bookmarks'
>>> path_insensitive('/HOME/Chris/.GTK-bookmarks')
'/home/chris/.gtk-bookmarks'
>>> path_insensitive("/HOME/Chris/I HOPE this doesn't exist")
"/HOME/Chris/I HOPE this doesn't exist"
"""
return _path_insensitive(path) or path
def _path_insensitive(path):
"""
Recursive part of path_insensitive to do the work.
"""
if path == '' or os.path.exists(path):
return path
base = os.path.basename(path) # may be a directory or a file
dirname = os.path.dirname(path)
suffix = ''
if not base: # dir ends with a slash?
if len(dirname) < len(path):
suffix = path[:len(path) - len(dirname)]
base = os.path.basename(dirname)
dirname = os.path.dirname(dirname)
if not os.path.exists(dirname):
dirname = _path_insensitive(dirname)
if not dirname:
return
# at this point, the directory exists but not the file
try: # we are expecting dirname to be a directory, but it could be a file
files = os.listdir(dirname)
except OSError:
return
baselow = base.lower()
try:
basefinal = next(fl for fl in files if fl.lower() == baselow)
except StopIteration:
return
if basefinal:
return os.path.join(dirname, basefinal) + suffix
else:
return
это простая рекурсивная функция для поиска Eli предлагает выше:
def find_sensitive_path(dir, insensitive_path):
insensitive_path = insensitive_path.strip(os.path.sep)
parts = insensitive_path.split(os.path.sep)
next_name = parts[0]
for name in os.listdir(dir):
if next_name.lower() == name.lower():
improved_path = os.path.join(dir, name)
if len(parts) == 1:
return improved_path
else:
return find_sensitive_path(improved_path, os.path.sep.join(parts[1:]))
return None
Создайте список каталогов;и создайте словарь, содержащий сопоставление имен файлов верхнего регистра с их фактическими именами файлов. Затем сделайте ввод в верхнем регистре и найдите его в словаре.