81b37cb7d2
refactor: clean up code to py39+ supported syntax - f-strings instead of format - latest typing support instead of pre 3.9 TitleCase - remove UTF-8 declarations. - many more changes Powered by https://github.com/asottile/pyupgrade/ + manual cleanups
63 lines
1.5 KiB
Python
63 lines
1.5 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
"""docfield utililtes"""
|
|
|
|
import frappe
|
|
|
|
|
|
def rename(doctype, fieldname, newname):
|
|
"""rename docfield"""
|
|
df = frappe.db.sql(
|
|
"""select * from tabDocField where parent=%s and fieldname=%s""", (doctype, fieldname), as_dict=1
|
|
)
|
|
if not df:
|
|
return
|
|
|
|
df = df[0]
|
|
|
|
if frappe.db.get_value("DocType", doctype, "issingle"):
|
|
update_single(df, newname)
|
|
else:
|
|
update_table(df, newname)
|
|
update_parent_field(df, newname)
|
|
|
|
|
|
def update_single(f, new):
|
|
"""update in tabSingles"""
|
|
frappe.db.begin()
|
|
frappe.db.sql(
|
|
"""update tabSingles set field=%s where doctype=%s and field=%s""",
|
|
(new, f["parent"], f["fieldname"]),
|
|
)
|
|
frappe.db.commit()
|
|
|
|
|
|
def update_table(f, new):
|
|
"""update table"""
|
|
query = get_change_column_query(f, new)
|
|
if query:
|
|
frappe.db.sql(query)
|
|
|
|
|
|
def update_parent_field(f, new):
|
|
"""update 'parentfield' in tables"""
|
|
if f["fieldtype"] in frappe.model.table_fields:
|
|
frappe.db.begin()
|
|
frappe.db.sql(
|
|
"""update `tab{}` set parentfield={} where parentfield={}""".format(f["options"], "%s", "%s"),
|
|
(new, f["fieldname"]),
|
|
)
|
|
frappe.db.commit()
|
|
|
|
|
|
def get_change_column_query(f, new):
|
|
"""generate change fieldname query"""
|
|
desc = frappe.db.sql("desc `tab%s`" % f["parent"])
|
|
for d in desc:
|
|
if d[0] == f["fieldname"]:
|
|
return "alter table `tab{}` change `{}` `{}` {}".format(f["parent"], f["fieldname"], new, d[1])
|
|
|
|
|
|
def supports_translation(fieldtype):
|
|
return fieldtype in ["Data", "Select", "Text", "Small Text", "Text Editor"]
|