73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# License: MIT. See LICENSE
|
|
|
|
import os
|
|
|
|
import frappe
|
|
from frappe.core.doctype.data_import.data_import import export_json, import_doc
|
|
|
|
|
|
def sync_fixtures(app=None):
|
|
"""Import, overwrite fixtures from `[app]/fixtures`"""
|
|
if app:
|
|
apps = [app]
|
|
else:
|
|
apps = frappe.get_installed_apps()
|
|
|
|
frappe.flags.in_fixtures = True
|
|
|
|
for app in apps:
|
|
fixtures_path = frappe.get_app_path(app, "fixtures")
|
|
if os.path.exists(fixtures_path):
|
|
import_doc(fixtures_path)
|
|
|
|
import_custom_scripts(app)
|
|
|
|
frappe.flags.in_fixtures = False
|
|
|
|
frappe.db.commit()
|
|
|
|
|
|
def import_custom_scripts(app):
|
|
"""Import custom scripts from `[app]/fixtures/custom_scripts`"""
|
|
if os.path.exists(frappe.get_app_path(app, "fixtures", "custom_scripts")):
|
|
for fname in os.listdir(frappe.get_app_path(app, "fixtures", "custom_scripts")):
|
|
if fname.endswith(".js"):
|
|
with open(frappe.get_app_path(app, "fixtures", "custom_scripts") + os.path.sep + fname) as f:
|
|
doctype = fname.rsplit(".", 1)[0]
|
|
script = f.read()
|
|
if frappe.db.exists("Client Script", {"dt": doctype}):
|
|
custom_script = frappe.get_doc("Client Script", {"dt": doctype})
|
|
custom_script.script = script
|
|
custom_script.save()
|
|
else:
|
|
frappe.get_doc({"doctype": "Client Script", "dt": doctype, "script": script}).insert()
|
|
|
|
|
|
def export_fixtures(app=None):
|
|
"""Export fixtures as JSON to `[app]/fixtures`"""
|
|
if app:
|
|
apps = [app]
|
|
else:
|
|
apps = frappe.get_installed_apps()
|
|
for app in apps:
|
|
for fixture in frappe.get_hooks("fixtures", app_name=app):
|
|
filters = None
|
|
or_filters = None
|
|
if isinstance(fixture, dict):
|
|
filters = fixture.get("filters")
|
|
or_filters = fixture.get("or_filters")
|
|
fixture = fixture.get("doctype") or fixture.get("dt")
|
|
print(
|
|
"Exporting {0} app {1} filters {2}".format(fixture, app, (filters if filters else or_filters))
|
|
)
|
|
if not os.path.exists(frappe.get_app_path(app, "fixtures")):
|
|
os.mkdir(frappe.get_app_path(app, "fixtures"))
|
|
|
|
export_json(
|
|
fixture,
|
|
frappe.get_app_path(app, "fixtures", frappe.scrub(fixture) + ".json"),
|
|
filters=filters,
|
|
or_filters=or_filters,
|
|
order_by="idx asc, creation asc",
|
|
)
|