From c3e3fa047b76d81ebcfcbaf2c183c1530a3810f1 Mon Sep 17 00:00:00 2001 From: Usman Nasir Date: Wed, 6 Nov 2019 23:37:32 +0500 Subject: [PATCH] Update upgrade.py --- plogical/upgrade.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/plogical/upgrade.py b/plogical/upgrade.py index 55235357b..318b0f957 100755 --- a/plogical/upgrade.py +++ b/plogical/upgrade.py @@ -1359,6 +1359,12 @@ class Upgrade: if items.find('SESSION_COOKIE_SECURE') > -1: SESSION_COOKIE_SECURE = 0 + DATABASE_ROUTERS = 1 + + for items in data: + if items.find('DATABASE_ROUTERS') > -1: + DATABASE_ROUTERS = 0 + Upgrade.stdOut('Restoring settings file!') writeToFile = open("/usr/local/CyberCP/CyberCP/settings.py", 'w') @@ -1369,6 +1375,7 @@ class Upgrade: writeToFile.writelines(" 'django.middleware.csrf.CsrfViewMiddleware',\n") if items.find('DATABASE_ROUTERS') > -1: + writeToFile.writelines(items) if SESSION_COOKIE_SECURE == 1: con = """SESSION_COOKIE_SECURE = True CSRF_COOKIE_SECURE = True @@ -1416,6 +1423,9 @@ CSRF_COOKIE_SECURE = True writeToFile.writelines("MEDIA_URL = '/home/cyberpanel/media/'\n") writeToFile.writelines('MEDIA_ROOT = MEDIA_URL\n') + if DATABASE_ROUTERS == 1: + writeToFile.writelines("DATABASE_ROUTERS = ['backup.backupRouter.backupRouter']\n") + writeToFile.close() Upgrade.stdOut('Settings file restored!')