Scribd
Upload
0 views5 pages

Python Postgresql Menu App

The document outlines a Python application that interacts with a PostgreSQL database to manage a 'homemade_product' table. It includes functions to create, insert, read, and drop the table, along with a main menu for user interaction. The application allows users to perform database operations through a simple command-line interface.

Uploaded by

patilharshada974
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
0 views5 pages

Python Postgresql Menu App

The document outlines a Python application that interacts with a PostgreSQL database to manage a 'homemade_product' table. It includes functions to create, insert, read, and drop the table, along with a main menu for user interaction. The application allows users to perform database operations through a simple command-line interface.

Uploaded by

patilharshada974
Copyright
© © All Rights Reserved
We take content rights seriously. If you suspect this is your content, claim it here.
Available Formats
Download as DOCX, PDF, TXT or read online on Scribd
You are on page 1/ 5

Python + PostgreSQL Menu App

# Import required libraries

import psycopg2 # For PostgreSQL database connection

import sys # For exiting the program cleanly

# Function to establish a connection with the PostgreSQL database

def connect_db():

connection = psycopg2.connect(

user="postgres", # Username for PostgreSQL

password="******", # Password for PostgreSQL

host='localhost', # Host address (127.0.0.1 or localhost)

port="5432", # Default PostgreSQL port

database='postgres' # Name of the database to connect to

return connection

# Function to create a new table using the provided SQL query

def create_table_into_db(query):

connection = connect_db()

cursor = connection.cursor()

cursor.execute(query) # Executes the create table query

print("✅ Table created successfully.")

connection.commit() # Save the changes to the database

cursor.close()

connection.close()
# Function to insert data into the table using the provided SQL
query

def insert_data_into_db(query):

connection = connect_db()

cursor = connection.cursor()

cursor.execute(query) # Executes the insert query

print("✅ Data inserted successfully.")

connection.commit()

cursor.close()

connection.close()

# Function to read and display all records from the homemade_product


table

def read_data_from_db():

connection = connect_db()

cursor = connection.cursor()

cursor.execute("SELECT * FROM homemade_product") # Fetch all


data

results = cursor.fetchall() # Get all rows

print("\n📋 Table Data:")

for row in results:

print(row) # Display each row

connection.commit()

cursor.close()

connection.close()

# Function to drop (delete) the table if it exists

def drop_table_from_db():
connection = connect_db()

cursor = connection.cursor()

cursor.execute("DROP TABLE IF EXISTS homemade_product") # Drop


table

print("❌ Table dropped successfully.")

connection.commit()

cursor.close()

connection.close()

# ---------------- Main menu loop ----------------

while True:

# Display options to the user

print("\n" + "-" * 75)

print("1. Create table in DB")

print("2. Insert data into table")

print("3. Read data from table")

print("4. Drop table from DB")

print("5. Exit")

# Ask user for input

choice = input("Enter your choice (1-5): ")

# Perform actions based on user choice

if choice == '1':

# SQL query to create table

query_create = '''

CREATE TABLE homemade_product


(

p_id INT,

p_name VARCHAR(20),

p_price FLOAT

'''

create_table_into_db(query_create)

elif choice == '2':

# SQL query to insert multiple rows of data

query_insert = '''

INSERT INTO homemade_product

VALUES (1875678, 'Termind', 86.34),

(1125447, 'Garam Masal', 100.36),

(8457123, 'Chilli Powder', 200.65),

(1475682, 'Tea Powder', 180.57)

'''

insert_data_into_db(query_insert)

elif choice == '3':

# Read and display all data from the table

read_data_from_db()

elif choice == '4':

# Drop the table if it exists

drop_table_from_db()
elif choice == '5':

# Exit the program

print("👋 Program exited.")

sys.exit(0)

else:

# Handle invalid menu choices

print("⚠️Please enter a valid choice (1-5).")

You might also like