← Back to branch summary

~psycopg/psycopg/2.0.x

« back to all changes in this revision

Viewing changes to examples/fetch.py

  • Committer: Federico Di Gregorio
  • Date: 2009-04-20 22:38:09 UTC
  • Revision ID: fog@erin-20090420223809-y59bb142mo3vadnw
Better detection of PostgreSQL version

Show diffs side-by-side

added added

removed removed

Lines of Context:
examples/fetch.py
 
1
# fetch.py -- example about declaring cursors
 
2
#
 
3
# Copyright (C) 2001-2005 Federico Di Gregorio  <fog@debian.org>
 
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 the
 
7
# Free Software Foundation; either version 2, or (at your option) any later
 
8
# version.
 
9
#
 
10
# This program is distributed in the hope that it will be useful, but
 
11
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTIBILITY
 
12
# or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 
13
# for more details.
 
14
#
 
15
 
 
16
## put in DSN your DSN string
 
17
 
 
18
DSN = 'dbname=test'
 
19
 
 
20
## don't modify anything below tis line (except for experimenting)
 
21
 
 
22
import sys
 
23
import psycopg2
 
24
 
 
25
if len(sys.argv) > 1:
 
26
    DSN = sys.argv[1]
 
27
 
 
28
print "Opening connection using dns:", DSN
 
29
conn = psycopg2.connect(DSN)
 
30
print "Encoding for this connection is", conn.encoding
 
31
 
 
32
curs = conn.cursor()
 
33
try:
 
34
    curs.execute("CREATE TABLE test_fetch (val int4)")
 
35
except:
 
36
    conn.rollback()
 
37
    curs.execute("DROP TABLE test_fetch")
 
38
    curs.execute("CREATE TABLE test_fetch (val int4)")
 
39
conn.commit()
 
40
 
 
41
# we use this function to format the output
 
42
 
 
43
def flatten(l):
 
44
    """Flattens list of tuples l."""
 
45
    return map(lambda x: x[0], l)
 
46
 
 
47
# insert 20 rows in the table
 
48
 
 
49
for i in range(20):
 
50
    curs.execute("INSERT INTO test_fetch VALUES(%s)", (i,))
 
51
conn.commit()
 
52
 
 
53
# does some nice tricks with the transaction and postgres cursors
 
54
# (remember to always commit or rollback before a DECLARE)
 
55
#
 
56
# we don't need to DECLARE ourselves, psycopg now support named
 
57
# cursors (but we leave the code here, comments, as an example of
 
58
# what psycopg is doing under the hood)
 
59
#
 
60
#curs.execute("DECLARE crs CURSOR FOR SELECT * FROM test_fetch")
 
61
#curs.execute("FETCH 10 FROM crs")
 
62
#print "First 10 rows:", flatten(curs.fetchall())
 
63
#curs.execute("MOVE -5 FROM crs")
 
64
#print "Moved back cursor by 5 rows (to row 5.)"
 
65
#curs.execute("FETCH 10 FROM crs")
 
66
#print "Another 10 rows:", flatten(curs.fetchall())
 
67
#curs.execute("FETCH 10 FROM crs")
 
68
#print "The remaining rows:", flatten(curs.fetchall())
 
69
 
 
70
ncurs = conn.cursor("crs")
 
71
ncurs.execute("SELECT * FROM test_fetch")
 
72
print "First 10 rows:", flatten(ncurs.fetchmany(10))
 
73
ncurs.scroll(-5)
 
74
print "Moved back cursor by 5 rows (to row 5.)"
 
75
print "Another 10 rows:", flatten(ncurs.fetchmany(10))
 
76
print "Another one:", list(ncurs.fetchone())
 
77
print "The remaining rows:", flatten(ncurs.fetchall())
 
78
conn.rollback()
 
79
 
 
80
curs.execute("DROP TABLE test_fetch")
 
81
conn.commit()