daviddoji/example-code-2e
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

minor refactoring to make it easier to call shorten()

Browse Source
This commit is contained in:
Luciano Ramalho
2025-05-22 14:28:42 -03:00
parent ec03da74ca
commit cf99650007
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
links/short.py
View File

@@ -60,8 +60,9 @@ def gen_unused_short(redirects: dict) -> Iterator[str]:
yield short yield short
def shorten(urls: list[str], redirects: dict, targets: dict) -> list[tuple[str, str]]: def shorten(urls: list[str]) -> list[tuple[str, str]]:
"""Return (short, long) pairs, appending directives to HTACCESS_SHORT as needed.""" """Return (short, long) pairs, appending directives to HTACCESS_SHORT as needed."""
redirects, targets = load_redirects()
iter_short = gen_unused_short(redirects) iter_short = gen_unused_short(redirects)
pairs = [] pairs = []
timestamp = strftime('%Y-%m-%d %H:%M:%S') timestamp = strftime('%Y-%m-%d %H:%M:%S')
@@ -86,8 +87,7 @@ def shorten(urls: list[str], redirects: dict, targets: dict) -> list[tuple[str,
def main() -> None: def main() -> None:
"""read URLS from filename arguments or stdin""" """read URLS from filename arguments or stdin"""
urls = [line.strip() for line in fileinput.input(encoding='utf-8')] urls = [line.strip() for line in fileinput.input(encoding='utf-8')]
redirects, targets = load_redirects() for short, long in shorten(urls):
for short, long in shorten(urls, redirects, targets):
print(f'{BASE_DOMAIN}/{short}\t{long}') print(f'{BASE_DOMAIN}/{short}\t{long}')