73 lines
2.3 KiB
Python
73 lines
2.3 KiB
Python
# Copyright (c) 2015, Frappe Technologies Pvt. Ltd. and Contributors
|
|
# MIT License. See license.txt
|
|
|
|
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:
|
|
if os.path.exists(frappe.get_app_path(app, "fixtures")):
|
|
fixture_files = sorted(os.listdir(frappe.get_app_path(app, "fixtures")))
|
|
for fname in fixture_files:
|
|
if fname.endswith(".json") or fname.endswith(".csv"):
|
|
import_doc(frappe.get_app_path(app, "fixtures", fname))
|
|
|
|
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")
|