From 4ca3e7b29c9cdc796da9d0a651610095f42f4494 Mon Sep 17 00:00:00 2001 From: Daniel Dybing Date: Wed, 11 Mar 2026 21:17:15 +0100 Subject: [PATCH] UI: Implement Fallout Pip-Boy inspired terminal theme --- tamigo.py | 192 +++++++++++++++++++++++++----------------------------- 1 file changed, 88 insertions(+), 104 deletions(-) diff --git a/tamigo.py b/tamigo.py index 8214e19..e77bf9f 100644 --- a/tamigo.py +++ b/tamigo.py @@ -2,6 +2,9 @@ import requests import questionary from rich.console import Console from rich.table import Table +from rich.theme import Theme +from rich.panel import Panel +from rich.text import Text import json import os import re @@ -11,10 +14,53 @@ from dotenv import load_dotenv # Load environment variables from .env if it exists load_dotenv() -console = Console() +# Fallout Pip-Boy Style Theme +fallout_theme = Theme({ + "info": "bold green", + "warning": "bold yellow", + "error": "bold red", + "success": "bold green", + "dim": "dim green", + "highlight": "bold #33ff33", + "table.header": "bold #33ff33", + "table.title": "bold #33ff33", +}) + +console = Console(theme=fallout_theme) + +# Questionary custom style +custom_style = questionary.Style([ + ('qmark', 'fg:#33ff33 bold'), # token in front of the question + ('question', 'fg:#33ff33 bold'), # question text + ('answer', 'fg:#33ff33'), # submitted answer text behind the question + ('pointer', 'fg:#33ff33 bold'), # pointer used in select input + ('highlighted', 'fg:#000000 bg:#33ff33 bold'), # pointed-at choice in select and checkbox + ('selected', 'fg:#33ff33'), # pointed-at choice in select and checkbox + ('separator', 'fg:#004400'), # separator in lists + ('instruction', 'fg:#004400 italic'), # user instructions for checkbox, reorder, etc. + ('text', 'fg:#33ff33'), # plain text + ('disabled', 'fg:#004400 italic') # disabled choices for select and checkbox +]) + +PIPO_ART = """ + _____ __ __ _____ _____ ____ + |_ _| /\ | \/ |_ _/ ____|/ __ \ + | | / \ | \ / | | || | __| | | | + | | / /\ \ | |\/| | | || | |_ | | | | + _| |_ / ____ \| | | |_| || |__| | |__| | + |_____/_/ \_\_| |_|_____\_____|\____/ + + [bold #33ff33]ROBCO INDUSTRIES (TM) TERMLINK PROTOCOL[/bold #33ff33] +""" BASE_URL = "https://api.tamigo.com" +def print_header(): + console.print(PIPO_ART, style="highlight") + console.print("[dim]-------------------------------------------[/dim]") + console.print("[highlight]WELCOME TO TAMIGO-OS v1.0.0[/highlight]") + console.print("[dim]-------------------------------------------[/dim]\n") + def parse_tamigo_date(date_str): """ Parses Tamigo date formats: @@ -45,22 +91,18 @@ class TamigoClient: self.employee_id = None def login(self, email, password): - # Try different URL patterns based on docs and common patterns urls = [ f"{BASE_URL}/login/application", f"{BASE_URL}/Login", - f"{BASE_URL}/Login/", f"{BASE_URL}/login", - f"{BASE_URL}/login/" ] - last_error = "" for url in urls: payload = { "Email": email, "Password": password, - "Name": "TamigoCLI", # For /login/application - "Key": password # For /login/application + "Name": "TamigoCLI", + "Key": password } try: headers = { @@ -72,7 +114,6 @@ class TamigoClient: if response.status_code == 200: try: data = response.json() - # Token can be in different fields self.session_token = data.get("SessionToken") or data.get("securitytoken") or data.get("Token") if self.session_token: self.user_info = data @@ -84,49 +125,12 @@ class TamigoClient: self.session_token = text self.user_info = {"Email": email} return True - - last_error = f"URL: {url}, Status: {response.status_code}" - except Exception as e: - last_error = f"URL: {url}, Error: {str(e)}" + except: + continue - console.print(f"[red]Login failed.[/red]") - console.print(f"[dim]Debug: {last_error}[/dim]") return False - def get_employee_id(self): - if self.employee_id: - return self.employee_id - - headers = { - "x-tamigo-token": self.session_token, - "securitytoken": self.session_token, - "Accept": "application/json" - } - - # Method A: User Info from Login - if self.user_info and self.user_info.get("EmployeeId"): - self.employee_id = self.user_info.get("EmployeeId") - return self.employee_id - - # Method B: My Overview - url = f"{BASE_URL}/shifts/myoverview" - try: - response = requests.get(url, headers=headers) - if response.status_code == 200: - shifts = response.json() - if shifts and len(shifts) > 0: - self.employee_id = shifts[0].get("EmployeeId") - return self.employee_id - except: - pass - - return None - def get_employee_actual_shifts(self, start_date_dt, end_date_dt): - """ - Fetches actual worked shifts using employee-accessible endpoints. - Tamigo's 'past' endpoint often limits to 60 days, so we fetch in chunks. - """ if not self.session_token: return None @@ -139,11 +143,8 @@ class TamigoClient: all_shifts = [] days_diff = (end_date_dt - start_date_dt).days - # Fetch in 60-day chunks moving backwards from end_date for i in range(0, days_diff + 1, 60): target_date = (end_date_dt - timedelta(days=i)).strftime("%Y-%m-%d") - - # Stop if we've moved past the start date current_dt = end_date_dt - timedelta(days=i) if current_dt < start_date_dt - timedelta(days=60): break @@ -159,10 +160,9 @@ class TamigoClient: response = requests.get(url, params={"securitytoken": self.session_token}) if response.status_code == 200: all_shifts.extend(response.json()) - except Exception as e: - console.print(f"[dim]Failed to fetch chunk at {target_date}: {e}[/dim]") + except: + pass - # Supplement with /shifts/period/ start_str = start_date_dt.strftime("%Y-%m-%d") end_str = end_date_dt.strftime("%Y-%m-%d") url_period = f"{BASE_URL}/shifts/period/{start_str}/{end_str}" @@ -179,14 +179,15 @@ class TamigoClient: def calculate_checkins(client): range_choice = questionary.select( - "Select period:", + "SELECT TEMPORAL RANGE:", choices=[ "Last 365 days", "Last 30 days", "This Month", "This Year", "Custom range..." - ] + ], + style=custom_style ).ask() end_date_dt = datetime.now() @@ -200,28 +201,24 @@ def calculate_checkins(client): elif range_choice == "This Year": start_date_dt = end_date_dt.replace(month=1, day=1) else: - # Custom range while True: - start_str = questionary.text("Start date (YYYY-MM-DD):", default=(datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d")).ask() - end_str = questionary.text("End date (YYYY-MM-DD):", default=datetime.now().strftime("%Y-%m-%d")).ask() + start_str = questionary.text("ENTER START DATE (YYYY-MM-DD):", default=(datetime.now() - timedelta(days=30)).strftime("%Y-%m-%d"), style=custom_style).ask() + end_str = questionary.text("ENTER END DATE (YYYY-MM-DD):", default=datetime.now().strftime("%Y-%m-%d"), style=custom_style).ask() try: start_date_dt = datetime.strptime(start_str, "%Y-%m-%d") end_date_dt = datetime.strptime(end_str, "%Y-%m-%d") if start_date_dt > end_date_dt: - console.print("[red]Start date must be before end date![/red]") + console.print("[error]ERROR: INVALID CHRONOLOGY. START MUST PRECEDE END.[/error]") continue break except ValueError: - console.print("[red]Invalid format. Please use YYYY-MM-DD.[/red]") + console.print("[error]ERROR: INVALID DATA FORMAT. USE YYYY-MM-DD.[/error]") - with console.status(f"[bold green]Fetching work history from {start_date_dt.strftime('%Y-%m-%d')} to {end_date_dt.strftime('%Y-%m-%d')}..."): + with console.status(f"[highlight]INITIALIZING DATA RETRIEVAL...[/highlight]"): data = client.get_employee_actual_shifts(start_date_dt, end_date_dt) if data: - work_days = {} # date_str -> {hours, text} - - # Filter data to ensure it's strictly within our requested range - # (Since the API chunks might return slightly more) + work_days = {} requested_start = start_date_dt.date() requested_end = end_date_dt.date() @@ -236,10 +233,7 @@ def calculate_checkins(client): date_str = dt.strftime("%Y-%m-%d") is_absent = item.get("IsAbsent", False) - - hours = (item.get("ActualShiftHours") or - item.get("CheckInOutShiftHours") or - item.get("Hours") or 0) + hours = (item.get("ActualShiftHours") or item.get("CheckInOutShiftHours") or item.get("Hours") or 0) if hours == 0 and item.get("StartTime") and item.get("EndTime"): st = parse_tamigo_date(item.get("StartTime")) @@ -248,91 +242,80 @@ def calculate_checkins(client): hours = (et - st).total_seconds() / 3600.0 has_actual = False - if hours > 0 or item.get("CheckInTime") or item.get("ActualStartTime"): - has_actual = True - if item.get("ActualShift") and item["ActualShift"].get("Shift", 0) > 0: - has_actual = True - hours = item["ActualShift"]["Shift"] - - if item.get("StartTime") and not is_absent: + if hours > 0 or item.get("CheckInTime") or item.get("ActualStartTime") or (item.get("StartTime") and not is_absent): has_actual = True if has_actual and not is_absent: if date_str not in work_days: - work_days[date_str] = { - "hours": float(hours), - "text": item.get("ActualShiftText") or item.get("ActivityName") or item.get("DepartmentName") or "Worked" - } + work_days[date_str] = {"hours": float(hours), "text": item.get("ActualShiftText") or item.get("ActivityName") or "WORKED"} else: work_days[date_str]["hours"] += float(hours) if not work_days: - console.print("[yellow]No work records found for this period.[/yellow]") + console.print("[warning]WARNING: NO LOG ENTRIES DETECTED FOR SPECIFIED PARAMETERS.[/warning]") return all_dates = sorted(work_days.keys(), reverse=True) - table = Table(title="Worked Days in Period", show_header=True, header_style="bold magenta") - table.add_column("Date", style="cyan") - table.add_column("Hours", justify="right") - table.add_column("Details", style="dim") + table = Table(title="LOG ENTRIES: ACTUAL WORKED SHIFTS", show_header=True, header_style="table.header", box=None) + table.add_column("DATE", style="highlight") + table.add_column("HOURS", justify="right") + table.add_column("DETAILS", style="dim") for day in all_dates: info = work_days[day] - table.add_row(day, f"{info['hours']:.2f}", str(info['text'])) + table.add_row(day, f"{info['hours']:.2f}", str(info['text']).upper()) console.print(table) - console.print(f"\n[bold]Work Statistics:[/bold]") - console.print(f" - Period: [cyan]{start_date_dt.strftime('%Y-%m-%d')} to {end_date_dt.strftime('%Y-%m-%d')}[/cyan]") - console.print(f" - Days Worked: [bold green]{len(all_dates)}[/bold green]") - total_hours = sum(d['hours'] for d in work_days.values()) - console.print(f" - Total Hours: [bold green]{total_hours:.2f}[/bold green]") + summary_text = f"PERIOD: {start_date_dt.strftime('%Y-%m-%d')} TO {end_date_dt.strftime('%Y-%m-%d')}\n" + summary_text += f"DAYS WORKED: {len(all_dates)}\n" + summary_text += f"TOTAL HOURS: {sum(d['hours'] for d in work_days.values()):.2f}" + + console.print(Panel(summary_text, title="SUMMARY STATISTICS", border_style="highlight", expand=False)) else: - console.print("[yellow]Could not retrieve any shift data for this period.[/yellow]") - + console.print("[error]FATAL ERROR: DATA RETRIEVAL FAILURE.[/error]") def show_profile(client): if client.user_info: console.print_json(data=client.user_info) else: - console.print("[yellow]No profile info available. Are you logged in?[/yellow]") - + console.print("[error]ERROR: ACCESS DENIED. SESSION TOKEN NOT FOUND.[/error]") def main(): + print_header() client = TamigoClient() - console.print("[bold blue]Welcome to Tamigo CLI[/bold blue]") - email = os.getenv("TAMIGO_EMAIL") if not email: - email = questionary.text("Email:").ask() + email = questionary.text("ENTER IDENTIFIER (EMAIL):", style=custom_style).ask() password = os.getenv("TAMIGO_PASSWORD") if not password: - password = questionary.password("Password:").ask() + password = questionary.password("ENTER ACCESS KEY:", style=custom_style).ask() if not email or not password: return - with console.status("[bold green]Logging in..."): + with console.status("[highlight]AUTHENTICATING WITH VAULT-TEC CLOUD...[/highlight]"): success = client.login(email, password) if success: - console.print("[bold green]Login successful![/bold green]") + console.print("[success]AUTHENTICATION SUCCESSFUL. ACCESS GRANTED.[/success]") menu(client) else: - console.print("[bold red]Login failed. Please check your credentials.[/bold red]") + console.print("[error]CRITICAL FAILURE: INVALID CREDENTIALS.[/error]") def menu(client): while True: choice = questionary.select( - "What would you like to do?", + "SELECT ACTION:", choices=[ "Calculate actual work days", "Show profile info", "Logout and Exit" - ] + ], + style=custom_style ).ask() if choice == "Calculate actual work days": @@ -340,6 +323,7 @@ def menu(client): elif choice == "Show profile info": show_profile(client) else: + console.print("[highlight]SYSTEM SHUTDOWN INITIATED...[/highlight]") break if __name__ == "__main__":