1

import sqlite3

     2

     3

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

     4

     5

def execute (expr, *args) :

     6

    c = conn.cursor()

     7

    c.execute(expr, args)

     8

     9

    return c

    10

    11

def execute_many (expr, iter) :

    12

    c = conn.cursor()

    13

    c.executemany(expr, iter)

    14

    15

    return c

    16

    17

def insert (expr, *args) :

    18

    return execute_commit(expr, *args).lastrowid

    19

    20

def insert_many (cb, expr, iter) :

    21

    """

    22

        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)

    23

    """

    24

    25

    c = conn.cursor()

    26

    27

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

    28

    29

    return commit(c)

    30

    31

def _lastrowid_adapter (c, iter, cb) :

    32

    for val, args in iter :

    33

        yield args

    34

    35

        cb(val, c.lastrowid)

    36

    37

def commit (cursor) :

    38

    try :

    39

        cursor.execute("COMMIT")

    40

    except sqlite3.OperationalError :

    41

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

    42

    43

    return cursor

    44

    45

def execute_commit (expr, *args) :

    46

    return commit(execute(expr, *args))

    47

    48

def execute_commit_many (expr, iter) :

    49

    return commit(execute_many(expr, iter))

    50

    51

select = execute

    52

    53

delete = execute_commit

    54

    55

delete_many = execute_commit_many

    56

    57

cursor = conn.cursor

    58