main.py

from db import db, Task
from tasks import create_task, read_all_tasks, update_task, delete_task
from scheduler import task_scheduler
from api import app
from prettytable import PrettyTable

import threading
import argparse

def tabulate_tasks(tasks):
    table = PrettyTable()
    table.field_names = ["Sno", "ID", "Name", "Execution Time", "Status", "Recurring", "Cron Schedule"]

    for sno, task in enumerate(tasks, start=1):
        table.add_row([sno, task.id, task.name, task.execution_time.strftime('%Y-%m-%d %H:%M:%S'), task.status, task.recurring, task.cron_schedule])

    return table.get_string()

def main():
    parser = argparse.ArgumentParser(description='Task Management Application')
    parser.add_argument('command', choices=['run', 'view', 'create', 'update', 'delete', 'get'], help='Command to execute')

    parser.add_argument('--id', type=int, help='Task ID')
    parser.add_argument('--name', type=str, help='Task Name')
    parser.add_argument('--execution_time', type=str, help='Task Execution Time (YYYY-MM-DD HH:MM:SS)')
    parser.add_argument('--cron_schedule', type=str, help='Cron schedule for recurring tasks (* * * * *)')

    args = parser.parse_args()

    if args.command == 'run':
        if not db.is_closed():
            db.close()

        db.connect()
        db.create_tables([Task], safe=True)
        threading.Thread(target=task_scheduler).start()

        app.run(debug=False)
    elif args.command == 'view':
        tasks = read_all_tasks()
        print(tabulate_tasks(tasks))
    elif args.command == 'create':
        create_task(args.id, args.name, args.execution_time, args.cron_schedule)
        print('Task created successfully.')
    elif args.command == 'update':
        update_task(args.id, args.name, args.execution_time, args.cron_schedule)
        print(f'Task {args.id} updated successfully.')
    elif args.command == 'delete':
        delete_task(args.id)
        print(f'Task {args.id} deleted successfully.')
    elif args.command == 'get':
        if args.id is not None:
            task = Task.get_or_none(Task.id == args.id)
            if task:
                print(f'Task found:\n{tabulate_tasks([task])}')
            else:
                print(f'Task with ID {args.id} not found.')
        else:
            print('Please provide a task ID with the --id option for the "get" command.')    

if __name__ == '__main__':
    main()