mirror of
https://github.com/frappe/frappe_docker.git
synced 2024-11-14 00:54:07 +00:00
58 lines
1.3 KiB
Python
58 lines
1.3 KiB
Python
import os
|
|
import frappe
|
|
import json
|
|
|
|
from frappe.utils import cint, get_sites
|
|
from check_connection import get_config, COMMON_SITE_CONFIG_FILE
|
|
|
|
|
|
def save_config(config):
|
|
with open(COMMON_SITE_CONFIG_FILE, 'w') as f:
|
|
return json.dump(config, f, indent=1, sort_keys=True)
|
|
|
|
|
|
def set_maintenance_mode(enable=True):
|
|
conf = get_config()
|
|
|
|
if enable:
|
|
conf.update({"maintenance_mode": 1, "pause_scheduler": 1})
|
|
save_config(conf)
|
|
|
|
if not enable:
|
|
conf.update({"maintenance_mode": 0, "pause_scheduler": 0})
|
|
save_config(conf)
|
|
|
|
|
|
def migrate_sites(maintenance_mode=False):
|
|
installed_sites = ":".join(get_sites())
|
|
sites = os.environ.get("SITES", installed_sites).split(":")
|
|
if not maintenance_mode:
|
|
maintenance_mode = cint(os.environ.get("MAINTENANCE_MODE"))
|
|
|
|
if maintenance_mode:
|
|
set_maintenance_mode(True)
|
|
|
|
for site in sites:
|
|
print('Migrating', site)
|
|
frappe.init(site=site)
|
|
frappe.connect()
|
|
try:
|
|
from frappe.migrate import migrate
|
|
migrate()
|
|
finally:
|
|
frappe.destroy()
|
|
|
|
# Disable maintenance mode after migration
|
|
set_maintenance_mode(False)
|
|
|
|
|
|
def main():
|
|
migrate_sites()
|
|
if frappe.redis_server:
|
|
frappe.redis_server.connection_pool.disconnect()
|
|
exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|