feat: projects and boards
This commit is contained in:
		@@ -5,7 +5,7 @@ from sqlalchemy import select, and_, union_all, func, Subquery, literal
 | 
			
		||||
 | 
			
		||||
from enums.profit_table_group_by import ProfitTableGroupBy
 | 
			
		||||
from models import DealService, Deal, DealStatusHistory, DealProductService, DealProduct, Service, Client, \
 | 
			
		||||
    ShippingWarehouse, BaseMarketplace, User
 | 
			
		||||
    ShippingWarehouse, BaseMarketplace, User, Project, Board
 | 
			
		||||
from schemas.statistics import GetProfitChartDataResponse, GetProfitChartDataRequest, ProfitChartDataItem, \
 | 
			
		||||
    GetProfitTableDataResponse, GetProfitTableDataRequest, ProfitTableDataItem, CommonProfitFilters
 | 
			
		||||
from services.base import BaseService
 | 
			
		||||
@@ -24,7 +24,7 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
                    'day',
 | 
			
		||||
                    Deal.created_at,
 | 
			
		||||
                ).label('date'),
 | 
			
		||||
                Deal.current_status,
 | 
			
		||||
                Deal.current_status_id,
 | 
			
		||||
            )
 | 
			
		||||
            .subquery()
 | 
			
		||||
        )
 | 
			
		||||
@@ -51,7 +51,7 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
                    'day',
 | 
			
		||||
                    last_statuses.c.changed_at,
 | 
			
		||||
                ).label('date'),
 | 
			
		||||
                Deal.current_status,
 | 
			
		||||
                Deal.current_status_id,
 | 
			
		||||
            )
 | 
			
		||||
            .join(last_statuses, last_statuses.c.deal_id == Deal.id)
 | 
			
		||||
            .subquery()
 | 
			
		||||
