tracim_v2/tracim/models/workspace_menu_entries.py

# coding=utf-8
import typing
from copy import copy

from tracim.models.applications import applications
from tracim.models.data import Workspace


class WorkspaceMenuEntry(object):
    """
    Application class with data needed for frontend
    """
    def __init__(
            self,
            label: str,
            slug: str,
            icon: str,
            hexcolor: str,
            route: str,
    ) -> None:
        self.slug = slug
        self.label = label
        self.route = route
        self.hexcolor = hexcolor
        self.icon = icon


dashboard_menu_entry = WorkspaceMenuEntry(
  slug='dashboard',
  label='Dashboard',
  route='/#/workspaces/{workspace_id}/dashboard',
  hexcolor='#252525',
  icon="",
)
all_content_menu_entry = WorkspaceMenuEntry(
  slug="contents/all",
  label="All Contents",
  route="/#/workspaces/{workspace_id}/contents",
  hexcolor="#fdfdfd",
  icon="",
)


def default_workspace_menu_entry(
    workspace: Workspace,
)-> typing.List[WorkspaceMenuEntry]:
    """
    Get default menu entry for a workspace
    """
    menu_entries = [
        copy(dashboard_menu_entry),
        copy(all_content_menu_entry),
    ]
    for app in applications:
        if app.main_route:
            new_entry = WorkspaceMenuEntry(
                slug=app.slug,
                label=app.label,
                hexcolor=app.hexcolor,
                icon=app.icon,
                route=app.main_route
            )
            menu_entries.append(new_entry)

    for entry in menu_entries:
        entry.route = entry.route.replace(
            '{workspace_id}',
            str(workspace.workspace_id)
        )

    return menu_entries