Аргументы запуска Python-скрипта при работе с файлами

Расширенные примеры работы с аргументами командной строки

Пример 1: argparse с проверкой существования файла

import argparse
import os

def file_exists(path):
    if not os.path.isfile(path):
        raise argparse.ArgumentTypeError(f'Файл {path} не существует')
    return path

parser = argparse.ArgumentParser(description='Подсчёт строк в файле')
parser.add_argument('file', type=file_exists, help='Путь к файлу')
parser.add_argument('--lines-start', type=int, default=1, help='Номер первой строки')
args = parser.parse_args()

with open(args.file, 'r') as f:
    lines = f.readlines()
print(f'Всего строк: {len(lines)}, начиная с {args.lines_start}')

$ python script.py nonexistent.txt
usage: script.py [-h] [--lines-start LINES_START] file
script.py: error: argument file: Файл nonexistent.txt не существует

$ python script.py existing.txt
Всего строк: 42, начиная с 1

Пример 2: Обработка нескольких файлов с помощью sys.argv и glob

import sys
import glob

if len(sys.argv) < 2:
    print('Укажите хотя бы один файл или шаблон (например, *.txt)')
    sys.exit(1)

files = []
for pattern in sys.argv[1:]:
    matched = glob.glob(pattern)
    if not matched:
        print(f'Предупреждение: нет файлов, соответствующих {pattern}')
    files.extend(matched)

print('Обрабатываются файлы:')
for f in files:
    with open(f, 'r') as fh:
        print(f'  {f}: {len(fh.readlines())} строк')

$ python script.py data/*.csv report.log
Обрабатываются файлы:
  data/a.csv: 100 строк
  data/b.csv: 200 строк
  report.log: 50 строк

Пример 3: argparse с подкомандами (subparsers) для разных операций с файлами

import argparse

def read_file(args):
    with open(args.file, 'r') as f:
        print(f.read())

def write_file(args):
    with open(args.file, 'w') as f:
        f.write(args.content or '')

parser = argparse.ArgumentParser(description='Файловый менеджер')
subparsers = parser.add_subparsers(dest='command')

read_parser = subparsers.add_parser('read', help='Чтение файла')
read_parser.add_argument('file')
read_parser.set_defaults(func=read_file)

write_parser = subparsers.add_parser('write', help='Запись в файл')
write_parser.add_argument('file')
write_parser.add_argument('--content', help='Текст для записи')
write_parser.set_defaults(func=write_file)

args = parser.parse_args()
if hasattr(args, 'func'):
    args.func(args)
else:
    parser.print_help()

$ python file_manager.py read example.txt
(содержимое файла)

$ python file_manager.py write new.txt --content 'Hello'
(создаёт файл с текстом)

Пример 4: Использование модуля fileinput для обработки файлов, переданных как аргументы

import fileinput
import sys

# Если файлы не указаны, читает stdin
for line in fileinput.input():
    if fileinput.isfirstline():
        print(f'--- {fileinput.filename()} ---')
    sys.stdout.write(line)

$ python script.py file1.txt file2.txt
--- file1.txt ---
(содержимое file1.txt)
--- file2.txt ---
(содержимое file2.txt)

Пример 5: getopt с длинными опциями и проверкой целочисленных значений

import sys
import getopt

def main():
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'n:c:', ['number=', 'count='])
    except getopt.GetoptError as err:
        print(str(err))
        sys.exit(1)

    number = None
    count = None
    for opt, val in opts:
        if opt in ('-n', '--number'):
            try:
                number = int(val)
            except ValueError:
                print('Опция --number требует целое число')
                sys.exit(1)
        elif opt in ('-c', '--count'):
            count = int(val)

    if args:
        files = args
    else:
        print('Укажите хотя бы один файл')
        sys.exit(1)

    print(f'Число: {number}, количество: {count}, файлы: {files}')

if __name__ == '__main__':
    main()

$ python script.py --number abc file.txt
Опция --number требует целое число