postgres_service.py

import psycopg2 as pg


class Postgres():
    def __init__(self):
        self.conn_string = "host='127.0.0.1' dbname='test' user='postgres' password=''"
        self.conn = pg.connect(self.conn_string)
        self.cursor = self.conn.cursor()

    def populate(self):
        self.cursor.execute("DROP TABLE IF EXISTS things;")
        self.cursor.execute("CREATE TABLE things (name varchar(20));")
        self.cursor.execute("INSERT INTO things(name) VALUES('Dre');")
        self.cursor.execute("INSERT INTO things(name) VALUES('Smalls');")
        self.cursor.execute("INSERT INTO things(name) VALUES('West');")
        self.cursor.execute("INSERT INTO things(name) VALUES('Combs');")
        self.cursor.execute("INSERT INTO things(name) VALUES('Flame');")

    def read(self):
        self.cursor.execute("SELECT * FROM things;")
        count = self.cursor.fetchall()
        return len(count)

    def disconnect(self):
        if self.conn:
            self.conn.close()

if __name__ == '__main__':
    postgres = Postgres()
    postgres.populate()
    postgres.read()
    postgres.disconnect()