short.py appends timestamps to short.htaccesss

ramalho · ramalho · commit ec03da74cac0 · 2025-05-22T13:44:46.000-03:00
diff --git a/links/short.py b/links/short.py
@@ -6,8 +6,8 @@
 This script reads lines from stdin or files named as arguments, then:
 
 1. retrieves or creates new short URLs, taking into account existing RedirectTemp
-   directives in custom.htacess or short.htacess;
-2. appends RedirectTemp directives for newly created short URLs to short.htacess;
+   directives in custom.htaccess or short.htaccess;
+2. appends RedirectTemp directives for newly created short URLs to short.htaccess;
 3. outputs the list of (short, long) URLs retrieved or created.
 
 """
@@ -17,21 +17,25 @@
 from collections.abc import Iterator
 from time import strftime
 
+HTACCESS_CUSTOM = 'custom.htaccess'
+HTACCESS_SHORT = 'short.htaccess'
+HTACCESS_FILES = (HTACCESS_CUSTOM, HTACCESS_SHORT)
 BASE_DOMAIN = 'fpy.li'
 
-def load_redirects():
+
+def load_redirects() -> tuple[dict, dict]:
     redirects = {}
     targets = {}
-    for filename in ('custom.htaccess', 'short.htaccess'):
+    for filename in HTACCESS_FILES:
         with open(filename) as fp:
             for line in fp:
                 if line.startswith('RedirectTemp'):
                     _, short, long = line.split()
                     short = short[1:]  # Remove leading slash
-                    assert short not in redirects, f"{filename}: duplicate redirect from {short}"
-                    # custom is live since 2022, we cannot change it remove duplicate targets
-                    if not filename.startswith('custom'):
-                        assert long not in targets, f"{filename}: Duplicate redirect to {long}"
+                    assert short not in redirects, f'{filename}: duplicate redirect from {short}'
+                    # htaccess.custom is live since 2022, we can't change it remove duplicate targets
+                    if filename != HTACCESS_CUSTOM:
+                        assert long not in targets, f'{filename}: duplicate redirect to {long}'
                     redirects[short] = long
                     targets[long] = short
     return redirects, targets
@@ -41,9 +45,7 @@ def load_redirects():
 
 
 def gen_short(start_len=1) -> Iterator[str]:
-    """
-    Generate every possible sequence of SDIGITS, starting with start_len
-    """
+    """Generate every possible sequence of SDIGITS, starting with start_len"""
     length = start_len
     while True:
         for short in itertools.product(SDIGITS, repeat=length):
@@ -52,22 +54,20 @@ def gen_short(start_len=1) -> Iterator[str]:
 
 
 def gen_unused_short(redirects: dict) -> Iterator[str]:
-    """
-    Generate next available short URL of len >= 2.
-    """
+    """Generate next available short URL of len >= 2."""
     for short in gen_short(2):
         if short not in redirects:
             yield short
 
 
-def shorten(urls: list[str], redirects: dict, targets: dict) -> list[tuple[str,str]]:
-    """return (short, long) pairs, appending directives to short.htaccess as needed"""
+def shorten(urls: list[str], redirects: dict, targets: dict) -> list[tuple[str, str]]:
+    """Return (short, long) pairs, appending directives to HTACCESS_SHORT as needed."""
     iter_short = gen_unused_short(redirects)
     pairs = []
     timestamp = strftime('%Y-%m-%d %H:%M:%S')
-    with open('short.htaccess', 'a') as fp:
+    with open(HTACCESS_SHORT, 'a') as fp:
         for long in urls:
-            assert BASE_DOMAIN not in long, f"{long} is a {BASE_DOMAIN} URL"
+            assert BASE_DOMAIN not in long, f'{long} is a {BASE_DOMAIN} URL'
             if long in targets:
                 short = targets[long]
             else:
@@ -79,16 +79,16 @@ def shorten(urls: list[str], redirects: dict, targets: dict) -> list[tuple[str,s
                     timestamp = None
                 fp.write(f'RedirectTemp /{short} {long}\n')
             pairs.append((short, long))
-            
+
     return pairs
 
 
-def main():
+def main() -> None:
     """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, redirects, targets):
-    		print(f'{BASE_DOMAIN}/{short}\t{long}')
+        print(f'{BASE_DOMAIN}/{short}\t{long}')
 
 
 if __name__ == '__main__':
diff --git a/ruff.toml b/ruff.toml
@@ -0,0 +1,4 @@
+line-length = 100
+[format]
+# Like Python's repr(), use single quotes for strings.
+quote-style = "single"
