Создание полноценного инструмента командной строки
На этом этапе мы объединим все, что узнали, чтобы создать полноценный инструмент командной строки. Давайте создадим простой менеджер задач, который позволит пользователям добавлять, просматривать и удалять задачи, используя аргументы командной строки.
Создание менеджера задач
- Создайте новый файл с именем
task_manager.py в каталоге /home/labex/project
- Добавьте следующий код:
import argparse
import json
import os
## File to store tasks
TASKS_FILE = "/home/labex/project/tasks.json"
def load_tasks():
"""Load tasks from the JSON file."""
if os.path.exists(TASKS_FILE):
with open(TASKS_FILE, 'r') as f:
try:
return json.load(f)
except json.JSONDecodeError:
return []
return []
def save_tasks(tasks):
"""Save tasks to the JSON file."""
with open(TASKS_FILE, 'w') as f:
json.dump(tasks, f, indent=2)
def main():
## Create the main parser
parser = argparse.ArgumentParser(description='Task Manager CLI')
subparsers = parser.add_subparsers(dest='command', help='Command to run')
## Add command
add_parser = subparsers.add_parser('add', help='Add a new task')
add_parser.add_argument('title', help='Task title')
add_parser.add_argument('-p', '--priority',
choices=['low', 'medium', 'high'],
default='medium',
help='Task priority (default: medium)')
add_parser.add_argument('-d', '--due',
help='Due date (format: YYYY-MM-DD)')
## List command
list_parser = subparsers.add_parser('list', help='List all tasks')
list_parser.add_argument('-p', '--priority',
choices=['low', 'medium', 'high'],
help='Filter tasks by priority')
list_parser.add_argument('-s', '--sort',
choices=['priority', 'title'],
default='priority',
help='Sort tasks by criteria (default: priority)')
## Delete command
delete_parser = subparsers.add_parser('delete', help='Delete a task')
delete_parser.add_argument('task_id', type=int, help='Task ID to delete')
## Parse arguments
args = parser.parse_args()
## Load existing tasks
tasks = load_tasks()
## Handle commands
if args.command == 'add':
## Add a new task
new_task = {
'id': len(tasks) + 1,
'title': args.title,
'priority': args.priority,
'due': args.due
}
tasks.append(new_task)
save_tasks(tasks)
print(f"Task added: {new_task['title']} (ID: {new_task['id']})")
elif args.command == 'list':
## List tasks
if not tasks:
print("No tasks found.")
return
## Filter by priority if specified
if args.priority:
filtered_tasks = [t for t in tasks if t['priority'] == args.priority]
else:
filtered_tasks = tasks
## Sort tasks
if args.sort == 'priority':
## Custom priority sorting
priority_order = {'high': 0, 'medium': 1, 'low': 2}
sorted_tasks = sorted(filtered_tasks, key=lambda x: priority_order[x['priority']])
else:
## Sort by title
sorted_tasks = sorted(filtered_tasks, key=lambda x: x['title'])
## Display tasks
print("ID | Title | Priority | Due Date")
print("-" * 50)
for task in sorted_tasks:
due_date = task['due'] if task['due'] else 'N/A'
print(f"{task['id']:2} | {task['title'][:20]:<20} | {task['priority']:<8} | {due_date}")
elif args.command == 'delete':
## Delete a task
task_id = args.task_id
task_found = False
for i, task in enumerate(tasks):
if task['id'] == task_id:
del tasks[i]
task_found = True
break
if task_found:
save_tasks(tasks)
print(f"Task {task_id} deleted.")
else:
print(f"Task {task_id} not found.")
else:
## No command specified
parser.print_help()
if __name__ == "__main__":
main()
Понимание менеджера задач
Этот скрипт демонстрирует несколько расширенных функций argparse:
- Subparsers (Подпарсеры) - Создание наборов аргументов, специфичных для команд
- Command-specific arguments (Аргументы, специфичные для команд) - Различные аргументы для команд add, list и delete
- Nested validation (Вложенная проверка) - Выбор приоритета ограничен определенными значениями
- Default values (Значения по умолчанию) - Предоставление разумных значений по умолчанию для необязательных аргументов
Тестирование менеджера задач
Давайте запустим скрипт с разными командами, чтобы увидеть, как он работает:
- Добавление задач:
python3 task_manager.py add "Complete Python tutorial" -p high -d 2023-12-31
Вывод:
Task added: Complete Python tutorial (ID: 1)
Добавьте еще несколько задач:
python3 task_manager.py add "Read documentation" -p medium
python3 task_manager.py add "Take a break" -p low -d 2023-12-25
- Просмотр задач:
python3 task_manager.py list
Вывод:
ID | Title | Priority | Due Date
--------------------------------------------------
1 | Complete Python tutor | high | 2023-12-31
2 | Read documentation | medium | N/A
3 | Take a break | low | 2023-12-25
- Просмотр задач с фильтрацией и сортировкой:
python3 task_manager.py list -s title
Вывод:
ID | Title | Priority | Due Date
--------------------------------------------------
1 | Complete Python tutor | high | 2023-12-31
2 | Read documentation | medium | N/A
3 | Take a break | low | 2023-12-25
- Фильтрация по приоритету:
python3 task_manager.py list -p high
Вывод:
ID | Title | Priority | Due Date
--------------------------------------------------
1 | Complete Python tutor | high | 2023-12-31
- Удаление задачи:
python3 task_manager.py delete 2
Вывод:
Task 2 deleted.
Убедитесь, что задача была удалена:
python3 task_manager.py list
Вывод:
ID | Title | Priority | Due Date
--------------------------------------------------
1 | Complete Python tutor | high | 2023-12-31
3 | Take a break | low | 2023-12-25
Получение справки по подкомандам
Argparse автоматически генерирует справку для каждой подкоманды:
python3 task_manager.py add --help
Вывод:
usage: task_manager.py add [-h] [-p {low,medium,high}] [-d DUE] title
positional arguments:
title Task title
options:
-h, --help show this help message and exit
-p {low,medium,high}, --priority {low,medium,high}
Task priority (default: medium)
-d DUE, --due DUE Due date (format: YYYY-MM-DD)
Ключевые моменты о подпарсерах
- Subparsers (Подпарсеры) создают наборы аргументов, специфичные для команд
- Каждый подпарсер может иметь свои собственные аргументы
- Параметр
dest указывает, где будет храниться имя команды
- Сообщения справки автоматически генерируются для каждой подкоманды
- Вы можете смешивать позиционные и необязательные аргументы в подпарсерах
Теперь у вас есть полностью функциональный инструмент командной строки, который демонстрирует мощь и гибкость argparse для обработки сложных сценариев аргументов.
