def alterAddColumn(database: str, table: str, default: any) -> int:
try:
if type(database) != str or type(table) != str:
return 1
checkData()
if validateIdentifier(database) and validateIdentifier(table):
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
# Get the dbNode
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(),
database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return 3 # table no existente
else:
tuplaTree = serializable.Read(
f"./Data/BPlusMode/{database}/{table}/", table)
res = tuplaTree.addColumn(default)
if res:
return res
else:
serializable.update(
f"./Data/BPlusMode/{database}/{table}/", table,
tuplaTree)
return 0
else:
return 2 # database no existente
else:
return 1
except:
return 1
def loadCSV(filepath, database, table):
if type(database) != str or type(table) != str or type(filepath) != str:
return []
checkData()
if validateIdentifier(database) and validateIdentifier(table):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return []
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return []
try:
res = []
import csv
PKsTree = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
with open(filepath, 'r') as file:
reader = csv.reader(file, delimiter=',')
for row in reader:
res.append(PKsTree.register(row))
serializable.update(f'./Data/BPlusMode/{database}/{table}/', table,
PKsTree)
return res
except:
return []
else:
return []
def alterTable(database: str, tableOld: str, tableNew: str) -> int:
if type(database) != str or type(tableOld) != str or type(tableNew) != str:
return 1
checkData()
if validateIdentifier(tableOld) and validateIdentifier(tableNew):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(), database)
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
rootT = tablesTree.getRoot()
if not tablesTree.search(rootT, tableOld):
return 3 # tableOLD no existente
elif tablesTree.search(rootT, tableNew):
return 4 # tableNEW existente
tablesTree.delete(rootT, tableOld)
serializable.Rename(f'./Data/BPlusMode/{database}/', tableOld,
tableNew)
tablesTree.add(tablesTree.getRoot(), tableNew)
serializable.update(f"./Data/BPlusMode/{database}/", database,
tablesTree)
return 0
else:
return 2 # db no existente
else:
return 1
def alterDropPK(database: str, table: str) -> int:
try:
if type(database) != str or type(table) != str:
return 1
checkData()
if validateIdentifier(database) and validateIdentifier(table):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return 2 # database no existente
else:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return 3 # table no existente
PKsTree = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
res = PKsTree.DeletePk()
if res:
return res
else:
serializable.update(
f'./Data/BPlusMode/{database}/{table}/', table,
PKsTree)
return 0 # exito
else:
return 1
except:
return 1
def truncate(database, table):
if type(database) != str or type(table) != str:
return 1
checkData()
if validateIdentifier(database) and validateIdentifier(table):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return 2 # database no existente
else:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return 3 # table no existente
PKsTree = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
try:
PKsTree.truncate()
serializable.update(f'./Data/BPlusMode/{database}/{table}/',
table, PKsTree)
return 0
except:
return 1
else:
return 1
def update(database, table, register, columns):
if type(database) != str or type(table) != str or type(
register) != dict or type(columns) != list:
return 1
checkData()
if validateIdentifier(database) and validateIdentifier(table):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return 2 # database no existente
else:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return 3 # table no existente
PKsTree = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
try:
res = PKsTree.update(register, columns)
serializable.update(f'./Data/BPlusMode/{database}/{table}/',
table, PKsTree)
return res
except:
return 1
else:
return 1
def extractRangeTable(database, table, columnNumber, lower, upper):
if type(database) != str or type(table) != str or type(
columnNumber) != int:
return None
checkData()
if validateIdentifier(database) and validateIdentifier(table):
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
# Get the dbNode
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(), database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if tablesTree.search(tablesTree.getRoot(), table):
table = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
tableList = list(table.lista().values())
validList = []
if columnNumber < 0 or columnNumber >= table.columns:
return None
try:
for i in tableList:
if type(i[columnNumber]) == str:
if str(i[columnNumber]) <= str(upper) and str(
i[columnNumber]) >= str(lower):
validList.append(i)
elif type(i[columnNumber]) == float:
if float(
i[columnNumber]) <= float(upper) and float(
i[columnNumber]) >= float(lower):
validList.append(i)
elif type(i[columnNumber]) == int:
if int(i[columnNumber]) <= int(upper) and int(
i[columnNumber]) >= int(lower):
validList.append(i)
elif type(i[columnNumber]) == bool:
if bool(i[columnNumber]) <= bool(upper) and bool(
i[columnNumber]) >= bool(lower):
validList.append(i)
except:
return None
return validList
else:
return None
else:
return None
else:
return None
def showTables(database):
if type(database) != str:
return 1
checkData()
if validateIdentifier(database):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
if dataBaseTree.search(dataBaseTree.getRoot(), database):
db = serializable.Read(f"./Data/BPlusMode/{database}/", database)
dbKeys = db.postOrder(db.getRoot())
return [] if len(dbKeys) == 0 else dbKeys[:-1].split("-")
else:
return None
else:
return None
def dropDatabase(database):
if type(database) != str:
return 1
checkData()
if validateIdentifier(database):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return 2
dataBaseTree.delete(root, database)
serializable.delete('./Data/BPlusMode/' + database)
serializable.update('./Data/BPlusMode/', 'Databases', dataBaseTree)
return 0
else:
return 1
def createDatabase(database):
if type(database) != str:
return 1
checkData()
if database and validateIdentifier(database):
dataBaseTree = serializable.Read('./Data/BPlusMode/', 'Databases')
root = dataBaseTree.getRoot()
if dataBaseTree.search(root, database):
return 2
else:
dataBaseTree.add(root, database)
serializable.write('./Data/BPlusMode/', database,
AVLTree.AVLTree())
serializable.update('./Data/BPlusMode/', 'Databases', dataBaseTree)
return 0
else:
return 1
def extractRow(database, table, columns):
if type(database) != str or type(table) != str or type(columns) != list:
return []
checkData()
if validateIdentifier(database) and validateIdentifier(table):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, database):
return [] # database no existente
else:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return [] # table no existente
PKsTree = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
return PKsTree.search(columns) # exito
else:
return []
def alterDatabase(dataBaseOld, dataBaseNew) -> int:
if type(dataBaseOld) != str or type(dataBaseNew) != str:
return 1
checkData()
if validateIdentifier(dataBaseOld) and validateIdentifier(dataBaseNew):
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
if not dataBaseTree.search(root, dataBaseOld):
return 2
if dataBaseTree.search(root, dataBaseNew):
return 3
dataBaseTree.delete(root, dataBaseOld)
root = dataBaseTree.getRoot()
serializable.Rename('./Data/BPlusMode/', dataBaseOld, dataBaseNew)
dataBaseTree.add(root, dataBaseNew)
serializable.update('./Data/BPlusMode/', 'Databases', dataBaseTree)
return 0
else:
return 1
def dropTable(database: str, table: str) -> int:
try:
if type(database) != str or type(table) != str:
return 1
checkData()
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(), database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
root = tablesTree.getRoot()
if not tablesTree.search(root, table):
return 3 # table no existente
else:
tablesTree.delete(root, table)
serializable.delete(f"./Data/BPlusMode/{database}/{table}")
serializable.update(f"./Data/BPlusMode/{database}/", database,
tablesTree)
return 0
else:
return 2
except:
return 1
def createTable(database, table, numberColumns):
if type(database) != str or type(table) != str or type(
numberColumns) != int:
return 1
# Validates identifier before searching
if validateIdentifier(database) and validateIdentifier(
table) and numberColumns > 0:
checkData()
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
# Get the dbNode
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(), database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if tablesTree.search(tablesTree.getRoot(), table):
return 3
else:
# Creates new table node
tablesTree.add(tablesTree.getRoot(), table)
serializable.update(f"./Data/BPlusMode/{database}/", database,
tablesTree)
# Creates bin file for the new table
serializable.write(f"./Data/BPlusMode/{database}/", table,
BplusTree.BPlusTree(5, numberColumns))
return 0
else:
return 2
else:
return 1
def alterDropColumn(database: str, table: str, columnNumber: int) -> int:
try:
if type(database) != str or type(table) != str or type(
columnNumber) != int:
return 1
checkData()
if validateIdentifier(database) and validateIdentifier(table):
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
# Get the dbNode
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(),
database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if not tablesTree.search(tablesTree.getRoot(), table):
return 3 # table no existente
else:
tuplaTree = serializable.Read(
f"./Data/BPlusMode/{database}/{table}/", table)
if columnNumber < 0 or columnNumber >= tuplaTree.columns:
return 5 # out of limit
else:
res = tuplaTree.dropColumn(columnNumber)
if res:
return res
else:
serializable.update(
f"./Data/BPlusMode/{database}/{table}/", table,
tuplaTree)
return 0
else:
return 2 # database no existente
else:
return 1
except:
return 1
def extractTable(database, table):
if type(database) != str or type(table) != str:
return None
checkData()
if validateIdentifier(database) and validateIdentifier(table):
# Get the databases tree
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
# Get the dbNode
databaseNode = dataBaseTree.search(dataBaseTree.getRoot(), database)
# If DB exist
if databaseNode:
tablesTree = serializable.Read(f"./Data/BPlusMode/{database}/",
database)
if tablesTree.search(tablesTree.getRoot(), table):
table = serializable.Read(
f'./Data/BPlusMode/{database}/{table}/', table)
return list(table.lista().values())
else:
return None
else:
return None
else:
return None
def showDatabases():
checkData()
dataBaseTree = serializable.Read('./Data/BPlusMode/', "Databases")
root = dataBaseTree.getRoot()
dbKeys = dataBaseTree.postOrder(root)
return [] if len(dbKeys) == 0 else dbKeys[:-1].split("-")
