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)
