Modul pickle

Modul pickle umožňuje ukládat do souboru objekty jako slovník a seznam, ale i samotnou třídu a později objekt znovu načíst.

Modul pickle podporuje ukládání objektů do souboru a jejich následné načtení jako stejný datový typ. Velmi se to hodí například při ukládání nastavení vašich programů: Uložíte slovník, načtete slovník. Totéž platí i pro čísla, řetězce, seznamy, třídy a ostatní datové typy. My budeme používat dvě základní funkce, a to:

• pickle.dump(objekt, soubor) – uloží objekt do souboru
• pickle.load(soubor) – načte objekt ze souboru

Ale konec teorie, zbytek si ukážeme na příkladech.

Uložení slovníku

import pickle
muj_slovnik={1:"jedna"}
soubor=file('zkouska.txt','w')
pickle.dump(muj_slovnik,soubor)
soubor.close()

Abychom do souboru mohli nějaký objekt zapsat, musíme ho otevřít v režimu 'w'. Načtení probíhá následujícím způsobem:

import pickle
soubor=file('zkouska.txt','r')
muj_slovnik=pickle.load(soubor)
soubor.close()

Jak vidíte, metoda pickle.load nepřijímá žádný parametr, aby věděla, jaký objekt načíst. Je to řešené tak, že každým zavoláním metody pickle.load() dostaneme následující prvek. Takže když potřebujeme získat hodnotu prvku, který jsme uložili jako druhý, musíme zavolat metodu pickle.load() dvakrát bez ohledu na to, že první objekt ani nevyužijeme.

Uložený slovník v souboru vypadá takto:

(dp0
I1
S'jedna'
p1
s.

Načtení a uložení třídy

Stejným způsobem, jak jsme pracovali se slovníkem, můžeme pracovat i se třídou:

# -*- coding: cp1250 -*-
import pickle

class Trida:
    def __init__(self,cislo):
        self.cislo=cislo
    def vypis(self):
        return self.cislo

moje_trida=Trida(123)
#uložení třídy:
soubor=file('zkouska.txt','w')
pickle.dump(moje_trida,soubor)
soubor.close()
del moje_trida

#načtení třídy
soubor=file('zkouska.txt','r')
moje_trida=pickle.load(soubor)
soubor.close()

print moje_trida.vypis()

Nejprve jsme si vytvořili třídu Trida s metodu vypis. Tato metoda vrátí hodnotu self.cislo. Pak jsme takto vytvořenou třídu uložili do souboru a vymazali. Následně jsme soubor otevřeli a znovu ji načetli. Jako důkaz, že se jedná o stejnou třídu, stačí zavolat metodu vypis a ta vrátí vámi zadanou hodnotu (v našem případě 123).

Ale pozor, modul pickle neumožňuje ukládat například udělátka Tkinteru nebo třídy obsahující tato udělátka, neboť dojde k vygenerování výjimky podobné této:

pickle.PicklingError: Can't pickle 'tkapp' object: <tkapp object at 0x00AEFC98>

Pro zajímavost, anglické slovo pickle znamená nakládat (např. okurky).