57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
import logging
|
|
|
|
def barename(file):
|
|
(val, _) = os.path.splitext(os.path.basename(file))
|
|
return val
|
|
|
|
def ext(file):
|
|
(_, val) = os.path.splitext(os.path.basename(file))
|
|
return val.lower()
|
|
|
|
def same_barename(lhs, rhs):
|
|
return barename(lhs) == barename(rhs)
|
|
|
|
def is_image(file):
|
|
return ext(file) in {'.jpg', '.jpeg', '.png', '.bmp', '.webp', '.jfif'}
|
|
|
|
def read_text(file):
|
|
try:
|
|
with open(file, encoding='utf-8', mode='r') as stream:
|
|
return stream.read().strip()
|
|
except Exception as e:
|
|
logging.warning(f" *** Error reading text file as utf-8: {file}: {e}")
|
|
|
|
try:
|
|
with open(file, encoding='latin-1', mode='r') as stream:
|
|
return stream.read().strip()
|
|
except Exception as e:
|
|
logging.warning(f" *** Error reading text file as latin-1: {file}: {e}")
|
|
|
|
def read_float(file):
|
|
try:
|
|
return float(read_text(file))
|
|
except Exception as e:
|
|
logging.warning(f" *** Could not parse to float in file {file}: {e}")
|
|
|
|
import os
|
|
|
|
def walk_and_visit(path, visit_fn, context=None):
|
|
names = [entry.name for entry in os.scandir(path)]
|
|
|
|
dirs = []
|
|
files = []
|
|
for name in names:
|
|
fullname = os.path.join(path, name)
|
|
|
|
if str(name).startswith('.'):
|
|
continue
|
|
|
|
if os.path.isdir(fullname):
|
|
dirs.append(fullname)
|
|
else:
|
|
files.append(fullname)
|
|
|
|
subcontext = visit_fn(files, context)
|
|
|
|
for subdir in dirs:
|
|
walk_and_visit(subdir, visit_fn, subcontext) |