[SOLVED] Python how to assign user inputs to variables

bindo · 04-19-2017, 08:28 AM

Hello all,

My first post here. Newbie to python and I want to improve the below code mainly for learning purpose. Let me first post it and then explain how I want to improve it.

Code:

VAR1, VAR2, VAR3= raw_input("Enter ip, port and name seperated by colan").split(":")
   ##user input should be something like, 1.1.1.1:422:name1

template ="""frontend fe:{0}:bind:{1}
        mode tcp
        bind {0}:{1} name {2}"""
  
BLOCK = template.format(VAR1,VAR2,VAR3)
  
with open("file.txt", "a") as myfile:
    myfile.write(BLOCK)

As you can see I'm appending to the "file.txt" based on user inputs. Once the user specify the ip address and the port (i.e: 10.1.2.2:443:somename) it would construct the text in variable "template" sending the correct value to the corresponding "{}".

I need to take this further and do the following. I'm really at a loss here guys.

- I want to get the user to specify the raw_input as a command line argument (i.e ./script.py FE-<name>:<ip>:<port> | BE-<name><ip>:<port>).

- I want to add another part to the template called "backend bk:<somename>:<someip>:<someport>". Script should pick up the correct user input to the right section (ie: "FE-<name>:<ip>:<port>" to the "frontend fe" section and "BE-<name><ip>:<port>" to the backend section.

- once user is done inputting the parameters I want to restart a service i.e:service abc restart. Please note that I want to wait before user adds all the entries before restarting the service.

Please help out gentleman. I really want to come up to speed.
many thanks in advance

-Bindo

astrogeek · 04-19-2017, 01:40 PM

raw_input is for processing interactive user input.

If you want to pass and process command line args then what you probably need are optparse or argparse, depending on Python version.

As far as the other sections of the code, what to do with those args, it would be more helpful if you could show us what you have done and what parts you are having trouble with. Whlie people here are happy to help, we cannot write the code for you.

bindo · 04-20-2017, 01:22 AM

Thanks astro for the advice. Well I wrote the below code and it doesn't get me anywhere.

Code:

#!/usr/bin/python
import argparse
import re
import string

p = argparse.ArgumentParser()
p.add_argument("INPUT")  
args = p.parse_args()
KKK= args.INPUT
bb=re.split(":|,", KKK)

def func_three(help):
    for i in help:
        #print help
        return help

#func_three(bb[0:3])
YY = var1, var2, var3 = func_three(bb[0:3])
print YY

The way to run this script should be "script.py <IP>:<PORT:<HOSTNAME>". i.e: script.py 192.168.1.10:80:name1 172.25.16.2:100:name3

As you can see if one argument is passed I have no problems. But when there are more arguments I cant determine how to workout the regexes and get this done via a loop.

So to recap, this is how i want the output to look like to proceed further. Please help out.

192.168.1.10
80
name1

172.25.16.2
100
name2

pan64 · 04-20-2017, 01:48 AM

I would try to use the argv array http://www.pythonforbeginners.com/sy...ython-sys-argv http://www.pythonforbeginners.com/ar...with-sys-argv:

Code:

for var in sys.argv[1:]:
    data = re.split(":|,", var)
    print data[0]
    print data[1]
    print data[2]

camp0 · 04-20-2017, 03:28 AM

I prefer the optparse style

Code:

import optparse

def option_parser():

    def get_comma_separated_args(option, opt, value, parser):
        setattr(parser.values, option.dest, value.split(','))

    p = optparse.OptionParser()

    p.add_option("-u","--url", dest="url",
        default="http://www.google.com", type="string",
        help="Sets the URL. Default value [%default]")

    p.add_option("-v", "--verbose",dest="verbose", default=False, action="store_true",
        help="Shows extra messages.")

    return p

if __name__ == '__main__':

    (options, args) = option_parser().parse_args()

bindo · 04-20-2017, 09:34 AM

Thanks all for trying to help out. I will try the suggestions out and get back to you. Its is very late in the night here in this part of the world/. Thanks once again.

-Bindo

bindo · 04-26-2017, 07:10 AM

I appologize for not having replied much sooner as I was taken ill.

Both solutions work and thanks so much for helping out Camp0 and and Pan64. You guys rock.