Python win32com.client и оператор "with"

Question

Добрый день,

Я пишу некоторые ETL-скрипты на Python и в настоящее время использую win32com.client для открытия и обновления некоторых подключений к данным в Excel.

Мой вопрос заключается в следующем: я должен использовать оператор с, чтобы открыть / закрыть "Excel.Application", как таковой

import win32com.client
xl = win32com.client.DispatchEx("Excel.Application")

def wb_ref(file):
    xl.DisplayAlerts = False
    with xl.workbooks.open(file) as wb:
        wb.RefreshAll()
        wb.Save()

wb_ref('C:/Users/smugs/Documents/folder_a/workbooks/test.xlsx')

Когда я попытался это сделать, возникает исключение, поэтому я, очевидно, неправильно его использую.

Traceback (most recent call last):
  File "C:/Users/smugs/Documents/Python Scripts/Work/km_jobs/utils/xl_conv.py", line 32, in <module>
    wb_ref( 'C:/Users/smugs/Documents/folder_a/workbooks/test.xlsx')
  File "C:/Users/smugs/Documents/Python Scripts/Work/km_jobs/utils/xl_conv.py", line 11, in wb_ref
    with xl.workbooks.open(file) as wb:
AttributeError: __enter__

Или мне нужно явно вызвать команду закрытия

def wb_ref(file):
    xl.DisplayAlerts = False
    wb = xl.workbooks.open(file)
    wb.RefreshAll()
    wb.Save()
    wb.Close()

wb_ref('C:/Users/smugs/Documents/folder_a/workbooks/test.xlsx')

Второй пример - то, что я использовал, и это работает. Я думаю, мне просто интересно, что является более питоническим способом написания вышеуказанной функции.

(к сведению - впервые спрашиваю, давний читатель)

Answer 1

Вы получаете AttributeError: __enter__ ошибка, потому что xl.workbooks.open не является менеджером контекста, и поэтому он не поддерживает with заявление.

Если вы хотите использовать with В своем коде вы можете использовать функцию закрытия из модуля contextlib в стандартной библиотеке, например:

from contextlib import closing

def wb_ref(file):
    xl.DisplayAlerts = False
    with closing(xl.workbooks.open(file)) as wb:
        wb.RefreshAll()
        wb.Save()

contextlib.closing автоматически позвонит close на объекте, который передается ему, когда код в with Блок завершил выполнение.