blob: 6515a14086db31db7760d06a3346a1b6728d5ef3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
import uwsgi
import psycopg2
def ugreen_wait_callback(conn, timeout=-1):
"""A wait callback useful to allow uWSGI/uGreen to work with Psycopg."""
while True:
state = conn.poll()
if state == psycopg2.extensions.POLL_OK:
break
elif state == psycopg2.extensions.POLL_READ:
uwsgi.green_wait_fdread(conn.fileno())
elif state == psycopg2.extensions.POLL_WRITE:
uwsgi.green_wait_fdwrite(conn.fileno())
else:
raise Exception("Unexpected result from poll: %r", state)
# set the wait callback
psycopg2.extensions.set_wait_callback(ugreen_wait_callback)
def application(env, start_response):
start_response('200 Ok', [('Content-type', 'text/html')])
# connect
conn = psycopg2.connect("dbname=prova user=postgres")
# get cursor
curs = conn.cursor()
yield "<table>"
# run query
curs.execute("SELECT * FROM tests")
while True:
row = curs.fetchone()
if not row:
break
yield "<tr><td>%s</td></tr>" % str(row)
yield "</table>"
conn.close()
|