summaryrefslogtreecommitdiffstats
path: root/girok/utils/display.py
diff options
context:
space:
mode:
Diffstat (limited to 'girok/utils/display.py')
-rw-r--r--girok/utils/display.py349
1 files changed, 11 insertions, 338 deletions
diff --git a/girok/utils/display.py b/girok/utils/display.py
index 362dfa3..563cb6e 100644
--- a/girok/utils/display.py
+++ b/girok/utils/display.py
@@ -1,351 +1,24 @@
-from datetime import datetime
-import calendar
+import shutil
from rich import print
-from rich.console import Console
-from rich.tree import Tree
from rich.align import Align
-from rich import box
+from rich.console import Console
from rich.style import Style
-from rich.table import Table, Column
from rich.text import Text
-from rich.align import Align
-from rich.console import Console
-from rich.live import Live
-import typer
-from rich import print
-from rich.console import Console
-from rich.table import Table
-from rich.progress import Progress
-import shutil
-import time
-import girok.utils.task as task_utils
-import girok.utils.calendar as calendar_utils
-import girok.utils.general as general_utils
-from girok.config import get_config
-import girok.constants as constants
+
+from girok.constants import DisplayBoxType
console = Console()
-cfg = get_config()
-def display_categories(cats_dict: dict, highlight_cat=None):
- tree = Tree("")
- for idx, cat in enumerate(cats_dict):
- display_category(
- cats_dict,
- cat,
- top_most=True,
- tree=tree,
- cumul_path=cat,
- highlight_cat=highlight_cat,
- color=cats_dict[cat]['color']
- )
- console.print(tree)
+def center_print(text: str, text_type: DisplayBoxType, wrap: bool = False) -> None:
+ style = Style(color=text_type.text_color_hex, bgcolor=text_type.bg_color_hex)
-def display_category(
- cats_dict: dict,
- cat: str,
- top_most=True,
- tree: Tree=None,
- cumul_path="",
- highlight_cat=None,
- color=None,
-):
- style = Style(
- color=constants.TABLE_HEADER_COLOR if highlight_cat != cumul_path else "#B9D66A",
- bold=True
+ width = (
+ shutil.get_terminal_size().columns // 2
+ if wrap
+ else shutil.get_terminal_size().columns
)
- # if top_most:
- circle = Text(constants.CIRCLE_EMOJI + " ", style=Style(color=constants.CIRCLE_COLOR[color]))
-
- dir_name = Text(f"{cat}", style=style)
- # if top_most:
- dir_name = Text.assemble(circle, dir_name)
-
- tree = tree.add(dir_name)
- for sub in cats_dict[cat]['subcategories'].keys():
- display_category(
- cats_dict[cat]['subcategories'],
- sub,
- top_most=False,
- tree=tree,
- cumul_path=cumul_path + "/" + sub,
- highlight_cat=highlight_cat,
- color=color
- )
-
-
-def center_print(text, type: str = None, wrap: bool = False):
- if type == "title":
- style = Style(
- color=constants.DISPLAY_TERMINAL_COLOR_TITLE_TEXT,
- bgcolor=constants.DISPLAY_TERMINAL_COLOR_TITLE_BACKGROUND
- )
- elif type == "success":
- style = Style(
- color=constants.DISPLAY_TERMINAL_COLOR_SUCCESS_TEXT,
- bgcolor=constants.DISPLAY_TERMINAL_COLOR_SUCCESS_BACKGROUND
- )
- elif type == "error":
- style = Style(
- color=constants.DISPLAY_TERMINAL_COLOR_ERROR_TEXT,
- bgcolor=constants.DISPLAY_TERMINAL_COLOR_ERROR_BACKGROUND
- )
- elif type == "warning":
- style = Style(
- color=constants.DISPLAY_TERMINAL_COLOR_WARNING_TEXT,
- bgcolor=constants.DISPLAY_TERMINAL_COLOR_WARNING_BACKGROUND
- )
- else:
- style = Style(
- color=constants.DISPLAY_TERMINAL_COLOR_TITLE_TEXT,
- bgcolor=constants.DISPLAY_TERMINAL_COLOR_TITLE_BACKGROUND
- )
- if wrap:
- width = shutil.get_terminal_size().columns // 2
- else:
- width = shutil.get_terminal_size().columns
-
content = Text(text, style=style)
- console.print(Align.center(content, style=style, width=width), height=100)
-
-
-def display_tasks_by_category(
- task_tree,
- color_dict,
- marked_task_id: int = None,
- marked_color: str = constants.TABLE_TASK_DELETED_COLOR
-):
- # print(task_tree)
- task_ids_cache = {}
- tree = Tree("")
- for idx, cat_name in enumerate(task_tree):
- tree.add(display_category_with_tasks(
- cat_name=cat_name,
- task_tree=task_tree[cat_name],
- color_dict=color_dict,
- top_most=True,
- task_ids_cache=task_ids_cache,
- marked_task_id=marked_task_id,
- marked_color=marked_color
- ))
- general_utils.cache_task_ids(cfg=cfg, cache=task_ids_cache)
- return tree
-
-
-def display_category_with_tasks(
- cat_name,
- task_tree: dict,
- color_dict,
- top_most,
- task_ids_cache,
- marked_task_id: int = None,
- marked_color: str = constants.TABLE_TASK_DELETED_COLOR,
-):
- dir_name = Text(f"{cat_name}", style=Style(color=constants.TABLE_HEADER_COLOR, bold=True))
- dir_name = Text.assemble(
- Text(f"{constants.CIRCLE_EMOJI} ", style=Style(color=constants.CIRCLE_COLOR[color_dict[cat_name]])),
- dir_name
- )
-
- tree = Tree(dir_name)
-
- for sub_cat_name in task_tree['sub_categories']:
- sub_tree = display_category_with_tasks(
- cat_name=sub_cat_name,
- task_tree=task_tree['sub_categories'][sub_cat_name],
- color_dict=color_dict,
- top_most=False,
- task_ids_cache=task_ids_cache,
- marked_task_id=marked_task_id,
- marked_color=marked_color
- )
- tree.add(sub_tree)
-
- for task in task_tree['tasks']:
- # Cache task id
- if not(task['task_id'] == marked_task_id and marked_color == constants.TABLE_TASK_DELETED_COLOR):
- task_ids_cache[len(task_ids_cache) + 1] = task['task_id']
- deadline = calendar_utils.get_date_obj_from_str_separated_by_T(str(task['deadline']))
- # deadline = datetime.strptime(task['deadline'], "%Y-%m-%dT%H:%M:%S")
- year, month, day = deadline.year, deadline.month, deadline.day
- h, m, s = str(deadline.time()).split(":")
-
- weekday_name = task_utils.get_weekday_name_from_date(year, month, day)
-
- h = int(h)
- is_time = task['is_time']
- afternoon = h >= 12
- if h > 12:
- h -= 12
-
- time = f" {h}:{m} {'PM' if afternoon else 'AM'} " if is_time else ' '
- remaining_days = task_utils.get_day_offset_between_two_dates(datetime.now(), deadline)
- day_offset_message = f"{remaining_days} days left" if remaining_days > 0 else f"{abs(remaining_days)} days passed"
-
- if remaining_days == 0:
- day_offset_message = "Due Today"
-
- if not(task['task_id'] == marked_task_id and marked_color == constants.TABLE_TASK_DELETED_COLOR):
- task_id = Text(f"[{len(task_ids_cache)}] ", style=Style(color="white" if task['task_id'] != marked_task_id else marked_color))
- else:
- task_id = ""
- task_name = Text(task['name'], style=Style(color=constants.TABLE_TASK_NAME_COLOR if task['task_id'] != marked_task_id else marked_color, bold=True))
- task_remaining_days = Text(f" [{day_offset_message}]", style=Style(color='#F5F7F2' if task['task_id'] != marked_task_id else marked_color))
- task_date = Text(f" - {year}/{month}/{day}{time}{weekday_name}", style=Style(color="white" if task['task_id'] != marked_task_id else marked_color))
- node_name = Text.assemble(task_id, task_name, task_remaining_days, task_date, style=Style(strike=True if task['task_id'] == marked_task_id and marked_color == constants.TABLE_TASK_DELETED_COLOR else False))
-
- tree.add(node_name)
-
- return tree
-
-
-def display_tasks_by_list(tasks: list, marked_task_id: int = None, color: str = constants.TABLE_TASK_DELETED_COLOR):
- table = build_task_table(tasks, marked_task_id, color)
- print(Align(table, align="center"))
-
-
-def build_task_table(tasks: list, marked_task_id: int = None, color: str = constants.TABLE_TASK_DELETED_COLOR):
- tasks = sorted(tasks, key=lambda x: calendar_utils.get_date_obj_from_str_separated_by_T(x['deadline']))
- num_tasks = len(tasks)
- table = Table(
- Column(header="ID", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TASK_NAME_COLOR)),
- Column(header="Name", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TASK_NAME_COLOR)),
- Column(header="Category", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TEXT_COLOR)),
- Column(header="Deadline", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color="green")),
- Column(header="Time", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TEXT_COLOR)),
- Column(header="Tag", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TEXT_COLOR)),
- Column(header="Priority", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TEXT_COLOR)),
- Column(header="Remaining", justify="center", header_style=Style(color=constants.TABLE_HEADER_COLOR), style=Style(color=constants.TABLE_TEXT_COLOR)),
- box=box.SIMPLE_HEAVY,
- show_lines=False if num_tasks > 15 else True,
- border_style=Style(color="#D7E1C9", bold=True)
- )
-
- task_ids_cache = {}
- for idx, task in enumerate(tasks):
- # Cache task id
- task_ids_cache[idx + 1] = task['task_id']
-
- date = calendar_utils.get_date_obj_from_str_separated_by_T(task['deadline'])
- deadline = f"{date.day} {calendar.month_name[date.month]} {date.year}"
- h, m, s = str(date.time()).split(":")
- h = int(h)
- is_time = task['is_time']
- afternoon = h >= 12
- if h > 12:
- h -= 12
- time = f"{h}:{m} {'PM' if afternoon else 'AM'}" if is_time else '-'
-
- remaining_days = task_utils.get_day_offset_between_two_dates(datetime.now(), date)
- day_offset_message = f"{remaining_days} days left" if remaining_days > 0 else f"{abs(remaining_days)} days passed"
- if remaining_days == 0:
- day_offset_message = "Due Today"
- circle = Text(constants.CIRCLE_EMOJI + " ", style=Style(color=constants.CIRCLE_COLOR[task['color']]))
- task_name = Text.assemble(circle, task['name'], style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TASK_NAME_COLOR, bold=True))
-
- table.add_row(
- Text(str(idx + 1), style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TEXT_COLOR)),
- task_name,
- Text(task['task_category_full_path'] if task['task_category_full_path'] else "None", style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TEXT_COLOR)),
- Text(deadline, style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_DEADLINE_COLOR)),
- Text(time, style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_DEADLINE_COLOR)),
- Text(task['tag'] if task['tag'] else '-', style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TEXT_COLOR)),
- Text(str(task['priority']) if task['priority'] else '-', style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TEXT_COLOR)),
- Text(day_offset_message, style=Style(color=color if marked_task_id == task['task_id'] else constants.TABLE_TEXT_COLOR))
- )
-
- general_utils.cache_task_ids(cfg=cfg, cache=task_ids_cache)
-
- return table
-
-
-def register_welcome():
- print()
- with Progress() as progress:
- task = progress.add_task("[green]Registering...", total=100)
-
- while not progress.finished:
- progress.update(task, advance=1)
- time.sleep(0.02)
-
- welcome_msg = """[yellow] __ __ ___
-/\ \ __/\ \ /\_ \
-\ \ \/\ \ \ \ __\//\ \ ___ ___ ___ ___ __
- \ \ \ \ \ \ \ /'__`\\\ \ \ /'___\ / __`\ /' __` __`\ /'__`\
- \ \ \_/ \_\ \/\ __/ \_\ \_/\ \__//\ \L\ \/\ \/\ \/\ \/\ __/
- \ `\___x___/\ \____\/\____\ \____\ \____/\ \_\ \_\ \_\ \____\\
- '\/__//__/ \/____/\/____/\/____/\/___/ \/_/\/_/\/_/\/____/
-
-
- __ ____ __ __
- /\ \__ /\ _`\ __ /\ \ /\ \
- \ \ ,_\ ___ \ \ \L\_\/\_\ _ __ ___\ \ \/'\ \ \ \
- \ \ \/ / __`\ \ \ \L_L\/\ \/\`'__\/ __`\ \ , < \ \ \
- \ \ \_/\ \L\ \ \ \ \/, \ \ \ \ \//\ \L\ \ \ \\\\\`\ \ \_\
- \ \__\ \____/ \ \____/\ \_\ \_\\\ \____/\ \_\ \_\ \/\_\\
- \/__/\/___/ \/___/ \/_/\/_/ \/___/ \/_/\/_/ \/_/
-
- [/yellow]"""
- l1 = " __ __ ___ "
- l2 = "/\ \ __/\ \ /\_ \ "
- l3 = "\ \ \/\ \ \ \ __\//\ \ ___ ___ ___ ___ __ "
- l4 = " \ \ \ \ \ \ \ /'__`\\\ \ \ /'___\ / __`\ /' __` __`\ /'__`\ "
- l5 = " \ \ \_/ \_\ \/\ __/ \_\ \_/\ \__//\ \L\ \/\ \/\ \/\ \/\ __/ "
- l6 = " \ `\___x___/\ \____\/\____\ \____\ \____/\ \_\ \_\ \_\ \____\\"
- l7 = " '\/__//__/ \/____/\/____/\/____/\/___/ \/_/\/_/\/_/\/____/"
-
- r1 = " __ ____ __ __ "
- r2 = " /\ \__ /\ _`\ __ /\ \ /\ \ "
- r3 = " \ \ ,_\ ___ \ \ \L\_\/\_\ _ __ ___\ \ \/'\ \ \ \ "
- r4 = " \ \ \/ / __`\ \ \ \L_L\/\ \/\`'__\/ __`\ \ , < \ \ \ "
- r5 = " \ \ \_/\ \L\ \ \ \ \/, \ \ \ \ \//\ \L\ \ \ \\\\\`\ \ \_\ "
- r6 = " \ \__\ \____/ \ \____/\ \_\ \_\\\ \____/\ \_\ \_\ \/\_\\"
- r7 = " \/__/\/___/ \/___/ \/_/\/_/ \/___/ \/_/\/_/ \/_/"
-
-
-# console.print(Align.center("""[yellow] __ __ ___
-# /\ \ __/\ \ /\_ \
-# \ \ \/\ \ \ \ __\//\ \ ___ ___ ___ ___ __
-# \ \ \ \ \ \ \ /'__`\\\ \ \ /'___\ / __`\ /' __` __`\ /'__`\
-# \ \ \_/ \_\ \/\ __/ \_\ \_/\ \__//\ \L\ \/\ \/\ \/\ \/\ __/
-# \ `\___x___/\ \____\/\____\ \____\ \____/\ \_\ \_\ \_\ \____\\
-# '\/__//__/ \/____/\/____/\/____/\/___/ \/_/\/_/\/_/\/____/
-
-
-# __ ____ __ __
-# /\ \__ /\ _`\ __ /\ \ /\ \
-# \ \ ,_\ ___ \ \ \L\_\/\_\ _ __ ___\ \ \/'\ \ \ \
-# \ \ \/ / __`\ \ \ \L_L\/\ \/\`'__\/ __`\ \ , < \ \ \
-# \ \ \_/\ \L\ \ \ \ \/, \ \ \ \ \//\ \L\ \ \ \\\\\`\ \ \_\
-# \ \__\ \____/ \ \____/\ \_\ \_\\\ \____/\ \_\ \_\ \/\_\\
-# \/__/\/___/ \/___/ \/_/\/_/ \/___/ \/_/\/_/ \/_/
-
-# [/yellow]"""))
-
- print("\n"*5)
- screen_width = shutil.get_terminal_size().columns // 2
- with Live("", refresh_per_second=4) as live:
- num_iters = (70 * screen_width) // 109
- for _ in range(num_iters):
- space = _
- msg = " " * space + l1 + "\n" + " " * space + l2 + "\n" + " " * space + l3 + "\n" + " " * space + l4 + "\n" + " " * space + l5 + "\n" + " " * space + l6 + "\n" + " " * space + l7 + "\n"
- live.update(Text(msg, style=Style(color="yellow")))
- time.sleep(0.009)
- print("\n")
-
- with Live("", refresh_per_second=4) as live:
- for _ in range(num_iters):
- space = _
- msg = " " * space + r1 + "\n" + " " * space + r2 + "\n" + " " * space + r3 + "\n" + " " * space + r4 + "\n" + " " * space + r5 + "\n" + " " * space + r6 + "\n" + " " * space + r7 + "\n"
- live.update(Text(msg, style=Style(color="yellow")))
- time.sleep(0.009)
-
-
- print("\n" * 1)
- console.print(" " * ((70 * screen_width) // 109 + 20) + "Enter [green]girok login[/green] to log-in to your account.")
- print("\n" * 10)
-
-
+ console.print(Align.center(content, style=style, width=width), height=50)
generated by cgit v1.2.3 (git 2.25.1) at 2024-06-12 00:23:49 +0000