Fix bug shortening Citi URLs

Author: impredicative

bitlyshortener/shortener.py

@@ -203,9 +203,8 @@ def _shorten_url(self, long_url: str) -> str:  # pylint: disable=too-many-locals
         short_url = response_json["link"]
 
         # Postprocess short URL
-        assert short_url.startswith("https://")
-        # if short_url.startswith('http://'):
-        #     short_url = short_url.replace('http://', 'https://', 1)
+        if short_url.startswith("http://"):  # Example: http://citi.us/2FPqsuZ
+            short_url = short_url.replace("http://", "https://", 1)
         if (not self._vanitize) or (self._vanitize and short_url.startswith("https://bit.ly/")):
             url_id = short_url.rpartition("/")[-1]
             short_url = f"https://j.mp/{url_id}"

setup.cfg

@@ -22,4 +22,4 @@ ignore = D100,D101,D102,D103,D104,D105,D106,D107,D203,D213
 
 [tool:pytest]
 addopts = --isort --mccabe --mypy --pycodestyle --pydocstyle --pylint --ignore=archive
-mccabe-complexity=12
+mccabe-complexity=13

setup.py

@@ -18,7 +18,7 @@ def parse_requirements(filename: str) -> List[str]:
     name="bitlyshortener",
     author="Ouroboros Chrysopoeia",
     author_email="[email protected]",
-    version="0.4.1",
+    version="0.4.2",
     description="High-volume Bitly V4 URL shortener with memory-cache",
     keywords="bitly url shortener",
     long_description=(_DIR / "README.md").read_text().strip(),