def add_default_user(self, uname: str, passw: str) -> bool: """ Create a default user login for PySQL Parameters ---------- uname: str username to be stored passw: str password to be stored Returns ------- bool True if .env file is created else False """ # authenticate user credentials authenticate = auth.Database(uname, passw).authenticate() try: if (authenticate is True): # create an environment file file = open(f"{self.directory}/.env", "w") # add user credentials to .env file file.write(f'UNAME="{uname}"' + "\n") file.write(f'PASSWD="{passw}"' + "\n") file.close() return True else: return False except: return False
def test_authenticate(self): """ Test `authenticate` function Returns True if connection with MySQL server can be made, else returns False """ result = auth.Database("root", "root").authenticate() self.assertEqual(result, True)
def test_auth_table_columns(self): """ Test `auth_table_columns` function Params :database: -> name of database :table: -> name of table to authenticate :query: -> name & type of column """ result = auth.Database("root", "root").auth_table_columns("test", "users", "name varchar(30)") self.assertEqual(result, True)
def test_auth_db(self): """ Test `auth_db` function Params :database: -> name of database to authenticate Returns True if database exists in MySQL server, else returns False """ result = auth.Database("root", "root").auth_db("test") self.assertEqual(result, True)
def test_auth_table(self): """ Test `auth_table` function Params :database: -> name of database :table: -> name of table to authenticate Returns True if table exists in database, else returns False """ result = auth.Database("root", "root").auth_table("test", "users") self.assertEqual(result, True)
def __init__(self, username: str, password: str, database: str, table: str): self.uname = username self.passw = password self.table = table self.db = database # create a `Database` class instance self._auth = auth.Database(self.uname, self.passw) self.connection = mc.connect(host="localhost", user=self.uname, password=self.passw, database=self.db, autocommit=True) self.cursor = self.connection.cursor(buffered=True)
def __init__(self, username: str, password: str): self.uname = username self.passw = password # create a `Database` class instance self._auth = auth.Database(self.uname, self.passw) # authenticate data using auth module authenticate = self._auth.authenticate() if authenticate is True: # initialize connection with MySQL server and cursor object for execution of commands self.connection = mc.connect(host="localhost", user=self.uname, password=self.passw, autocommit=True) self.cursor = self.connection.cursor(buffered=True) else: raise exceptions.AuthenticationError()
# check if a default user exists check = user.User().check_default_user() if check is True: # get default user's name def_user = user.User().get_default_user() print(f"\n[+] Default user {def_user[0]} authenticated") uname = def_user[0] passwd = def_user[1] else: uname = input("Username: "******"Password: "******"\n[-] User could not be authenticated") exit() print(f"\n[+] User authenticated") time.sleep(1) print(info.menu) # package class instances as objects for calling functions ddl_obj = ddl.DDL(uname, passwd) dml_obj = dml.DML(uname, passwd) exp_obj = export.Export(uname, passwd) imp_obj = imports.Import(uname, passwd)