mirror of
https://github.com/VinylDNS/vinyldns
synced 2025-08-22 18:17:07 +00:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
#!/usr/bin/python
|
||
|
|
import mysql.connector
|
||
|
|
import VinylDNSProto_pb2
|
||
|
|
import sys
|
||
|
|
import os
|
||
|
|
|
||
|
|
# arguments
|
||
|
|
user_name = sys.argv[1]
|
||
|
|
make_support = sys.argv[2][0].upper()=='T' if len(sys.argv) >= 2 else None
|
||
|
|
|
||
|
|
# environment variables to connect to database
|
||
|
|
db_user = os.environ.get('DB_USER')
|
||
|
|
db_pass = os.environ.get('DB_PASS')
|
||
|
|
db_host = os.environ.get('DB_HOST')
|
||
|
|
db_name = os.environ.get('DB_NAME')
|
||
|
|
db_port = os.environ.get('DB_PORT')
|
||
|
|
|
||
|
|
cnx = mysql.connector.connect(user=db_user, password=db_pass, host=db_host, database=db_name, port=db_port)
|
||
|
|
|
||
|
|
query = "SELECT data FROM user where user_name = %(user_name)s"
|
||
|
|
update = "UPDATE user SET data = %(pb)s WHERE user_name = %(user_name)s"
|
||
|
|
|
||
|
|
cursor = cnx.cursor(dictionary=True)
|
||
|
|
|
||
|
|
cursor.execute(query, { 'user_name': user_name })
|
||
|
|
user = VinylDNSProto_pb2.User()
|
||
|
|
for row in cursor:
|
||
|
|
user_raw = row['data']
|
||
|
|
user = VinylDNSProto_pb2.User()
|
||
|
|
user.ParseFromString(user_raw)
|
||
|
|
print("FOUND USER NAME {}, IS SUPPORT = {}".format(user.userName, user.isSupport))
|
||
|
|
|
||
|
|
if make_support is not None:
|
||
|
|
print("Updating {}, Support = {}".format(user_name, make_support))
|
||
|
|
user.isSupport = make_support
|
||
|
|
cursor.execute(update, { 'pb': user.SerializeToString(), 'user_name': user_name })
|
||
|
|
cnx.commit()
|
||
|
|
else:
|
||
|
|
print("Skipping making support as no make support value provided")
|
||
|
|
|
||
|
|
cursor.close()
|
||
|
|
cnx.close()
|