67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
"""Views for the feeds app.
|
|
|
|
IndexView - /
|
|
FeedsView - /feeds
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import typing
|
|
|
|
from django.db import connection
|
|
from django.views.generic.base import TemplateView
|
|
from django.views.generic.list import ListView
|
|
|
|
from feeds.models import Feed
|
|
|
|
|
|
def get_database_size() -> int:
|
|
"""Get the size of a database.
|
|
|
|
Returns:
|
|
The size of the database in megabytes.
|
|
"""
|
|
with connection.cursor() as cursor:
|
|
# Query to get the size of the database
|
|
query = "SELECT pg_database_size('feedvault')"
|
|
cursor.execute(sql=query)
|
|
|
|
if not cursor:
|
|
return 0
|
|
|
|
size_in_bytes = cursor.fetchone()[0] # type: ignore # noqa: PGH003
|
|
|
|
if not size_in_bytes:
|
|
return 0
|
|
|
|
return int(size_in_bytes / (1024 * 1024))
|
|
|
|
|
|
class IndexView(TemplateView):
|
|
"""Index page."""
|
|
|
|
template_name = "index.html"
|
|
|
|
def get_context_data(self: IndexView, **kwargs: dict) -> dict:
|
|
"""Add feed count and database size to context data."""
|
|
context: dict = super().get_context_data(**kwargs)
|
|
context["feed_count"] = Feed.objects.count()
|
|
context["database_size"] = get_database_size()
|
|
return context
|
|
|
|
|
|
class FeedsView(ListView):
|
|
"""Feeds page."""
|
|
|
|
model = Feed
|
|
template_name = "feeds.html"
|
|
context_object_name = "feeds"
|
|
paginate_by = 100
|
|
ordering: typing.ClassVar[list[str]] = ["-created_at"]
|
|
|
|
def get_context_data(self: FeedsView, **kwargs: dict) -> dict:
|
|
"""Add feed count and database size to context data."""
|
|
context: dict = super().get_context_data(**kwargs)
|
|
context["feed_count"] = Feed.objects.count()
|
|
context["database_size"] = get_database_size()
|
|
return context
|