mirror of https://github.com/snachodog/mybuddy.git
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
from collections import OrderedDict
|
|
|
|
from typing import NamedTuple, List, Any
|
|
|
|
from django.urls import include, path
|
|
from rest_framework import routers
|
|
from rest_framework.schemas import get_schema_view
|
|
|
|
from . import views
|
|
|
|
|
|
class ExtraPath(NamedTuple):
|
|
path: str
|
|
reverese_name: str
|
|
route: Any
|
|
|
|
|
|
class CustomRouterWithExtraPaths(routers.DefaultRouter):
|
|
extra_api_urls: List[ExtraPath]
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.extra_api_urls = []
|
|
|
|
def add_detail_path(self, url_path, reverese_name, *args, **kwargs):
|
|
self.extra_api_urls = self.extra_api_urls or []
|
|
kwargs["name"] = reverese_name
|
|
self.extra_api_urls.append(
|
|
ExtraPath(url_path, reverese_name, path(url_path, *args, **kwargs))
|
|
)
|
|
|
|
def get_api_root_view(self, api_urls=None):
|
|
api_root_dict = OrderedDict()
|
|
list_name = self.routes[0].name
|
|
for prefix, viewset, basename in self.registry:
|
|
api_root_dict[prefix] = list_name.format(basename=basename)
|
|
for extra_path in self.extra_api_urls:
|
|
api_root_dict[extra_path.path] = extra_path.reverese_name
|
|
|
|
return self.APIRootView.as_view(api_root_dict=api_root_dict)
|
|
|
|
@property
|
|
def urls(self):
|
|
return super().urls + [e.route for e in self.extra_api_urls]
|
|
|
|
|
|
router = CustomRouterWithExtraPaths()
|
|
router.register(r"bmi", views.BMIViewSet)
|
|
router.register(r"changes", views.DiaperChangeViewSet)
|
|
router.register(r"children", views.ChildViewSet)
|
|
router.register(r"feedings", views.FeedingViewSet)
|
|
router.register(r"head-circumference", views.HeadCircumferenceViewSet)
|
|
router.register(r"height", views.HeightViewSet)
|
|
router.register(r"notes", views.NoteViewSet)
|
|
router.register(r"pumping", views.PumpingViewSet)
|
|
router.register(r"sleep", views.SleepViewSet)
|
|
router.register(r"tags", views.TagViewSet)
|
|
router.register(r"temperature", views.TemperatureViewSet)
|
|
router.register(r"timers", views.TimerViewSet)
|
|
router.register(r"tummy-times", views.TummyTimeViewSet)
|
|
router.register(r"weight", views.WeightViewSet)
|
|
|
|
router.add_detail_path("profile", "profile", views.ProfileView.as_view())
|
|
router.add_detail_path(
|
|
"schema",
|
|
"openapi-schema",
|
|
get_schema_view(
|
|
title="Baby Buddy API",
|
|
version=1,
|
|
description="API documentation for the Baby Buddy application",
|
|
),
|
|
)
|
|
|
|
|
|
app_name = "api"
|
|
|
|
urlpatterns = [
|
|
path("api/", include(router.urls)),
|
|
path("api/auth/", include("rest_framework.urls", namespace="rest_framework")),
|
|
]
|