patx/micropie

clean up a few typos in url shortner, make rate limit middleware require db name

Commit 83ea2b1 · patx · 2025-12-28T19:26:12-05:00

Changeset

83ea2b1f6589bd3b1a86db9b0221fcb46ca8485e

Parents

fda0eedb6ad6ea2f9f685f1ae778d48cc9bc9d30

View source at this commit

Comments

No comments yet.

Log in to comment

Diff

diff --git a/examples/url_shortener/main.py b/examples/url_shortener/main.py
index 7a9b9d1..2d7aef2 100644
--- a/examples/url_shortener/main.py
+++ b/examples/url_shortener/main.py
@@ -62,6 +62,7 @@ app = Shorty(session_backend=MkvSessionBackend(
 app.middlewares.append(
     MongoRateLimitMiddleware(
         mongo_uri=MONGO_URI,
+        db_name=DB_NAME,
         allowed_hosts=None,          # don't enforce host allowlist, change in prod
         trust_proxy_headers=False,   # change in prod
         require_cf_ray=False,
diff --git a/examples/url_shortener/middlewares/rate_limit.py b/examples/url_shortener/middlewares/rate_limit.py
index 983f4a3..29b93fd 100644
--- a/examples/url_shortener/middlewares/rate_limit.py
+++ b/examples/url_shortener/middlewares/rate_limit.py
@@ -43,7 +43,7 @@ class MongoRateLimitMiddleware(HttpMiddleware):
     def __init__(
         self,
         mongo_uri: str,
-        db_name: str = "vegy_security",
+        db_name: str,
         collection_name: str = "rate_limits_global",
         *,
         allowed_hosts: Set[str] | None = None,
diff --git a/examples/url_shortener/sessions/mongo_session.py b/examples/url_shortener/sessions/mongo_session.py
index 1b47583..a97e863 100644
--- a/examples/url_shortener/sessions/mongo_session.py
+++ b/examples/url_shortener/sessions/mongo_session.py
@@ -29,7 +29,7 @@ class MkvSessionBackend(SessionBackend):
         *,
         key_prefix: str = "sess:",
     ) -> None:
-        self.store = Mkv(mongo_uri, db_name=db_name, collection_name=collection_name)
+        self.store = Mkv(mongo_uri, db_name, collection_name)
         self.key_prefix = key_prefix
 
     def _k(self, session_id: str) -> str: