-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathwrite data in db with input command.py
56 lines (41 loc) · 1.34 KB
/
write data in db with input command.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
#db connection and execution
import psycopg2
from dbconnector import dbconnect
def connect():
""" Connect to the PostgreSQL database server """
conn = None
try:
# read connection parameters
params = dbconnect()
# connect to the PostgreSQL server
print('Connecting to the PostgreSQL database...')
conn = psycopg2.connect(**params)
# create a cursor
cur = conn.cursor()
# execute a statement
print('PostgreSQL database version:')
cur.execute('SELECT version()')
# display the PostgreSQL database server version
db_version = cur.fetchone()
print(db_version)
#what data to insert in table
ps_insert_date = """ INSERT INTO person(name, age, gender) VALUES(%s,%s,%s)"""
name = input("Enter your name here : ")
age = input("Enter your age here : ")
gender = input("Enter your gender here (male/female): ")
#collect data to insert in table from user
record_to_insert = (name, age, gender)
cur.execute(ps_insert_date, record_to_insert)
conn.commit()
count = cur.rowcount
print(count, "Record inserted successfully into person table")
# close the communication with the PostgreSQL
cur.close()
except (Exception, psycopg2.DatabaseError) as error:
print(error)
finally:
if conn is not None:
conn.close()
print('Database connection closed.')
if __name__ == '__main__':
connect()