1

# DeGAL - A pretty simple web image gallery

     2

# Copyright (C) 2007 Tero Marttila

     3

# http://marttila.de/~terom/degal/

     4

#

     5

# This program is free software; you can redistribute it and/or modify

     6

# it under the terms of the GNU General Public License as published by

     7

# the Free Software Foundation; either version 2 of the License, or

     8

# (at your option) any later version.

     9

#

    10

# This program is distributed in the hope that it will be useful,

    11

# but WITHOUT ANY WARRANTY; without even the implied warranty of

    12

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

    13

# GNU General Public License for more details.

    14

#

    15

# You should have received a copy of the GNU General Public License

    16

# along with this program; if not, write to the

    17

# Free Software Foundation, Inc.,

    18

# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

    19

#

    20

    21

import sqlite3

    22

    23

conn = sqlite3.connect("db/degal.db")

    24

    25

def execute (expr, *args) :

    26

    c = conn.cursor()

    27

    c.execute(expr, args)

    28

    29

    return c

    30

    31

def execute_many (expr, iter) :

    32

    c = conn.cursor()

    33

    c.executemany(expr, iter)

    34

    35

    return c

    36

    37

def insert (expr, *args) :

    38

    return execute_commit(expr, *args).lastrowid

    39

    40

def insert_many (cb, expr, iter) :

    41

    """

    42

        Perform an executemany with the given iterator (which must yield (cb_val, args) tuples), calling the given callback with the args (cb_val, row_id)

    43

    """

    44

    45

    c = conn.cursor()

    46

    47

    c.executemany(expr, _lastrowid_adapter(c, iter, cb))

    48

    49

    return commit(c)

    50

    51

def _lastrowid_adapter (c, iter, cb) :

    52

    for val, args in iter :

    53

        yield args

    54

    55

        cb(val, c.lastrowid)

    56

    57

def commit (cursor) :

    58

    try :

    59

        cursor.execute("COMMIT")

    60

    except sqlite3.OperationalError :

    61

        pass    # ffs. INSERT just doesn't do anything otherwise

    62

    63

    return cursor

    64

    65

def execute_commit (expr, *args) :

    66

    return commit(execute(expr, *args))

    67

    68

def execute_commit_many (expr, iter) :

    69

    return commit(execute_many(expr, iter))

    70

    71

select = execute

    72

    73

delete = execute_commit

    74

    75

delete_many = execute_commit_many

    76

    77

cursor = conn.cursor

    78