ZeroNet/update.py

128 lines
4.8 KiB
Python
Raw Normal View History

import os
import sys
2019-04-04 13:28:02 +02:00
import json
import re
import shutil
def update():
2019-04-04 13:28:02 +02:00
from Config import config
config.parse(silent=True)
2019-04-04 13:28:02 +02:00
if getattr(sys, 'source_update_dir', False):
if not os.path.isdir(sys.source_update_dir):
os.makedirs(sys.source_update_dir)
source_path = sys.source_update_dir.rstrip("/")
else:
source_path = os.getcwd().rstrip("/")
2019-09-04 20:15:49 +02:00
if config.dist_type.startswith("bundle_linux"):
runtime_path = os.path.normpath(os.path.dirname(sys.executable) + "/../..")
else:
runtime_path = os.path.dirname(sys.executable)
2019-04-04 13:28:02 +02:00
updatesite_path = config.data_dir + "/" + config.updatesite
sites_json = json.load(open(config.data_dir + "/sites.json"))
updatesite_bad_files = sites_json.get(config.updatesite, {}).get("cache", {}).get("bad_files", {})
print(
"Update site path: %s, bad_files: %s, source path: %s, runtime path: %s, dist type: %s" %
(updatesite_path, len(updatesite_bad_files), source_path, runtime_path, config.dist_type)
2019-04-04 13:28:02 +02:00
)
updatesite_content_json = json.load(open(updatesite_path + "/content.json"))
inner_paths = list(updatesite_content_json.get("files", {}).keys())
inner_paths += list(updatesite_content_json.get("files_optional", {}).keys())
2019-04-04 13:28:02 +02:00
# Keep file only in ZeroNet directory
inner_paths = [inner_path for inner_path in inner_paths if re.match("^(core|bundle)", inner_path)]
2019-04-04 13:28:02 +02:00
# Checking plugins
plugins_enabled = []
plugins_disabled = []
if os.path.isdir("%s/plugins" % source_path):
for dir in os.listdir("%s/plugins" % source_path):
if dir.startswith("disabled-"):
plugins_disabled.append(dir.replace("disabled-", ""))
else:
plugins_enabled.append(dir)
print("Plugins enabled:", plugins_enabled, "disabled:", plugins_disabled)
update_paths = {}
2019-04-04 13:28:02 +02:00
for inner_path in inner_paths:
if ".." in inner_path:
continue
2019-04-09 16:21:39 +02:00
inner_path = inner_path.replace("\\", "/").strip("/") # Make sure we have unix path
2019-04-04 13:28:02 +02:00
print(".", end=" ")
if inner_path.startswith("core"):
dest_path = source_path + "/" + re.sub("^core/", "", inner_path)
elif inner_path.startswith(config.dist_type):
dest_path = runtime_path + "/" + re.sub("^bundle[^/]+/", "", inner_path)
else:
continue
2019-04-04 13:28:02 +02:00
if not dest_path:
continue
# Keep plugin disabled/enabled status
2019-08-13 21:07:26 +02:00
match = re.match(re.escape(source_path) + "/plugins/([^/]+)", dest_path)
2019-04-04 13:28:02 +02:00
if match:
plugin_name = match.group(1).replace("disabled-", "")
if plugin_name in plugins_enabled: # Plugin was enabled
dest_path = dest_path.replace("plugins/disabled-" + plugin_name, "plugins/" + plugin_name)
elif plugin_name in plugins_disabled: # Plugin was disabled
dest_path = dest_path.replace("plugins/" + plugin_name, "plugins/disabled-" + plugin_name)
print("P", end=" ")
dest_dir = os.path.dirname(dest_path)
if dest_dir and not os.path.isdir(dest_dir):
os.makedirs(dest_dir)
if dest_dir != dest_path.strip("/"):
update_paths[updatesite_path + "/" + inner_path] = dest_path
num_ok = 0
num_rename = 0
num_error = 0
for path_from, path_to in update_paths.items():
print("-", path_from, "->", path_to)
2019-06-12 02:57:18 +02:00
if not os.path.isfile(path_from):
print("Missing file")
continue
data = open(path_from, "rb").read()
try:
open(path_to, 'wb').write(data)
num_ok += 1
except Exception as err:
2019-04-04 13:28:02 +02:00
try:
print("Error writing: %s. Renaming old file as workaround..." % err)
path_to_tmp = path_to + "-old"
if os.path.isfile(path_to_tmp):
os.unlink(path_to_tmp)
os.rename(path_to, path_to_tmp)
num_rename += 1
open(path_to, 'wb').write(data)
shutil.copymode(path_to_tmp, path_to) # Copy permissions
print("Write done after rename!")
num_ok += 1
2019-04-04 13:28:02 +02:00
except Exception as err:
print("Write error after rename: %s" % err)
num_error += 1
print("* Updated files: %s, renamed: %s, error: %s" % (num_ok, num_rename, num_error))
2019-04-04 13:28:02 +02:00
if __name__ == "__main__":
print('please update zeronet-conservancy via git. usually it can be done via single commnad')
print(' git pull')
print('although it depends on your branches setup')
print('updating through 1update site is not considered safe at the moment')
print('if you really want to use it, edit this file')
2017-06-05 01:07:39 +02:00
if False:
sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src")) # Imports relative to src
update()