Dotaz: python - snimek obrazovky (screenshot)

[petr]

Ahoj! Snazim se naucit python a rad bych udelal programek na ukladani snimku obrazovky (screenshoty). Mam tedy neco ve stylu:
import sys
import Image
import ImageGrab
.
.
.
img=ImageGrab.grab()
.
.
.

Python Imaging Library (PIL) jsem nainstaloval. Interpreter mi ale hlasi, ze:

Traceback (most recent call last):
  File "./programek.py", line 17, in ?
    import ImageGrab
  File "/usr/lib/python2.4/site-packages/PIL/ImageGrab.py", line 34, in ?
    import _grabscreen
ImportError: No module named _grabscreen

Nic jsem k tomu na netu nenasel, akorat tady maji ten samej problem co ja. Nevite nekdo co s tim?

Odpovědi

[petr]

Aha, tak tady je problem: The current version works on Windows only.

[petr]

Takze dalsi otazka je, jak udelat screenshoty v pythonu?

[Michal Vyskočil]

Pokud nevadí, že se jedná o vyloženě skriptovací řešení, tak z Pythonu spusť program xwd.

[petr]

No to je prave problem, protoze ja bych to chtel nejdriv ukazat jako nahled pomoci pygtk, ale tyto programky jako scrot a pod., to rovnou ukladaji

[trancelius]

něco jako

roura = os.popen("xwd -root")
data = roura.read()

a budeš mít ten obrázek bez psaní na disk

[Michal Karas]

Ahoj, nešel by použít PythonMagick?

[petr]

Diki vsem, pipe funguje, akorat ze xwd uklada do, pro mne, nestandardniho .xwd, ktery bych pak musel prekonvertovat do .png, jenze na to potrebuji zmineny pythonmagick, coz je dalsi zavislost (a tech chci mit minimum). Tak nakonec jsem to vyresil salamounsky, pouzivam muj oblibeny scrot, ukladam do /tmp, z toho delam nahled a v pak pripadne to ukladam (copy & rename) do zvoleneho adresare a souboru. Ehm, trochu divne, a proto bych se chtel zeptat, zda se to takhle dela, zda je to standardni.

jinak tady je kod s pipe, ktery jsem vygeneroval:

imagePipe = os.popen("xwd -root")
imageData = imagePipe.read()
imageFile = open('vystup.xwd','w')
imageFile.write(imageData)
imageFile.close()

I v tomhle bych potreboval poradit, zda jsem to alespon napsal ciste, nekomplikovane. Diki vsem.

[Jan Martinek]

Jak jsem vyrozuměl, čisté řešení v podobě knihovny + modulu do Pythonu v Linuxu neexistuje. Zato hotových programů je několik. Když už se tedy smířím se spouštěním externích programů, zmínil bych ještě program import. Má spoustu parametrů, umí psát do png souboru i do standardního výstupu. Zkus třeba

import -window root screenshot.png

Netvrdím, že je lepší či horší - je to prostě taky možnost.

[Jan Martinek]

... pripadne to ukladam (copy & rename) do zvoleneho adresare a souboru. Ehm, trochu divne, a proto bych se chtel zeptat, zda se to takhle dela, zda je to standardni.

Já myslím, že jo. Tady moc na vybranou asi není. Snad jen, že místo copy & rename dej pouze rename, ať se ti v /tmp nehromadí soubory. Jestli budeš číst png z roury, tak se bez dočasného souboru objedeš úplně.

jinak tady je kod s pipe, ktery jsem vygeneroval:

imagePipe = os.popen("xwd -root")
imageData = imagePipe.read()
imageFile = open('vystup.xwd','w')
imageFile.write(imageData)
imageFile.close()

I v tomhle bych potreboval poradit, zda jsem to alespon napsal ciste, nekomplikovane. Diki vsem.

Hehe, nedokážu posoudit, myslím, že je to v pohodě
Hlavně že to funguje. Snad jen jako alternativu můžeš vyzkoušet tohle:

#!/usr/bin/env python
import subprocess

cmd = ['xwd', '-root']
process = subprocess.Popen(cmd, stdout = subprocess.PIPE)
file('vystup.xwd', 'w').write(process.stdout.read())

Vřele doporučuji si zvyknout na modul subprocess. Možná je to trochu komplikovanější, ale zato se s externími programy zachází jednotně, lépe se předávají argumenty a můžeš úplně zapomenout na

os.system
os.spawn*
os.popen*
popen2.*
commands.*

Jo a snad možná místo "open" je lepší psát "file".

[Jan Martinek]

Nerad vyhrabávám staré vlákno, ale náhodou jsem natrefil na řešení. Screenshot v čistém pythonu (tedy bez pouštění externích programů) lze získat takto:

#!/usr/bin/env python
import gtk, gc

width = gtk.gdk.screen_width()
height = gtk.gdk.screen_height()
screenshot = gtk.gdk.Pixbuf.get_from_drawable(
             gtk.gdk.Pixbuf(gtk.gdk.COLORSPACE_RGB, True, 8, width, height),
             gtk.gdk.get_default_root_window(),
             gtk.gdk.colormap_get_system(),
             0, 0, 0, 0, width, height)

screenshot.save('screenshot.png', 'png')
del screenshot
gc.collect()

Více informací viz PyGTK FAQ.