mypy fixes

repology · Sep 17, 2024 · edc36d5 · edc36d5
1 parent 0b29846
commit edc36d5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/repologyapp/template_functions.py b/repologyapp/template_functions.py
@@ -52,4 +52,4 @@ def url_for_static(**args: Any) -> Any:
 
 def needs_ipv6_notice(*variants: str) -> bool:
     now = datetime.datetime.now()
-    return now.month == 6 and now.day == 6 and ':' not in flask.request.environ.REMOTE_ADDR and flask.request.endpoint != 'index'
+    return now.month == 6 and now.day == 6 and ':' not in flask.request.environ['REMOTE_ADDR'] and flask.request.endpoint != 'index'
diff --git a/repologyapp/template_functions_extra.py b/repologyapp/template_functions_extra.py
@@ -20,5 +20,5 @@
 from repologyapp.views import registry as view_registry
 
 
-def current_endpoint_group_in(**args: Any) -> str | None:
+def current_endpoint_group_in(**args: Any) -> bool:
     return view_registry.get_current_endpoint_group() in args['groups']
diff --git a/repologyapp/view_registry.py b/repologyapp/view_registry.py
@@ -111,4 +111,5 @@ def register_in_flask(self, app: flask.Flask) -> None:
             registrant.register_in_flask(app)
 
     def get_current_endpoint_group(self) -> str | None:
+        assert flask.request.endpoint
         return self._group_by_endpoint.get(flask.request.endpoint)