Skip to content
Snippets Groups Projects
Commit 9097330d authored by Auke Klazema's avatar Auke Klazema
Browse files

SW-598: Check on argument before asking password

parent ed029174
No related branches found
No related tags found
No related merge requests found
......@@ -71,6 +71,13 @@ if __name__ == '__main__':
dbPort = options.dbPort
dbUser = options.dbUser
# print sys.argv
if len(args) != 1:
parser.print_help()
sys.exit(1)
filename = str(args[0])
dbPassword = getpass.getpass()
host = "{}:{}".format(dbHost, dbPort)
......@@ -81,13 +88,6 @@ if __name__ == '__main__':
# calling stored procedures only works from the pg module for some reason.
db2 = pg.connect(user=dbUser, host=dbHost, dbname=dbName, port=dbPort, passwd=dbPassword)
# print sys.argv
if len(args) != 1:
parser.print_help()
sys.exit(1)
filename = str(args[0])
lines = get_rotation_lines(filename)
for line in lines:
(stationname, rotation0, rotation1) = get_rotation(line)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment