Use new View subclass for PatternPortfolioView - refs #2078

pull/2080/head
Simon Willison 2023-05-25 17:07:20 -07:00
rodzic df5fd7342c
commit e990fbc00e
2 zmienionych plików z 49 dodań i 10 usunięć

Wyświetl plik

@ -17,6 +17,7 @@ import secrets
import sys
import threading
import time
import types
import urllib.parse
from concurrent import futures
from pathlib import Path
@ -1361,7 +1362,7 @@ class Datasette:
r"/-/allow-debug$",
)
add_route(
PatternPortfolioView.as_view(self),
wrap_view(PatternPortfolioView, self),
r"/-/patterns$",
)
add_route(DatabaseDownload.as_view(self), r"/(?P<database>[^\/\.]+)\.db$")
@ -1673,7 +1674,42 @@ def _cleaner_task_str(task):
return _cleaner_task_str_re.sub("", s)
def wrap_view(view_fn, datasette):
def wrap_view(view_fn_or_class, datasette):
is_function = isinstance(view_fn_or_class, types.FunctionType)
if is_function:
return wrap_view_function(view_fn_or_class, datasette)
else:
if not isinstance(view_fn_or_class, type):
raise ValueError("view_fn_or_class must be a function or a class")
return wrap_view_class(view_fn_or_class, datasette)
def wrap_view_class(view_class, datasette):
async def async_view_for_class(request, send):
instance = view_class()
if inspect.iscoroutinefunction(instance.__call__):
return await async_call_with_supported_arguments(
instance.__call__,
scope=request.scope,
receive=request.receive,
send=send,
request=request,
datasette=datasette,
)
else:
return call_with_supported_arguments(
instance.__call__,
scope=request.scope,
receive=request.receive,
send=send,
request=request,
datasette=datasette,
)
return async_view_for_class
def wrap_view_function(view_fn, datasette):
@functools.wraps(view_fn)
async def async_view_fn(request, send):
if inspect.iscoroutinefunction(view_fn):

Wyświetl plik

@ -6,7 +6,7 @@ from datasette.utils import (
tilde_encode,
tilde_decode,
)
from .base import BaseView
from .base import BaseView, View
import secrets
import urllib
@ -57,13 +57,16 @@ class JsonDataView(BaseView):
)
class PatternPortfolioView(BaseView):
name = "patterns"
has_json_alternate = False
async def get(self, request):
await self.ds.ensure_permissions(request.actor, ["view-instance"])
return await self.render(["patterns.html"], request=request)
class PatternPortfolioView(View):
async def get(self, request, datasette):
await datasette.ensure_permissions(request.actor, ["view-instance"])
return Response.html(
await datasette.render_template(
"patterns.html",
request=request,
view_name="patterns",
)
)
class AuthTokenView(BaseView):