Thông báo: Download 4 khóa học Python từ cơ bản đến nâng cao tại đây.
Hướng dẫn sử dụng Rich, Typer, và SQLite trên terminal bằng Python
Trong hướng dẫn này, chúng ta sẽ xây dựng một ứng dụng dòng lệnh (CLI app) để quản lý danh sách công việc và việc cần làm (todo list). Ứng dụng này sẽ sử dụng:
- Typer: Tạo giao diện dòng lệnh một cách dễ dàng và tiện lợi.
- Rich: Tạo giao diện terminal với màu sắc và định dạng đẹp mắt.
- SQLite3: Quản lý dữ liệu bằng cơ sở dữ liệu đơn giản.
Công nghệ sử dụng và cài đặt trong Python
- Rich: Thêm văn bản đẹp mắt và định dạng vào terminal.
- Typer: Xây dựng giao diện CLI dễ dàng và linh hoạt.
- SQLite3: Module cơ sở dữ liệu được tích hợp sẵn trong Python.
Cài đặt Rich và Typer thông qua pip
.
pip install rich typer
SQLite3 là module có sẵn trong Python, không cần cài đặt thêm.
Bước 1: Khởi tạo ứng dụng với Typer
Tạo file todocli.py
và định nghĩa các chức năng cơ bản như thêm, xóa, cập nhật, hoàn thành và hiển thị công việc.
Code mẫu:
Bài viết này được đăng tại [free tuts .net]
import typer app = typer.Typer() @app.command(short_help='Thêm một công việc mới') def add(task: str, category: str): typer.echo(f"Đã thêm: {task}, danh mục: {category}") @app.command(short_help='Xóa công việc theo vị trí') def delete(position: int): typer.echo(f"Đã xóa công việc tại vị trí: {position}") @app.command(short_help='Cập nhật công việc') def update(position: int, task: str = None, category: str = None): typer.echo(f"Cập nhật công việc tại vị trí: {position}") @app.command(short_help='Đánh dấu hoàn thành công việc') def complete(position: int): typer.echo(f"Đã hoàn thành công việc tại vị trí: {position}") @app.command(short_help='Hiển thị danh sách công việc') def show(): typer.echo("Danh sách công việc:") if __name__ == "__main__": app()
Chạy lệnh sau trong terminal để kiểm tra:
python todocli.py --help python todocli.py add "Làm bài tập Python" "Học tập"
Bước 2: Thêm Rich để hiển thị danh sách công việc đẹp hơn
Cập nhật hàm show()
để hiển thị danh sách công việc dưới dạng bảng với Rich:
Code mẫu:
from rich.console import Console from rich.table import Table console = Console() @app.command(short_help='Hiển thị danh sách công việc') def show(): tasks = [ ("Học Python", "Học tập"), ("Tập thể dục", "Thể thao"), ] console.print("[bold magenta]Danh sách công việc:[/bold magenta]") table = Table(show_header=True, header_style="bold blue") table.add_column("#", style="dim", width=6) table.add_column("Công việc", min_width=20) table.add_column("Danh mục", min_width=12, justify="right") table.add_column("Trạng thái", min_width=12, justify="right") def get_category_color(category): COLORS = {'Học tập': 'green', 'Thể thao': 'cyan', 'Giải trí': 'red'} return COLORS.get(category, 'white') for idx, task in enumerate(tasks, start=1): c = get_category_color(task[1]) table.add_row(str(idx), task[0], f'[{c}]{task[1]}[/{c}]', "❌") console.print(table)
Chạy lệnh sau để kiểm tra:
python todocli.py show
Bước 3: Tạo lớp Todo để quản lý công việc
Tạo file model.py
và định nghĩa lớp Todo
để lưu thông tin công việc:
import datetime class Todo: def __init__(self, task, category, date_added=None, date_completed=None, status=1, position=None): self.task = task self.category = category self.date_added = date_added or datetime.datetime.now().isoformat() self.date_completed = date_completed self.status = status # 1: Chưa hoàn thành, 2: Đã hoàn thành self.position = position def __repr__(self): return f"({self.task}, {self.category}, {self.date_added}, {self.date_completed}, {self.status}, {self.position})"
Bước 4: Tạo cơ sở dữ liệu SQLite
Tạo file database.py
để quản lý các thao tác với cơ sở dữ liệu.
Code mẫu:
import sqlite3 from typing import List from model import Todo conn = sqlite3.connect('todos.db') c = conn.cursor() def create_table(): c.execute("""CREATE TABLE IF NOT EXISTS todos ( task text, category text, date_added text, date_completed text, status integer, position integer )""") create_table() def insert_todo(todo: Todo): c.execute('SELECT COUNT(*) FROM todos') count = c.fetchone()[0] todo.position = count with conn: c.execute('INSERT INTO todos VALUES (?, ?, ?, ?, ?, ?)', (todo.task, todo.category, todo.date_added, todo.date_completed, todo.status, todo.position)) def get_all_todos() -> List[Todo]: c.execute('SELECT * FROM todos') results = c.fetchall() return [Todo(*result) for result in results] def delete_todo(position: int): with conn: c.execute("DELETE FROM todos WHERE position=?", (position,))
Bước 5: Tích hợp tất cả vào todocli.py
Kết nối các chức năng CLI với cơ sở dữ liệu.
Code mẫu:
from database import get_all_todos, insert_todo, delete_todo from model import Todo @app.command(short_help='Thêm công việc') def add(task: str, category: str): todo = Todo(task, category) insert_todo(todo) show() @app.command(short_help='Xóa công việc') def delete(position: int): delete_todo(position - 1) show() @app.command(short_help='Hiển thị danh sách công việc') def show(): tasks = get_all_todos() console.print("[bold magenta]Danh sách công việc:[/bold magenta]") table = Table(show_header=True, header_style="bold blue") table.add_column("#", style="dim", width=6) table.add_column("Công việc", min_width=20) table.add_column("Danh mục", min_width=12, justify="right") table.add_column("Trạng thái", min_width=12, justify="right") for idx, task in enumerate(tasks, start=1): is_done = "✅" if task.status == 2 else "❌" table.add_row(str(idx), task.task, task.category, is_done) console.print(table)
Kết bài
Bạn đã hoàn thành ứng dụng CLI quản lý công việc với đầy đủ tính năng. Hãy thử thêm, xóa, và cập nhật công việc để kiểm tra.
Ý tưởng cải tiến:
- Hiển thị riêng công việc đã hoàn thành và chưa hoàn thành.
- Thêm tính năng phân tích thời gian hoàn thành công việc.
- Sắp xếp thứ tự công việc.
Hãy sáng tạo và mở rộng ứng dụng này nhé!