psycopg2_postgres_table_create_insert_query_quickstart.py

1import psycopg2
2
3# Connect to an existing database
4conn = psycopg2.connect("dbname=test user=postgres")
5
6# Open a cursor to perform database operations
7cur = conn.cursor()
8
9# Execute a command: this creates a new table
10cur.execute("CREATE TABLE test (id serial PRIMARY KEY, num integer, data varchar);")
11
12# Pass data to fill a query placeholders and let Psycopg perform
13# the correct conversion (no more SQL injections!)
14cur.execute("INSERT INTO test (num, data) VALUES (%s, %s)",
15            (100, "abc'def"))
16
17# Query the database and obtain data as Python objects
18cur.execute("SELECT * FROM test;")
19print(cur.fetchone())
20# Expected output: (1, 100, "abc'def")
21
22# Make the changes to the database persistent
23conn.commit()
24
25# Close communication with the database
26cur.close()
27conn.close()