115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
import frappe
|
|
from frappe import _
|
|
from frappe.database.schema import DBTable
|
|
|
|
class MariaDBTable(DBTable):
|
|
def create(self):
|
|
additional_definitions = ""
|
|
engine = self.meta.get("engine") or "InnoDB"
|
|
varchar_len = frappe.db.VARCHAR_LEN
|
|
|
|
# columns
|
|
column_defs = self.get_column_definitions()
|
|
if column_defs:
|
|
additional_definitions += ',\n'.join(column_defs) + ',\n'
|
|
|
|
# index
|
|
index_defs = self.get_index_definitions()
|
|
if index_defs:
|
|
additional_definitions += ',\n'.join(index_defs) + ',\n'
|
|
|
|
# child table columns
|
|
if self.meta.get("istable") or 0:
|
|
additional_definitions += ',\n'.join(
|
|
(
|
|
f"parent varchar({varchar_len})",
|
|
f"parentfield varchar({varchar_len})",
|
|
f"parenttype varchar({varchar_len})",
|
|
"index parent(parent)"
|
|
)
|
|
) + ',\n'
|
|
|
|
# create table
|
|
query = f"""create table `{self.table_name}` (
|
|
name varchar({varchar_len}) not null primary key,
|
|
creation datetime(6),
|
|
modified datetime(6),
|
|
modified_by varchar({varchar_len}),
|
|
owner varchar({varchar_len}),
|
|
docstatus int(1) not null default '0',
|
|
idx int(8) not null default '0',
|
|
{additional_definitions}
|
|
index modified(modified))
|
|
ENGINE={engine}
|
|
ROW_FORMAT=DYNAMIC
|
|
CHARACTER SET=utf8mb4
|
|
COLLATE=utf8mb4_unicode_ci"""
|
|
|
|
frappe.db.sql(query)
|
|
|
|
def alter(self):
|
|
for col in self.columns.values():
|
|
col.build_for_alter_table(self.current_columns.get(col.fieldname.lower()))
|
|
|
|
add_column_query = []
|
|
modify_column_query = []
|
|
add_index_query = []
|
|
drop_index_query = []
|
|
|
|
columns_to_modify = set(self.change_type + self.add_unique + self.set_default)
|
|
|
|
for col in self.add_column:
|
|
add_column_query.append("ADD COLUMN `{}` {}".format(col.fieldname, col.get_definition()))
|
|
|
|
for col in columns_to_modify:
|
|
modify_column_query.append("MODIFY `{}` {}".format(col.fieldname, col.get_definition()))
|
|
|
|
for col in self.add_index:
|
|
# if index key does not exists
|
|
if not frappe.db.has_index(self.table_name, col.fieldname + '_index'):
|
|
add_index_query.append("ADD INDEX `{}_index`(`{}`)".format(col.fieldname, col.fieldname))
|
|
|
|
for col in self.drop_index + self.drop_unique:
|
|
if col.fieldname != 'name': # primary key
|
|
current_column = self.current_columns.get(col.fieldname.lower())
|
|
unique_constraint_changed = current_column.unique != col.unique
|
|
if unique_constraint_changed and not col.unique:
|
|
# nosemgrep
|
|
unique_index_record = frappe.db.sql("""
|
|
SHOW INDEX FROM `{0}`
|
|
WHERE Key_name=%s
|
|
AND Non_unique=0
|
|
""".format(self.table_name), (col.fieldname), as_dict=1)
|
|
if unique_index_record:
|
|
drop_index_query.append("DROP INDEX `{}`".format(unique_index_record[0].Key_name))
|
|
index_constraint_changed = current_column.index != col.set_index
|
|
# if index key exists
|
|
if index_constraint_changed and not col.set_index:
|
|
# nosemgrep
|
|
index_record = frappe.db.sql("""
|
|
SHOW INDEX FROM `{0}`
|
|
WHERE Key_name=%s
|
|
AND Non_unique=1
|
|
""".format(self.table_name), (col.fieldname + '_index'), as_dict=1)
|
|
if index_record:
|
|
drop_index_query.append("DROP INDEX `{}`".format(index_record[0].Key_name))
|
|
|
|
try:
|
|
for query_parts in [add_column_query, modify_column_query, add_index_query, drop_index_query]:
|
|
if query_parts:
|
|
query_body = ", ".join(query_parts)
|
|
query = "ALTER TABLE `{}` {}".format(self.table_name, query_body)
|
|
frappe.db.sql(query)
|
|
|
|
except Exception as e:
|
|
# sanitize
|
|
if e.args[0]==1060:
|
|
frappe.throw(str(e))
|
|
elif e.args[0]==1062:
|
|
fieldname = str(e).split("'")[-2]
|
|
frappe.throw(_("{0} field cannot be set as unique in {1}, as there are non-unique existing values").format(
|
|
fieldname, self.table_name))
|
|
elif e.args[0]==1067:
|
|
frappe.throw(str(e.args[1]))
|
|
else:
|
|
raise e
|