Skip to content
Snippets Groups Projects
Commit e7ef0ba9 authored by Julian Volland's avatar Julian Volland
Browse files

changed name

parent 2a98edf6
No related branches found
No related tags found
No related merge requests found
#!/usr/bin/env python3
import psycopg2
from psycopg2 import sql
import sys
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-f','--file', required=True, help="Filelocation")
parser.add_argument('-t','--tablename', required=True, help="Tablename")
parser.add_argument('-n','--name', required=True, help="Databasename")
parser.add_argument('-u','--user', required=True, help="Databaseuser")
parser.add_argument('-p','--password', required=False, help="userpassword")
parser.add_argument('-s','--schema', required=True, help="Schema")
args = parser.parse_args()
filelocation = args.file
tablename = args.tablename
databasename = args.name
databaseuser = args.user
userpassword = args.password
schema = args.schema
if userpassword == None:
connectionstring = ("dbname=" + databasename + " user=" + databaseuser)
else:
connectionstring = ("dbname=" + databasename + " user=" + databaseuser + " password=" + userpassword)
conn = psycopg2.connect(connectionstring)
cur = conn.cursor()
file = open(filelocation, "rt")
lines = file.readlines()
file.close()
try:
cur.execute(sql.SQL("DROP TABLE {}").format(sql.Identifier(tablename)))
except:
conn.rollback()
finally:
cur.execute(sql.SQL("CREATE TABLE {} (id serial PRIMARY KEY, {} varchar)").format(sql.Identifier(tablename),sql.Identifier(tablename)))
for name in lines:
cur.execute(sql.SQL("INSERT INTO {}.{} ({}) VALUES (%s)").format(sql.Identifier(schema),sql.Identifier(tablename),sql.Identifier(tablename)),(name[:-1], ))
conn.commit()
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