1

import sqlite3

     2

     3

try :

     4

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

     5

     6

except sqlite3.OperationalError :

     7

    conn = None

     8

     9

def execute (expr, *args) :

    10

    c = conn.cursor()

    11

    c.execute(expr, args)

    12

    13

    return c

    14

    15

def execute_many (expr, iter) :

    16

    c = conn.cursor()

    17

    c.executemany(expr, iter)

    18

    19

    return c

    20

    21

def insert (expr, *args) :

    22

    return execute_commit(expr, *args).lastrowid

    23

    24

def insert_many (cb, expr, iter) :

    25

    """

    26

        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)

    27

    """

    28

    29

    c = conn.cursor()

    30

    31

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

    32

    33

    return commit(c)

    34

    35

def _lastrowid_adapter (c, iter, cb) :

    36

    for val, args in iter :

    37

        yield args

    38

    39

        cb(val, c.lastrowid)

    40

    41

def commit (cursor) :

    42

    try :

    43

        cursor.execute("COMMIT")

    44

    except sqlite3.OperationalError :

    45

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

    46

    47

    return cursor

    48

    49

def execute_commit (expr, *args) :

    50

    return commit(execute(expr, *args))

    51

    52

def execute_commit_many (expr, iter) :

    53

    return commit(execute_many(expr, iter))

    54

    55

select = execute

    56

    57

delete = execute_commit

    58

    59

delete_many = execute_commit_many

    60

    61

if conn :

    62

    cursor = conn.cursor

    63

    64