Python (10.) - Vstup a výstup

V dnešním díle se podíváme na to, jaké možnosti běžného (terminál, soubory) vstupu a výstupu nám Python nabízí.

2.3.2005 15:00 | Aleš Hakl | přečteno 26941×

Již v předchozích dílech jsme zmínili příkaz print pro vypisování dat na standardní výstup, toto samozřejmě není jediná možnost výstupu. Podobně jsme též zmínili funkci raw_input načítající řádek od uživatele, také to samozřejmě není jediný prostředek vstupu dat, ačkoli pro jednoduché načítání řetězců od uživatele je tato funkce jistě jednou z nejvhodnějších.

Nezmínili jsme funkci input, jež je stejná jako funkce raw_input, navíc ovšem řetězec zadaný uživatelem vyhodnotí, jako kdyby byl výrazem v programu a vrátí výslednou hodnotu, v praxi se tato funkce v Pytonu ovšem příliš často nepoužívá.

Python, podobně jako unix obecně, celkem nerozlišuje, z čeho čteme a kam píšeme, pro programátora mezi souborem, terminálem či třeba síťovým spojením není velký rozdíl, všechny tyto případy jsou reprezentovány objektem s přibližně stejnými metodami, nazývejme takový objekt stream (proud). Samozřejmě stream implementuje pouze metody, které pro něj mají smysl. Například čtení ze standardního výstupu (většinou) smysl nemá.

Standardní vstup a výstup

Zpočátku můžeme za zajímavé považovat streamy reprezentující standardní vstup, výstup a chybový výstup. Tyto proudy nalezneme ve standardním modulu sys jako proměnné stdin, stdout a stderr.

Samozřejmě, že tyto proudy jsou proměnné jako každé jiné a je tedy možné změnit stream používaný pro vstup či výstup a tím dosáhnout nejrůznějších žádoucích efektů.

Soubory

Streamy také můžeme použít k přístupu k souborům, slouží k tomu funkce open, která je přímo součástí interpretu jazyka Python. Tato funkce má dva argumenty, jméno souboru a režim, ve kterém bude otevřen.

Režim je řetězec, jehož první znak určuje jestli soubor chceme číst ('r'), přepsat ('w') nebo zapisovat na jeho konec ('a'). Dále můžeme uvést znak '+' pro zápis i čtení zároveň, 'r+' nám tedy umožní soubor jak číst tak zapisovat a 'w+' totéž, s tím že bude nejdříve smazán původní obsah souboru. Méně vyspělé operační systémy též mohou vyžadovat znak 'b' indikující binární soubor, ten se zapisuje na druou pozici, za jeden ze znaků 'rwa'.

Pokud režim neuvedeme použije se výchozí hodnota 'r'.

Čtení ze streamu

Pokud tedy chceme přečíst soubor po jednotlivých řádcích a něco s nimi provést (což je dle mého názoru asi nejčastější operace, kterou chceme v unixu se souborem provádět), uděláme to například takto:

f = open('/etc/inittab','r')
i = 0 

for l in f.xreadlines():
    i = i + 1
    print i,l.rstrip()   # ořízneme bíle znaky na konci řádku

                         # (tj. minimálně znak nového řádku)
                         # a řádek očíslujeme.

Zápis do proudu

Přesnou kopii souboru můžeme s využtím výše uvedených metod vytvořit nějak takto:

src = open('10.html','r')  # zdrojový soubor

dst = open('foo','w')      # cílový soubor

while True:
    buf = src.read(4096)   # čteme 4KiB najednou
    dst.write(buf)
    if len(buf) < 4096:
        break

Posun ve streamu

Aktuální pozici ve streamu můžeme, pokud to daný stream podporuje, měnit. Je zřejmé, že je nesmysl pokoušet se měnit pozici ve streamu reprezentujícím nějaké zařízení nebo síťové spojení, cestovat časem Python opravdu neumí ^_^.

Ostatní metody

Atributy

Streamy také disponují několika atributy. Atribut closed má hodnotu True, pokud je stream uzavřen, name obsahuje jméno souboru nebo případně nějaký popis streamu, mode obsahuje řetězec určující režim (druhý parametr funkce open).

Stream ovšem nemusí vůbec být soubor (nebo něco, co se z pohledu rozhraní operačního systému chová jako soubor), ale může to být téměř libovolná třída. Ve standardní knihovně Pythonu tudíž nalezneme nejrůznější třídy různým způsobem obalující jiné streamy (například provádějící komprimaci, převod kódování...) či jenom prostě implementující rozhraní streamu, protože autoři předpokládali, že je vhodné, aby daná třída vypadala jako stream.

Dnešní díl tedy zakončíme souhrnnou ukázkou. Napíšeme program, který nám umožní na základě jednoho textového souboru zkopírovat konkrétní byte z jednoho souboru do druhého (k čemu to může být dobré, nechám na představivosti čtenáře).

Vstupem bude textový soubor se seznamem offsetů, vstupní binární soubor a výstupem bude jeden soubor obsahující dané byty.

#!/usr/bin/env python
# -*- coding: utf-8; -*-

import sys

src_name = raw_input("Jméno vstupního souboru?")
txt_name = raw_input("Jméno souboru se seznamem offsetů?")
dst_name = raw_input("Jméno výstupního souboru?")

txt = open(txt_name,'r')
src = open(src_name,'rb') 
dst = open(dst_name,'wb')


for off_s in txt.xreadlines():
    off = long(off_s)  # Převedeme na číslo

    src.seek(off)      # nastavíme pozici
    byte = src.read(1) # přečteme 1 byte


    if byte=='':       # případně upozorníme uživatele na možnou chybu
        sys.stderr.write("Offset za koncem vstupního souboru")

    dst.write(byte)


# uklidíme po sobě
src.close() 
dst.close() 
txt.close()

Doufám, že vás dnešní poněkud vykonstruovaná ukázka neodradila a sejdeme se u dalšího dílu, tentokrát se podíváme na řetězce poněkud podrobněji.

Online verze článku: http://www.linuxsoft.cz/article.php?id_article=732