From 9700ff26070af0262af02ded05e7e2a48871666e Mon Sep 17 00:00:00 2001 From: Eden Kirin Date: Sun, 27 Aug 2023 11:16:31 +0200 Subject: [PATCH] Cleanup --- app/controllers/machine.py | 52 +++++--------------------------------- 1 file changed, 6 insertions(+), 46 deletions(-) diff --git a/app/controllers/machine.py b/app/controllers/machine.py index 3bda879..9cf9c27 100644 --- a/app/controllers/machine.py +++ b/app/controllers/machine.py @@ -1,17 +1,10 @@ from __future__ import annotations -from dataclasses import dataclass -from typing import TYPE_CHECKING, Optional, cast, TypeVar, Type, Generic +from typing import TYPE_CHECKING, Optional -from litestar import Controller, delete, get, post, put -from litestar.contrib.repository.filters import LimitOffset, SearchFilter, FilterTypes +from litestar import Controller, get +from litestar.contrib.repository.filters import LimitOffset, SearchFilter from litestar.di import Provide -from litestar.pagination import ( - AbstractAsyncOffsetPaginator, - AbstractSyncClassicPaginator, -) -from litestar.status_codes import HTTP_200_OK -from sqlalchemy import ScalarResult, func, select from sqlalchemy.ext.asyncio import AsyncSession from app.domain.machine import ( @@ -35,24 +28,6 @@ def provides_service(db_session: AsyncSession) -> Service: return Service(Repository(session=db_session)) -# class MachineOffsetPaginator(AbstractAsyncOffsetPaginator[Machine]): -# def __init__( -# self, async_session: AsyncSession -# ) -> None: # 'async_session' dependency will be injected here. -# self.async_session = async_session -# -# async def get_total(self) -> int: -# return cast( -# "int", await self.async_session.scalar(select(func.count(Machine.id))) -# ) -# -# async def get_items(self, limit: int, offset: int) -> list[Machine]: -# people: ScalarResult = await self.async_session.scalars( -# select(Machine).slice(offset, limit) -# ) -# return list(people.all()) - - class MachineController(Controller): dto = MachineWriteDTO return_dto = MachineReadDTO @@ -81,24 +56,9 @@ class MachineController(Controller): content = await service.list(*filters) return ObjectListResponse(content=content) - # @post() - # async def create_author(self, data: Machine, service: Service) -> Machine: - # return await service.create(data) - # @get(DETAIL_ROUTE) - async def get_machine(self, service: Service, machine_id: int) -> ObjectResponse[Machine]: + async def get_machine( + self, service: Service, machine_id: int + ) -> ObjectResponse[Machine]: content = await service.get(machine_id) return ObjectResponse(content=content) - - # - # @put(DETAIL_ROUTE) - # async def update_author( - # self, data: Machine, service: Service, author_id: UUID - # ) -> Machine: - # """Update an author.""" - # return await service.update(author_id, data) - # - # @delete(DETAIL_ROUTE, status_code=HTTP_200_OK) - # async def delete_author(self, service: Service, author_id: UUID) -> Machine: - # """Delete Author by ID.""" - # return await service.delete(author_id)