main.py

#!/usr/bin/env python3
import os
import json
import argparse
from pathlib import Path
from datetime import datetime, timedelta
from typing import List, Dict, Any, Optional

class ReminderManager:
    def __init__(self, date_file: str="~/.config/rem_cli/reminders.json"):
        self.date_file = Path(date_file).expanduser()
        self.reminders = self._load_reminders()

    def _load_reminders(self) -> List[Dict[str, Any]]:
        """Loading reminders from file"""
        if not self.date_file.exists():
            return []
        
        try:
            with open(self.date_file, 'r', encoding='utf-8') as f:
                return json.load(f)
        except (json.JSONDecodeError, FileNotFoundError):
            return[]

    def _save_reminders(self):
        """Saving reminders to a file"""
        self.date_file.parent.mkdir(parents=True, exist_ok=True)
        with open(self.date_file, 'w', encoding='utf-8') as f:
            json.dump(self.reminders, f, ensure_ascii=False, indent=2)
            
    def _parse_date(self, date_str: Optional[str]) -> Optional[str]:
        """Parsing date in ISO format"""
        if not date_str:
            return None
        
        # Trying
        date_formats = [
            '%Y-%m-%d',      # 2024-01-15
            '%d-%m-%Y',      # 15-01-2024
            '%m-%d-%Y',      # 01-15-2024
        ]
        
        for fmt in date_formats:
            try:
                dt = datetime.strptime(date_str, fmt)
                return dt.isoformat()
            except ValueError:
                continue
        
        # Catching error! :_)
        return date_str
            
    def add_reminder(self, text: str, priority: str = "medium", due_date: str = None):
        """Adding new reminder"""
        reminder = {
            "id": len(self.reminders) + 1,
            "text": text,
            "priority": priority,
            "created": datetime.now().isoformat(),
            "due_date": due_date,
            "completed": False
        }
        self.reminders.append(reminder)
        self._save_reminders()
        return reminder
    
    def _format_due_date(self, due_date_str: Optional[str]) -> str:
        """Formate date for print"""
        if not due_date_str:
            return ""
        
        try:
            # Trying parse as ISO
            due_date = datetime.fromisoformat(due_date_str)
            return f" | 📅 {due_date.strftime('%d.%m.%Y')}"
        except ValueError:
            # If not ISO
            return f" | 📅 {due_date_str}"
    
    def list_reminders(self, show_completed: bool = False):
        """Printing reminders list"""
        current_reminders = [
            r for r in self.reminders
            if show_completed or not r['completed']
        ]
        
        if not current_reminders:
            print("🎉 No active TODO!")
            return
        
        print("\n" + "=" * 60)
        print("📋 YOUR TODO")
        print("="*60)
        
        for reminder in current_reminders:
            status = "✅ " if reminder['completed'] else "⏳ "
            priority_icon = self._get_priority_icon(reminder['priority'])
            due_info = self._format_due_date(reminder['due_date'])

            print(f"{status}{priority_icon} [{reminder['id']}] {reminder['text']}{due_info}")
        
        print("="*60)
        print(f"Total: {len(current_reminders)} | Completed: {sum(1 for r in self.reminders if r['completed'])}")
        print("="*60 + "\n")
        
    def complete_reminder(self, reminder_id: int):
        """Marking reminder as completed"""
        for reminder in self.reminders:
            if reminder['id'] == reminder_id:
                reminder['completed'] = True
                self._save_reminders()
                return True
        return False
    
    def _get_priority_icon(self, priority: str) -> str:
        """Returning icon for reminder"""
        icons = {
            "high": "🔴",
            "medium": "🟡", 
            "low": "🟢"
        }
        return icons.get(priority, "⚪")
    
    def cleanup_completed(self):
        """Removing completed reminders"""
        self.reminders = [r for r in self.reminders if not r['completed']]
        self._save_reminders()
        
    def delete_reminder(self, reminder_id: int):
        """Deleting todo"""
        self.reminders = [r for r in self.reminders if r['id'] != reminder_id]
        # Recalculate ID's
        for i, reminder in enumerate(self.reminders, 1):
            reminder['id'] = i
        self._save_reminders()

def main():
    parser = argparse.ArgumentParser(description="TODO System")
    subparsers = parser.add_subparsers(dest='command')
    
    # Adding todo
    add_parser = subparsers.add_parser('add', help='Add TODO')
    add_parser.add_argument('text', help='Text of TODO')
    add_parser.add_argument('-p', '--priority', choices=['high', 'medium', 'low'], 
                          default='medium', help='Priority')
    add_parser.add_argument('-d', '--due-date', help='Due date (YYYY-MM-DD)')
    
    # Todo viewer
    list_parser = subparsers.add_parser('list', help='Show TODO')
    list_parser.add_argument('-a', '--all', action='store_true', 
                           help='Show all TODO (include completed)')
    
    # Complete todo
    complete_parser = subparsers.add_parser('complete', help='Complete TODO')
    complete_parser.add_argument('id', type=int, help='ID of TODO')
    
    # Deleting todo
    delete_parser = subparsers.add_parser('delete', help='Delete TODO')
    delete_parser.add_argument('id', type=int, help='ID todo')
    
    # Cleanup
    cleanup_parser = subparsers.add_parser('cleanup', help='Delete completed TODO')
    
    args = parser.parse_args()
    manager = ReminderManager()
    
    if args.command == 'add':
        manager.add_reminder(args.text, args.priority, args.due_date)
        print(f"✅ Added: {args.text}")
    
    elif args.command == 'list':
        manager.list_reminders(args.all)
    
    elif args.command == 'complete':
        if manager.complete_reminder(args.id):
            print(f"✅ TODO {args.id} completed!")
        else:
            print(f"❌ TODO {args.id} not found!")
            
    elif args.command == 'delete':
        if any(r['id'] == args.id for r in manager.reminders):
            manager.delete_reminder(args.id)
            print(f"🗑️ TODO {args.id} deleted!")
        else:
            print(f"❌ TODO {args.id} not found!")
    
    elif args.command == 'cleanup':
        before = len(manager.reminders)
        manager.cleanup_completed()
        after = len(manager.reminders)
        print(f"🧹 Deleted {before - after} completed TODO's!")
    
    else:
        # Showing todo if not args
        manager.list_reminders()

if __name__ == "__main__":
    main()