jamie/ucsinfer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Elborated rename function to recommend

Browse Source
This commit is contained in:
Jamie Hardt
2025-08-30 20:36:36 -07:00
parent 47829c5427
commit 3009d3831e
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
ucsinfer/__main__.py
View File

@@ -28,13 +28,19 @@ def print_recommendation(path: str | None, text: str, ctx: InferenceContext,
print(f"- {i}: {r} ({cat}-{subcat})")
if interactive_rename and path is not None:
response = input("Enter number, or any key to skip: ")
if match(r'^[0-9]+\n', response):
selection = int(response)
response = input("Enter number, t <text> for alternate text, or "
"return to skip: ")
if m := match(r'^([0-9]+)', response):
selection = int(m.group(1))
if 0 <= selection < len(recs):
new_name = recs[selection] + '_' + os.path.basename(path)
new_path = os.path.join(os.path.dirname(path), new_name)
os.rename(path, new_path)
elif m := match(r'^t (.*)', response):
print("searching for new matches")
text = m.group(1)
print_recommendation(path, text, ctx, True)
@click.group(epilog="For more information see "