update mongodb issue

Author: recursivezero

.env.local

@@ -3,6 +3,7 @@ MONGO_URI=mongodb://<username>:<password>@127.0.0.1:27017/?authSource=admin&retr
 DOMAIN=https://rzro.link
 PORT=8001
 API_VERSION=""
+APP_NAMe="LOCAL"
 
 MONGO_URI="mongodb://<username>@<password>:127.0.0.1:27017/?retryWrites=true&w=majority&appName=<APP-NAME>"
 DATABASE_NAME="<DB_NAME"

README.md

@@ -157,10 +157,10 @@ tiny/
 poetry install
 ```
 
-### 3. Install with MongoDB Support (Optional)
+### 3. Install with other dependent packages
 
 ```sh
-poetry install --with mongodb
+poetry install --all-extras --with dev
 ```
 
 ---
@@ -178,7 +178,7 @@ poetry run tiny dev
 ```
 
 Open:
-http://127.0.0.1:8000
+<http://127.0.0.1:8000>
 
 ---
 
@@ -260,11 +260,11 @@ poetry run tiny dev
 
 ### API Base URL
 
-http://127.0.0.1:8000/api
+<http://127.0.0.1:8000/api>
 
 ### Swagger Docs
 
-http://127.0.0.1:8000/api/docs
+<http://127.0.0.1:8000/api/docs>
 
 ### Shorten URL
 

app/api/fast_api.py

@@ -2,28 +2,29 @@
 import re
 import traceback
 from datetime import datetime, timezone
+from typing import Any
 
 from fastapi import APIRouter, FastAPI, Request
 from fastapi.responses import HTMLResponse, JSONResponse
 from pydantic import BaseModel, Field
 
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from pymongo.errors import PyMongoError
-else:
-    try:
-        from pymongo.errors import PyMongoError
-    except ImportError:
-
-        class PyMongoError(Exception):
-            pass
-
-
 from app import __version__
 from app.db import data as db_data
 from app.utils.helper import generate_code, is_valid_url, sanitize_url
 
+PyMongoError: Any
+try:
+    from pymongo.errors import PyMongoError as _RealPyMongoError
+
+    PyMongoError = _RealPyMongoError
+except (ImportError, ModuleNotFoundError):
+    # 2. Fallback: Define our own only if the real one fails
+    class _FallbackPyMongoError(Exception):
+        pass
+
+    # Assign our fallback to the same local name
+    PyMongoError = _FallbackPyMongoError
+
 SHORT_CODE_PATTERN = re.compile(r"^[A-Za-z0-9]{6}$")
 MAX_URL_LENGTH = 2048
 

app/main.py

@@ -2,48 +2,55 @@
 import os
 from contextlib import asynccontextmanager
 from pathlib import Path
-from typing import Optional
+from typing import Any, Optional
 
 from fastapi import FastAPI, Form, Request
 from fastapi.responses import HTMLResponse, PlainTextResponse, RedirectResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
 from starlette.middleware.sessions import SessionMiddleware
 
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from pymongo.errors import PyMongoError
-else:
-    try:
-        from pymongo.errors import PyMongoError
-    except ImportError:
-
-        class PyMongoError(Exception):
-            pass
-
-
-# fallback for offline mode
-
 from app.api.fast_api import app as api_app
 from app.db import data as db_data
-from app.utils.qr import generate_qr_with_logo
-
 from app.utils.config import load_env
 from app.utils.helper import (
     format_date,
     generate_code,
     is_valid_url,
     sanitize_url,
 )
+from app.utils.qr import generate_qr_with_logo
+
+load_env()  # ✅ load env ONCE
+
+RED = "\033[31m"
+GREEN = "\033[32m"
+BLUE = "\033[34m"
+RESET = "\033[0m"
+
+app_name = os.getenv("APP_NAME", "TinyURL")
+print(f"Environment loaded as {BLUE}{app_name}{RESET}")
+
+# 1. MongoDB error handling: Try to import the real exception class first
+PyMongoError: Any
+try:
+    from pymongo.errors import PyMongoError as _RealPyMongoError
+
+    PyMongoError = _RealPyMongoError
+except (ImportError, ModuleNotFoundError):
+    # 2. Fallback: Define our own only if the real one fails
+    class _FallbackPyMongoError(Exception):
+        pass
+
+    # Assign our fallback to the same local name
+    PyMongoError = _FallbackPyMongoError
 
 
 # -----------------------------
 # Lifespan: env + DB connect ONCE
 # -----------------------------
 @asynccontextmanager
 async def lifespan(app: FastAPI):
-    load_env()  # ✅ load env ONCE
     connected = db_data.connect_db()  # ✅ connect DB ONCE
     app.state.db_available = connected
     yield

pyproject.toml

@@ -29,7 +29,7 @@ dependencies = [
   "validators>=0.35.0,<0.36.0",
   "itsdangerous>=2.2.0,<3.0.0",
   "python-multipart>=0.0.22,<0.0.23",
-  "jinja2 = ^3.1.2",
+  "jinja2>=3.1.2",
 ]
 [project.optional-dependencies]
 mongodb = ["pymongo>=4.16.0,<5.0.0"]