feat: URL shortener service that engineers can run via this repo or copy/paste to their own projects.

Author: justinjohnson-dev

projects/url_shortener/app/__init__.py

@@ -0,0 +1,50 @@
+import os
+import sys
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
+
+from app.models.url_shortener import UrlShortenerValidation
+from app.services.url_shortener import URLShortener
+
+
+def run_url_shortener():
+	entered_url = input("Enter the URL you want to shorten: ")
+
+	# Validate the entered URL
+	url_validation = UrlShortenerValidation(url=entered_url)
+
+	# Extract the validated URL and convert it to string
+	validated_url = str(url_validation.url)
+
+	# Pass the string URL to URLShortener
+	url_shortener = URLShortener(url=validated_url)
+
+	result = url_shortener.shorten_url()
+	if isinstance(result, str):
+		print(f"Shortened URL: {result}")
+	else:
+		print(f"Error occurred: {result}")
+	return result
+
+if __name__ == "__main__":
+	run_url_shortener()
+    # run_url_shortener_with_other_library()
+
+# def run_url_shortener_with_other_library():
+#     entered_url = input("Enter the URL you want to shorten: ")
+
+#     # Validate the entered URL
+#     url_validation = UrlShortenerValidation(url=entered_url)
+
+#     # Extract the validated URL and convert it to string
+#     validated_url = str(url_validation.url)
+
+#     # Pass the string URL to URLShortener
+#     url_shortener = URLShortener(url=validated_url, shortener=None) # enter the shortener you want to use
+
+#     result = url_shortener.shorten_url()
+#     if isinstance(result, str):
+#         print(f"Shortened URL: {result}")
+#     else:
+#         print(f"Error occurred: {result}")
+#     return result

projects/url_shortener/app/main.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel, HttpUrl
+
+
+class UrlShortenerValidation(BaseModel):
+	url: HttpUrl

projects/url_shortener/app/models/__init__.py

@@ -0,0 +1,29 @@
+from configparser import ParsingError
+from typing import Optional, Union
+
+import pyshorteners
+from pydantic import BaseModel, PrivateAttr, ValidationError
+
+
+class URLShortener(BaseModel):
+	url: str
+	_shortener: Optional[pyshorteners.Shortener] = PrivateAttr()
+
+	class Config:
+		arbitrary_types_allowed = True
+
+	def __init__(self, **data):
+		shortener = data.pop("shortener", None) or pyshorteners.Shortener()
+		super().__init__(**data)
+		self._shortener = shortener
+
+	def shorten_url(self) -> Union[str, Exception]:
+		if self._shortener is None:
+			raise ValueError("Shortener is not initialized.")
+
+		try:
+			return str(self._shortener.tinyurl.short(self.url))
+		except (ParsingError, ValidationError) as error:
+			return error
+		except Exception as generic_error:
+			return generic_error