Скриншот окна с использованием Python

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

В настоящее время у меня есть скрипт, который делает скриншот всего экрана:

import sys
from PyQt4.QtGui import QPixmap, QApplication
from datetime import datetime

date = datetime.now()
filename = date.strftime('%Y-%m-%d_%H-%M-%S.jpg')
app = QApplication(sys.argv)
QPixmap.grabWindow(QApplication.desktop().winId()).save(filename, 'jpg')

Но я хотел бы иметь только выбранное окно. Я знаю, что проблема возникает из-за GrabWindow. Но я не знаю, как это решить.