wn/authservice
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use postgres

Browse Source
This commit is contained in:
Wolfgang Hottgenroth
2021-09-11 23:32:25 +02:00
parent 216e1c0684
commit 3541226964
1 changed files with 8 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
asadduser.py
View File

@ -1,6 +1,6 @@
#!/usr/bin/python #!/usr/bin/python
import mariadb import psycopg2
from pbkdf2 import crypt from pbkdf2 import crypt
import argparse import argparse
import os import os
@ -23,30 +23,26 @@ password = args.password
application = args.application application = args.application
DB_USER = os.environ["DB_USER"] DB_NAME = "authservice"
DB_PASS = os.environ["DB_PASS"]
DB_HOST = os.environ["DB_HOST"]
DB_NAME = os.environ["DB_NAME"]
pwhash = crypt(password, iterations=100000) pwhash = crypt(password, iterations=100000)
conn = None conn = None
cur = None cur = None
try: try:
conn = mariadb.connect(user = DB_USER, password = DB_PASS, conn = psycopg2.connect(database = DB_NAME)
host = DB_HOST, database = DB_NAME)
conn.autocommit = False conn.autocommit = False
cur = conn.cursor() cur = conn.cursor()
cur.execute(""" cur.execute("""
INSERT INTO users (login, pwhash) INSERT INTO user_t (login, pwhash)
VALUES(?, ?) VALUES(%s, %s)
""", [user, pwhash]) """, [user, pwhash])
cur.execute(""" cur.execute("""
INSERT INTO user_applications_mapping (application, user) INSERT INTO user_application_mapping_t (application,"user")
VALUES( VALUES(
(SELECT id FROM applications WHERE name = ?), (SELECT id FROM application_t WHERE name = %s),
(SELECT id FROM users WHERE login = ?) (SELECT id FROM user_t WHERE login = %s)
) )
""", [application, user]) """, [application, user])
conn.commit() conn.commit()