mirror of
https://github.com/brycedrennan/imaginAIry
synced 2024-10-31 03:20:40 +00:00
20 lines
514 B
Python
20 lines
514 B
Python
|
import os
|
||
|
|
from urllib.parse import urlparse
|
||
|
|
|
||
|
|
|
||
|
|
def format_filename(format_template: str, data: dict) -> str:
|
||
|
|
"""
|
||
|
|
Formats the filename based on the provided template and variables.
|
||
|
|
"""
|
||
|
|
if not isinstance(format_template, str):
|
||
|
|
raise TypeError("format argument must be a string")
|
||
|
|
|
||
|
|
filename = format_template.format(**data)
|
||
|
|
return filename
|
||
|
|
|
||
|
|
|
||
|
|
def get_url_file_name(url):
|
||
|
|
parsed = urlparse(url)
|
||
|
|
model_name, _ = os.path.splitext(os.path.basename(parsed.path))
|
||
|
|
return model_name
|