#!/usr/bin/python
# -*- coding: utf-8 -*-

import psycopg2
import sys
import xlwt

con = None

try:
    i=0
    con = psycopg2.connect(database="testdb", user="postgres", password="pass123",  port="5432")

    cur = con.cursor()
    cur.execute("SELECT * FROM company")
    workbook = xlwt.Workbook()
    sheet = workbook.add_sheet("Sheet Name")
    while True:

        row = cur.fetchone()

        if row == None:
            break

        print row[0], row[1], row[2]
        sheet.write(i,0, row[0])
        sheet.write(i,1, row[1])
        sheet.write(i,2, row[2])
        i+=1
    workbook.save("kh.xls")
except psycopg2.DatabaseError, e:
    print 'Error %s' % e
    sys.exit(1)


finally:

    if con:
        con.close()