fix: revert some changes
This commit is contained in:
parent
8b47c6e5a0
commit
58d8859ddb
2 changed files with 3 additions and 71 deletions
|
|
@ -2082,22 +2082,16 @@ class LinkTableField(DynamicTableField):
|
||||||
) -> None:
|
) -> None:
|
||||||
super().__init__(doctype, fieldname, parent_doctype, alias=alias)
|
super().__init__(doctype, fieldname, parent_doctype, alias=alias)
|
||||||
self.link_fieldname = link_fieldname
|
self.link_fieldname = link_fieldname
|
||||||
self.table = self._get_joined_table()
|
self.table = frappe.qb.DocType(self.doctype)
|
||||||
self.field = self.table[self.fieldname]
|
self.field = self.table[self.fieldname]
|
||||||
|
|
||||||
def _get_joined_table(self):
|
|
||||||
table = frappe.qb.DocType(self.doctype)
|
|
||||||
if self.doctype == self.parent_doctype:
|
|
||||||
table = table.as_(f"tab{self.doctype}_{self.link_fieldname}")
|
|
||||||
return table
|
|
||||||
|
|
||||||
def apply_select(self, query: QueryBuilder, engine: "Engine" = None) -> QueryBuilder:
|
def apply_select(self, query: QueryBuilder, engine: "Engine" = None) -> QueryBuilder:
|
||||||
table = self._get_joined_table()
|
table = frappe.qb.DocType(self.doctype)
|
||||||
query = self.apply_join(query, engine=engine)
|
query = self.apply_join(query, engine=engine)
|
||||||
return query.select(getattr(table, self.fieldname).as_(self.alias or None))
|
return query.select(getattr(table, self.fieldname).as_(self.alias or None))
|
||||||
|
|
||||||
def apply_join(self, query: QueryBuilder, engine: "Engine" = None) -> QueryBuilder:
|
def apply_join(self, query: QueryBuilder, engine: "Engine" = None) -> QueryBuilder:
|
||||||
table = self._get_joined_table()
|
table = frappe.qb.DocType(self.doctype)
|
||||||
main_table = frappe.qb.DocType(self.parent_doctype)
|
main_table = frappe.qb.DocType(self.parent_doctype)
|
||||||
if not query.is_joined(table):
|
if not query.is_joined(table):
|
||||||
query = query.left_join(table).on(table.name == getattr(main_table, self.link_fieldname))
|
query = query.left_join(table).on(table.name == getattr(main_table, self.link_fieldname))
|
||||||
|
|
|
||||||
|
|
@ -1179,68 +1179,6 @@ class TestDBQuery(IntegrationTestCase):
|
||||||
data = get()
|
data = get()
|
||||||
self.assertEqual(len(data["values"]), 1)
|
self.assertEqual(len(data["values"]), 1)
|
||||||
|
|
||||||
def test_self_referential_link_joins(self):
|
|
||||||
"""Test that joined aliases are distinct, when a DocType has multiple links to itself."""
|
|
||||||
|
|
||||||
if not frappe.db.exists("DocType", "Self Linked DocType"):
|
|
||||||
frappe.get_doc(
|
|
||||||
{
|
|
||||||
"doctype": "DocType",
|
|
||||||
"custom": 1,
|
|
||||||
"module": "Custom",
|
|
||||||
"name": "Self Linked DocType",
|
|
||||||
"naming_rule": "Random",
|
|
||||||
"autoname": "hash",
|
|
||||||
"fields": [
|
|
||||||
{
|
|
||||||
"label": "Title",
|
|
||||||
"fieldname": "title",
|
|
||||||
"fieldtype": "Data",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"label": "Parent Ref",
|
|
||||||
"fieldname": "parent_ref",
|
|
||||||
"fieldtype": "Link",
|
|
||||||
"options": "Self Linked DocType",
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"label": "Sibling Ref",
|
|
||||||
"fieldname": "sibling_ref",
|
|
||||||
"fieldtype": "Link",
|
|
||||||
"options": "Self Linked DocType",
|
|
||||||
},
|
|
||||||
],
|
|
||||||
}
|
|
||||||
).insert()
|
|
||||||
else:
|
|
||||||
frappe.db.delete("Self Linked DocType")
|
|
||||||
|
|
||||||
first_link = frappe.get_doc({"doctype": "Self Linked DocType", "title": "Reference1"}).insert()
|
|
||||||
second_link = frappe.get_doc({"doctype": "Self Linked DocType", "title": "Reference2"}).insert()
|
|
||||||
frappe.get_doc(
|
|
||||||
{
|
|
||||||
"doctype": "Self Linked DocType",
|
|
||||||
"title": "Linked Doc",
|
|
||||||
"parent_ref": first_link.name,
|
|
||||||
"sibling_ref": second_link.name,
|
|
||||||
}
|
|
||||||
).insert()
|
|
||||||
|
|
||||||
fields = ["name", "parent_ref.title as parent_title", "sibling_ref.title as sibling_title"]
|
|
||||||
data = frappe.get_all(
|
|
||||||
"Self Linked DocType",
|
|
||||||
fields=fields,
|
|
||||||
)
|
|
||||||
self.assertEqual(len(data), 3)
|
|
||||||
|
|
||||||
query = frappe.qb.get_query(
|
|
||||||
"Self Linked DocType",
|
|
||||||
fields=fields,
|
|
||||||
).get_sql()
|
|
||||||
|
|
||||||
self.assertIn("LEFT JOIN `tabSelf Linked DocType` `tabSelf Linked DocType_parent_ref`", query)
|
|
||||||
self.assertIn("LEFT JOIN `tabSelf Linked DocType` `tabSelf Linked DocType_sibling_ref`", query)
|
|
||||||
|
|
||||||
def test_select_star_expansion(self):
|
def test_select_star_expansion(self):
|
||||||
count = frappe.get_list("Language", [{"SUM": 1}, {"COUNT": "*"}], as_list=1, order_by=None)[0]
|
count = frappe.get_list("Language", [{"SUM": 1}, {"COUNT": "*"}], as_list=1, order_by=None)[0]
|
||||||
self.assertEqual(count[0], frappe.db.count("Language"))
|
self.assertEqual(count[0], frappe.db.count("Language"))
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue