Adding scripts to update support users (#956)

docker/admin/update-support-user.py

@@ -0,0 +1,42 @@
+#!/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()