def insert_into_table(user_id): try: # Prepared Statement sql = "INSERT INTO BSqnOU0gA6.users_dateTime (user_id, user_name, creation_date) VALUES (%s, %s, %s)" # Connect to database and get cursor conn, cursor = connect() # Check if id exists and if it does it will get the next available id (1 above the last line in table) select(table="BSqnOU0gA6.users_dateTime", where=["user_id", user_id], conn=conn, cursor=cursor) cursorLength = cursor.rowcount if cursorLength > 0: select(table="BSqnOU0gA6.users_dateTime", conn=conn, cursor=cursor) last_id = (list(cursor)[-1])[0] user_id = last_id + 1 data = request.json # Get data from json payload date = datetime.now( ) # Get current date and time for creation date field in users table post("BSqnOU0gA6.users_dateTime", [ user_id, str(data.get("user_name")), date.strftime("%Y-%m-%d %H:%M:%S") ], conn=conn, cursor=cursor) # Disconnect from Database disconnect(conn, cursor) # If user generation succeeded return {"status": "ok", "user_added": data.get("user_name")}, 200 except Exception as err: return {"status": "error", "reason": "id already exist"}, 500
def select_from_table(user_id): try: # Connect to database and get cursor conn, cursor = connect() select(table="BSqnOU0gA6.users_dateTime", where=["user_id", user_id], conn=conn, cursor=cursor) # set the cursor name = None for row in cursor: name = row[1] disconnect(conn, cursor) # Disconnect from Database if name is not None: return {"status": "ok", "user_name": name}, 200 else: return {"status": "error", "reason": "no such id"}, 500 except Exception as err: print(err)
def get_user_name(user_id): # user db_connection library to select data from db conn, cursor = connect() user_name = select( table='BSqnOU0gA6.users_dateTime', select_value="*", where=["user_id", user_id], conn=conn, cursor=cursor ) # SELECT user_name FROM BSqnOU0gA6.users_dateTime WHERE user_id = %s if user_name != 0: for row in cursor: disconnect(conn, cursor) return "<h1 id='user'>" + row[1] + "</h1>" else: disconnect(conn, cursor) return "<h1 id='error'>" + 'no such user: ' + user_id + "</h1>"
from selenium.webdriver.common.by import By from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC import requests from Module.db_connector import connect, disconnect, select # Backend check part try: # Connect to database and get cursor conn, cursor = connect() # Get all data from config table select('BSqnOU0gA6.config', conn=conn, cursor=cursor) last_line = list(cursor)[cursor.arraysize] user_id = input("Insert requested id for entry creation: ") http_link = str(last_line[1]) browser = str(last_line[2]) user_name = str(last_line[3]) # request a post action to store data inside the database requests.post("%s/%s" % (http_link, user_id), json={"user_name": user_name}) # Check if user created successfully data = requests.get("%s/%s" % (http_link, user_id)) if data.status_code == 200 and data.json()["user_name"] == user_name: print(
requests.post("http://127.0.0.1:5000/users/{}".format(user_id), json={"user_name": user_name}) # Check if user created successfully data = requests.get("http://127.0.0.1:5000/users/{}".format(user_id)) if data.status_code == 200 and data.json()["user_name"] in user_name: print( "\nStatus code is \'%i\' and user name is \'%s\' as requested by user.\n" % (data.status_code, user_name)) # Connect to database and get cursor conn, cursor = connect() # Create table and select query select('BSqnOU0gA6.users_dateTime', where=["user_id", user_id], conn=conn, cursor=cursor) for row in cursor: print( "User\'s ID and Name are \'%s\' and \'%s\' and the values the user asked for are ID \'%s\' and Name " "\'%s\'." % (row[0], row[1], user_id, user_name)) disconnect(conn, cursor) else: if data.status_code != 200: print( "Status code is \'%i\', the user was not created as requested." % data.status_code) raise Exception("Test Failed")