AttentionThis page documents an earlier version. Go to the latest (v2.1)version.
Install the psycopg2 driver
Install the python PostgreSQL driver using the following command. You can get further details for the driver here .
$ pip install psycopg2-binary
This tutorial assumes that you have:
- installed YugabyteDB and created a universe with YSQL enabled. If not, please follow these steps in the Quick Start guide .
Writing the python code
Create a file
yb-sql-helloworld.py and add the following content to it.
import psycopg2 # Create the database connection. conn = psycopg2.connect("host=127.0.0.1 port=5433 dbname=postgres user=postgres password=postgres") # Open a cursor to perform database operations cur = conn.cursor() # Create the table. cur.execute( """ CREATE TABLE employee (id int PRIMARY KEY, name varchar, age int, language varchar); """) print "Created table employee" # Insert a row. cur.execute("INSERT INTO employee (id, name, age, language) VALUES (%s, %s, %s, %s)", (1, 'John', 35, 'Python')) print "Inserted (id, name, age, language) = (1, 'John', 35, 'Python')" # Query the row. cur.execute("SELECT name, age, language FROM employee WHERE id = 1;") row = cur.fetchone() print "Query returned: %s, %s, %s" % (row, row, row) # Close the connection. cur.close() conn.close()
Running the application
To run the application, type the following:
$ python yb-sql-helloworld.py
You should see the following output.
Created table employee Inserted (id, name, age, language) = (1, 'John', 35, 'Python') Query returned: John, 35, Python