python postgresql select user input

Question

I'm having a problem and I think that I'm missing something very basic. I'm kind of a noob if it comes to programming but I guess I learn quick. I'm trying to make a Python script that asks for user input and then performs Postgresql SELECT. Then, the outcome of SELECT should be put into another script that uses SSH but I can't wrap my head around that SELECT query. There is a basic example of my code, if you have any tips I would appreciate greatly:

print('Diagnostic tool')
print('')
print('Please insert account ID:')
input()
try:
    conn = psycopg2.connect("dbname='psql' user='user' host='localhost' 
    password='pasword'")
    cur = conn.cursor()
    cur.execute("SELECT * FROM exampletable WHERE acc_id = #userinput ")
    cur.fetchall()
except:
    print ("Could not establish connection to Database")

As shown above- how do I perform a query that uses SELECT WHERE table name is user input (acc_id)?

kRazzy R · Accepted Answer · 2018-12-10 00:57:30Z

1

you can use format function on a string.

id = input("Please enter your id")
# id should be passed to format. 
query = "SELECT * FROM exampletable WHERE acc_id = {} ".format(id)
try:
   conn = psycopg2.connect("dbname='psql' user='user' host='localhost' 
   password='pasword'")
   cur = conn.cursor()
   cur.execute(query )
   cur.fetchall()
except:
 print ("Could not establish connection to Database")

edited Dec 10, 2018 at 0:57

kRazzy R

1,6091 gold badge20 silver badges46 bronze badges

answered Aug 2, 2018 at 15:10

chandu ram naidu kodavatikanti

1011 silver badge9 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

hendooza Over a year ago

Thanks, I'll try that!

NiVeR · Accepted Answer · 2018-08-02 15:09:21Z

0

You should pass the input as a parameter to the query. If the input string is saved in userInput you can do:

cur.execute("SELECT * FROM exampletable WHERE acc_id = %s", (userInput,))

Note: for input if your purpose is to take just the input without evaluationg as python code you should use the function raw_input().

edited Aug 2, 2018 at 15:09

answered Aug 2, 2018 at 15:05

NiVeR

9,8064 gold badges33 silver badges36 bronze badges

2 Comments

hendooza Over a year ago

Great, I'll try that. Thanks for the answer!

hendooza Over a year ago

My purpose is to perform query to DB that matches columns based on acc_id provided by user input and then my python script uses that output to fill ssh@xxx (where xxx is value of one column) and then uses command "arp vlanxxx (from column) userIP(from column) to check connectivity.

Collectives™ on Stack Overflow

python postgresql select user input

2 Answers 2

1 Comment

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related