chapter10

Bazy danych

Przykład 9.4: sqlite

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

import os
import sqlite3

def connect_db():
    # przylaczenie bazy danych
    db = sqlite3.connect('test.db')
    return db

def ini_db():
    try:
        db = connect_db()
        cur = db.cursor()
        cur.executescript('''\
CREATE TABLE "user" (
  "id" INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL,
  "username" TEXT,
  "password" TEXT
);
        ''')
        db.commit()
    except sqlite3.Error as e:
        print("DB error:", e.args[0])
    except Error as e:
        print("rror:", e.args[0])

def new_user(db, name, password):
    try:
        cursor = db.cursor()
        cursor.executescript("insert into user(username, password) values ('%s','%s')" % (name, password))
        # nie powoduje błędu z O'Connor:
        #   cursor.execute("insert into user(username, password) values (?,?)" , (name, password))
        db.commit()
    except sqlite3.Error as e:
        print("DB error:", e.args[0])
        print('nieudane  wstawienie %s ', name)

ini_db()
db=connect_db()
new_user(db, 'Tomek','haslotomka')
new_user(db, 'Tomek','haslotomka')
new_user(db, "O''Connor",'haslo2')
new_user(db, 'Romek','hasloromka')

cursor = db.cursor()
for (id, username, password) in cursor.execute('select id, username, password from user'):
    print('%s %s:%s' % (id, username, password))

Objaśnienie:

Istnieje wiele różnych baz danych. Powyższy przykład dotyczy najprostszej z nich: SQLite. Dane są pamiętane w przypadku tej bazy w jednym pliku. W przypadku bardziej zaawansowanych baz danych komunikacja odbywa się za pośrednictwem sieci. Dla SQLite potrzebny jest tylko odpowiedni moduł – w przypadku Pythona jest to sqlite3.

Połączenie z bazą danych (funkcja connect) ustala zazwyczaj jedynie parametry komunikacji. W przypadku SQLite polecenie sqlite3.connect('test.db') tworzy bazę – o ile wcześniej ona nie istniała.

Komunikacja z bazą odbywa się z użyciem języka SQL. Polecenia nie są przesyłane wprost poprzez obiekt połączenia (db w powyższym przykładzie), ale przez obiekt kursora (ang. cursor). Umożliwia to stworzenie wielu kursorów dla jednej bazy – na przykład dwa kursory do przeglądania równolegle dwóch tabel.

Do wysłania polecenia SQL można użyć funkcji executescript lub execute. W tym drugim przypadku jako wynik funkcji dostajemy rezultat zapytania w postacie generowanych wierszy.

Polecenie SQL ‘CREATE TABLE "user" … ‘ tworzy tabelę o nazwie „user”. Polecenie ‘insert into ….’ wstawia wiersze do tej tabeli. Natomiast ‘select ...’ odczytuje wstawione dane.

Funkcja commit() w obiekcie połączenia z bazą powoduje zakończenie zapytania i zapisanie danych na dysku.