use postgres
This commit is contained in:
parent
216e1c0684
commit
3541226964
20
asadduser.py
20
asadduser.py
@ -1,6 +1,6 @@
|
||||
#!/usr/bin/python
|
||||
|
||||
import mariadb
|
||||
import psycopg2
|
||||
from pbkdf2 import crypt
|
||||
import argparse
|
||||
import os
|
||||
@ -23,30 +23,26 @@ password = args.password
|
||||
application = args.application
|
||||
|
||||
|
||||
DB_USER = os.environ["DB_USER"]
|
||||
DB_PASS = os.environ["DB_PASS"]
|
||||
DB_HOST = os.environ["DB_HOST"]
|
||||
DB_NAME = os.environ["DB_NAME"]
|
||||
DB_NAME = "authservice"
|
||||
|
||||
pwhash = crypt(password, iterations=100000)
|
||||
|
||||
conn = None
|
||||
cur = None
|
||||
try:
|
||||
conn = mariadb.connect(user = DB_USER, password = DB_PASS,
|
||||
host = DB_HOST, database = DB_NAME)
|
||||
conn = psycopg2.connect(database = DB_NAME)
|
||||
conn.autocommit = False
|
||||
|
||||
cur = conn.cursor()
|
||||
cur.execute("""
|
||||
INSERT INTO users (login, pwhash)
|
||||
VALUES(?, ?)
|
||||
INSERT INTO user_t (login, pwhash)
|
||||
VALUES(%s, %s)
|
||||
""", [user, pwhash])
|
||||
cur.execute("""
|
||||
INSERT INTO user_applications_mapping (application, user)
|
||||
INSERT INTO user_application_mapping_t (application,"user")
|
||||
VALUES(
|
||||
(SELECT id FROM applications WHERE name = ?),
|
||||
(SELECT id FROM users WHERE login = ?)
|
||||
(SELECT id FROM application_t WHERE name = %s),
|
||||
(SELECT id FROM user_t WHERE login = %s)
|
||||
)
|
||||
""", [application, user])
|
||||
conn.commit()
|
||||
|
Loading…
x
Reference in New Issue
Block a user