Как принять html файл в python?

Question

[Herr_Konrad]

Здравствуйте! Имеется ввод типа

python test.py test.htm

Вопрос заключается в том, как правильно принять данный html файл что бы его потом обработать?
Я реализовал этот алгоритм при помощи with:

with open("test.html", "r") as f:

    test = f.read()
    soup = BeautifulSoup(test, 'lxml')

Но этот пример работает при запуске python test.py, а нужно получить html файл именно из input.

Comments

[Сергей Горностаев]

И в чём проблема? Используйте input.

Answer 1

[galaxy]

argv[1]

Answer 2

[soremix]

именно из input

Из аргументов что ли?

import sys

args = sys.argv

if len(args) > 1:
    html_doc = args[1]
else:
    html_doc = None

print(html_doc)

Развернутый код, чтобы было понятно, что происходит. Можно сжать в строку:

html_doc = None if len(sys.argv) < 2 else sys.argv[1]

Если аргументов нет - все равно список sys.argv будет содержать в себе один элемент - имя файла, так что нужно проверять длину списка.

Для большого кол-ва аргументов понадобится argparse
https://docs.python.org/3/library/argparse.html

Comments

[Herr_Konrad]

Получается он принимает весь мой html файл для обработки?

[soremix]

Herr_Konrad, нет, он принимает название файла. Дальше уже открывайте его через open()

[Herr_Konrad]

SoreMix, спасибо большое!