@@ -97,8 +97,7 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
 | 
			
		||||
        return data
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _get_stmt_deal_services(sub_filtered_status_history: Subquery):
 | 
			
		||||
    def _get_stmt_deal_services(self, sub_filtered_status_history: Subquery):
 | 
			
		||||
        return (
 | 
			
		||||
            select(
 | 
			
		||||
                Deal.id.label("deal_id"),
 | 
			
		||||
@@ -112,10 +111,24 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
            .join(DealService, Deal.id == DealService.deal_id)
 | 
			
		||||
            .join(Service, DealService.service_id == Service.id)
 | 
			
		||||
            .join(sub_filtered_status_history, Deal.id == sub_filtered_status_history.c.deal_id)
 | 
			
		||||
            .where(and_(Deal.is_deleted == False, Deal.is_accounted == True))
 | 
			
		||||
            .where(
 | 
			
		||||
                and_(
 | 
			
		||||
                    Deal.is_deleted == False,
 | 
			
		||||
                    Deal.is_accounted == True,
 | 
			
		||||
                    Deal.is_completed == True if self.is_completed_only else True
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
            .group_by(Deal.id, "date")
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _board_ids_for_project(project_id: int, stmt):
 | 
			
		||||
        board_ids_stmt = (
 | 
			
		||||
            select(Board.id)
 | 
			
		||||
            .where(Board.project_id == project_id)
 | 
			
		||||
        )
 | 
			
		||||
        return stmt.where(Deal.board_id.in_(board_ids_stmt))
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _apply_filters(request: CommonProfitFilters, stmt_deal_services, stmt_deal_product_services):
 | 
			
		||||
        if request.client_id != -1:
 | 
			
		||||
@@ -127,9 +140,21 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
            stmt_deal_product_services = stmt_deal_product_services.where(
 | 
			
		||||
                Deal.base_marketplace_key == request.base_marketplace_key)
 | 
			
		||||
 | 
			
		||||
        if request.deal_status_id != -1:
 | 
			
		||||
            stmt_deal_services = stmt_deal_services.where(Deal.current_status == request.deal_status_id)
 | 
			
		||||
            stmt_deal_product_services = stmt_deal_product_services.where(Deal.current_status == request.deal_status_id)
 | 
			
		||||
        if request.project_id != -1:
 | 
			
		||||
            stmt_deal_services = ProfitStatisticsService._board_ids_for_project(request.project_id, stmt_deal_services)
 | 
			
		||||
            stmt_deal_product_services = ProfitStatisticsService._board_ids_for_project(
 | 
			
		||||
                request.project_id,
 | 
			
		||||
                stmt_deal_product_services,
 | 
			
		||||
            )
 | 
			
		||||
 | 
			
		||||
            if request.board_id != -1:
 | 
			
		||||
                stmt_deal_services = stmt_deal_services.where(Deal.board_id == request.board_id)
 | 
			
		||||
                stmt_deal_product_services = stmt_deal_product_services.where(Deal.board_id == request.board_id)
 | 
			
		||||
 | 
			
		||||
                if request.deal_status_id != -1:
 | 
			
		||||
                    stmt_deal_services = stmt_deal_services.where(Deal.current_status_id == request.deal_status_id)
 | 
			
		||||
                    stmt_deal_product_services = stmt_deal_product_services.where(
 | 
			
		||||
                        Deal.current_status_id == request.deal_status_id)
 | 
			
		||||
 | 
			
		||||
        if request.manager_id != -1:
 | 
			
		||||
            stmt_deal_services = stmt_deal_services.where(Deal.manager_id == request.manager_id)
 | 
			
		||||
@@ -137,8 +162,7 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
 | 
			
		||||
        return stmt_deal_services, stmt_deal_product_services
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _get_stmt_product_services():
 | 
			
		||||
    def _get_stmt_product_services(self):
 | 
			
		||||
        return (
 | 
			
		||||
            select(
 | 
			
		||||
                Deal.id.label("deal_id"),
 | 
			
		||||
@@ -154,7 +178,13 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
            .join(Service, DealProductService.service_id == Service.id)
 | 
			
		||||
            .where(and_(Deal.is_deleted == False, Deal.is_accounted == True))
 | 
			
		||||
            .where(
 | 
			
		||||
                and_(
 | 
			
		||||
                    Deal.is_deleted == False,
 | 
			
		||||
                    Deal.is_accounted == True,
 | 
			
		||||
                    Deal.is_completed == True if self.is_completed_only else True,
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
            .group_by(Deal.id)
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
@@ -226,16 +256,47 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _join_and_group_by_statuses(stmt):
 | 
			
		||||
    def _join_and_group_by_projects(stmt):
 | 
			
		||||
        return (
 | 
			
		||||
            select(
 | 
			
		||||
                Deal.current_status.label("grouped_value"),
 | 
			
		||||
                Project.id,
 | 
			
		||||
                Project.name.label("grouped_value"),
 | 
			
		||||
                func.count(stmt.c.deal_id).label("deals_count"),
 | 
			
		||||
                func.sum(stmt.c.revenue).label("revenue"),
 | 
			
		||||
                func.sum(stmt.c.profit).label("profit"),
 | 
			
		||||
            )
 | 
			
		||||
            .join(Deal, Deal.id == stmt.c.deal_id)
 | 
			
		||||
            .group_by(Deal.current_status)
 | 
			
		||||
            .join(Board, Board.id == Deal.board_id)
 | 
			
		||||
            .join(Project, Project.id == Board.project_id)
 | 
			
		||||
            .group_by(Project.id, Project.name)
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _join_and_group_by_boards(stmt):
 | 
			
		||||
        return (
 | 
			
		||||
            select(
 | 
			
		||||
                Board.id,
 | 
			
		||||
                Board.name.label("grouped_value"),
 | 
			
		||||
                func.count(stmt.c.deal_id).label("deals_count"),
 | 
			
		||||
                func.sum(stmt.c.revenue).label("revenue"),
 | 
			
		||||
                func.sum(stmt.c.profit).label("profit"),
 | 
			
		||||
            )
 | 
			
		||||
            .join(Deal, Deal.id == stmt.c.deal_id)
 | 
			
		||||
            .join(Board, Board.id == Deal.board_id)
 | 
			
		||||
            .group_by(Board.id, Board.name)
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def _join_and_group_by_statuses(stmt):
 | 
			
		||||
        return (
 | 
			
		||||
            select(
 | 
			
		||||
                Deal.current_status_id.label("grouped_value"),
 | 
			
		||||
                func.count(stmt.c.deal_id).label("deals_count"),
 | 
			
		||||
                func.sum(stmt.c.revenue).label("revenue"),
 | 
			
		||||
                func.sum(stmt.c.profit).label("profit"),
 | 
			
		||||
            )
 | 
			
		||||
            .join(Deal, Deal.id == stmt.c.deal_id)
 | 
			
		||||
            .group_by(Deal.current_status_id)
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
@@ -320,6 +381,7 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
 | 
			
		||||
    async def _get_data_grouped_by_date(self, request: CommonProfitFilters, is_chart: bool = True):
 | 
			
		||||
        self.date_from, self.date_to = request.date_range
 | 
			
		||||
        self.is_completed_only = request.is_completed_only
 | 
			
		||||
        self.filters = request
 | 
			
		||||
 | 
			
		||||
        sub_deals_dates = self._get_deals_dates(request.deal_status_id)
 | 
			
		||||
@@ -399,6 +461,18 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
 | 
			
		||||
        return await self._table_data_from_stmt(stmt_grouped_by_clients)
 | 
			
		||||
 | 
			
		||||
    async def _get_table_grouped_by_projects(self, request: GetProfitTableDataRequest) -> GetProfitTableDataResponse:
 | 
			
		||||
        sub_grouped_by_deals = self._get_common_table_grouped(request)
 | 
			
		||||
        stmt_grouped_by_projects = self._join_and_group_by_projects(sub_grouped_by_deals)
 | 
			
		||||
 | 
			
		||||
        return await self._table_data_from_stmt(stmt_grouped_by_projects)
 | 
			
		||||
 | 
			
		||||
    async def _get_table_grouped_by_boards(self, request: GetProfitTableDataRequest) -> GetProfitTableDataResponse:
 | 
			
		||||
        sub_grouped_by_deals = self._get_common_table_grouped(request)
 | 
			
		||||
        stmt_grouped_by_boards = self._join_and_group_by_boards(sub_grouped_by_deals)
 | 
			
		||||
 | 
			
		||||
        return await self._table_data_from_stmt(stmt_grouped_by_boards)
 | 
			
		||||
 | 
			
		||||
    async def _get_table_grouped_by_statuses(self, request: GetProfitTableDataRequest) -> GetProfitTableDataResponse:
 | 
			
		||||
        date_from, date_to = request.date_range
 | 
			
		||||
 | 
			
		||||
@@ -449,6 +523,10 @@ class ProfitStatisticsService(BaseService):
 | 
			
		||||
                return await self._get_table_grouped_by_dates(request)
 | 
			
		||||
            case ProfitTableGroupBy.BY_CLIENTS:
 | 
			
		||||
                return await self._get_table_grouped_by_clients(request)
 | 
			
		||||
            case ProfitTableGroupBy.BY_PROJECTS:
 | 
			
		||||
                return await self._get_table_grouped_by_projects(request)
 | 
			
		||||
            case ProfitTableGroupBy.BY_BOARDS:
 | 
			
		||||
                return await self._get_table_grouped_by_boards(request)
 | 
			
		||||
            case ProfitTableGroupBy.BY_STATUSES:
 | 
			
		||||
                return await self._get_table_grouped_by_statuses(request)
 | 
			
		||||
            case ProfitTableGroupBy.BY_WAREHOUSES:
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user