postgres

#python 3.9.5
#pip install psycopg2
#выставлены значения по умолчанию, в аргументах функции create_connection
#можно вбивать свои в таком же порядке в аргументах при запуске из консоли

import psycopg2
import sys

def create_connection(database="lab6", user="postgres", password="5786", host="127.0.0.1", port="5432"):
  # print([database, user, password, host, port])
  try:
    connection = psycopg2.connect(
      database=database,
      user=user,
      password=password,
      host=host,
      port=port
    )
    print("Database opened successfully\n")
    return connection
  except psycopg2.OperationalError as e:
    print('Unable to connect!\n{0}'.format(e))
    if not str(e):
      print('Check database, user and password')
    # sys.exit(-1)

with open('C:\\Users\\vladt\\Desktop\\users.txt') as file:
  users_line = file.readlines()
  if not users_line[-1].endswith('\n'):
    users_line[-1] = users_line[-1] + 'n'
  print(users_line)
users = []
for i in range(len(users_line)):
  users += [users_line[i][:-1].split(',')]
# users += [users_line[i].split(',')]
print(users)


args = sys.argv[1:]
s = 0
f = 0
for el in users:
  print(el)
  con = create_connection(user=el[0], password=el[1])
  if con:
    s += 1
  else:
    f += 1
print('s=', s, '\nf=', f)
# cur = con.cursor()
# try:
#   cur.execute('''CREAT TABLE a
#        (ADMISSION INT PRIMARY KEY NOT NULL,
#        NAME TEXT NOT NULL,
#        AGE INT NOT NULL,
#        COURSE CHAR(50),
#        DEPARTMENT CHAR(50));''')
#   print("Table created successfully")
# except Exception as e:
#   print(e)
# con.commit()
# con.close()