style: Black-ish queries

* feat(minor): Added DocType to frappe.query_builder namespace
This commit is contained in:
Gavin D'souza 2021-10-07 17:21:13 +05:30 committed by abhishek
parent f3d810414a
commit 2622f3398e
3 changed files with 11 additions and 6 deletions

View file

@ -8,7 +8,7 @@ import frappe
from frappe.model.base_document import get_controller
from frappe.modules import get_module_path, scrub_dt_dn
from frappe.utils import get_datetime_str
from frappe.query_builder import DocType
def caclulate_hash(path: str) -> str:
"""Calculate md5 hash of the file in binary mode
@ -106,7 +106,9 @@ def import_file_by_path(path, force=False, data_import=False, pre_process=None,
# not using db.set_value to avoid making changes in tabSingles
if doc["doctype"] == "DocType":
doctype_table = frappe.qb.DocType("DocType")
frappe.qb.update(doctype_table).set(doctype_table.migration_hash, calculated_hash).where(doctype_table.name == doc["name"]).run()
frappe.qb.update(doctype_table).set(
doctype_table.migration_hash, calculated_hash
).where(doctype_table.name == doc["name"]).run()
if original_modified:
update_modified(original_modified, doc)
@ -138,7 +140,7 @@ def read_doc_from_file(path):
def update_modified(original_modified, doc):
# since there is a new timestamp on the file, update timestamp in
if doc["doctype"] == doc["name"] and doc["name"] != "DocType":
singles_table = frappe.qb.DocType("Singles")
singles_table = DocType("Singles")
frappe.qb.update(
singles_table
@ -150,8 +152,8 @@ def update_modified(original_modified, doc):
singles_table.doctype == doc["name"]
).run()
else:
doctype_table = frappe.qb.DocType(doc['doctype'])
doctype_table = DocType(doc['doctype'])
frappe.qb.update(doctype_table
).set(
doctype_table.modified, original_modified

View file

@ -1,2 +1,2 @@
from pypika import *
from frappe.query_builder.utils import Column, get_query_builder, patch_query_execute
from frappe.query_builder.utils import Column, DocType, get_query_builder, patch_query_execute

View file

@ -44,6 +44,9 @@ def get_attr(method_string):
methodname = method_string.split('.')[-1]
return getattr(import_module(modulename), methodname)
def DocType(*args, **kwargs):
return frappe.qb.DocType(*args, **kwargs)
def patch_query_execute():
"""Patch the Query Builder with helper execute method
This excludes the use of `frappe.db.sql` method while