139 lines
4.1 KiB
Python
Executable File
139 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import sys
|
|
|
|
|
|
def main():
|
|
if sys.version_info.major < 3:
|
|
print("Error: Python 3.x is required")
|
|
sys.exit(0)
|
|
|
|
if "--silent" not in sys.argv:
|
|
print("- Starting ZeroNet...")
|
|
|
|
main = None
|
|
try:
|
|
import main
|
|
main.start()
|
|
except Exception as err: # Prevent closing
|
|
import traceback
|
|
try:
|
|
import logging
|
|
logging.exception("Unhandled exception: %s" % err)
|
|
except Exception as log_err:
|
|
print("Failed to log error:", log_err)
|
|
traceback.print_exc()
|
|
from Config import config
|
|
error_log_path = config.log_dir + "/error.log"
|
|
traceback.print_exc(file=open(error_log_path, "w"))
|
|
print("---")
|
|
print("Please report it: https://github.com/HelloZeroNet/ZeroNet/issues/new?assignees=&labels=&template=bug-report.md")
|
|
if sys.platform.startswith("win") and "python.exe" not in sys.executable:
|
|
displayErrorMessage(err, error_log_path)
|
|
|
|
if main and (main.update_after_shutdown or main.restart_after_shutdown): # Updater
|
|
if main.update_after_shutdown:
|
|
print("Shutting down...")
|
|
prepareShutdown()
|
|
import update
|
|
print("Updating...")
|
|
update.update()
|
|
if main.restart_after_shutdown:
|
|
print("Restarting...")
|
|
restart()
|
|
else:
|
|
print("Shutting down...")
|
|
prepareShutdown()
|
|
print("Restarting...")
|
|
restart()
|
|
|
|
|
|
def displayErrorMessage(err, error_log_path):
|
|
import ctypes
|
|
import urllib.parse
|
|
import subprocess
|
|
|
|
MB_YESNOCANCEL = 0x3
|
|
MB_ICONEXCLAIMATION = 0x30
|
|
|
|
ID_YES = 0x6
|
|
ID_NO = 0x7
|
|
ID_CANCEL = 0x2
|
|
|
|
err_message = "%s: %s" % (type(err).__name__, err)
|
|
err_title = "Unhandled exception: %s\nReport error?" % err_message
|
|
|
|
res = ctypes.windll.user32.MessageBoxW(0, err_title, "ZeroNet error", MB_YESNOCANCEL | MB_ICONEXCLAIMATION)
|
|
if res == ID_YES:
|
|
import webbrowser
|
|
report_url = "https://github.com/HelloZeroNet/ZeroNet/issues/new?assignees=&labels=&template=bug-report.md&title=%s"
|
|
webbrowser.open(report_url % urllib.parse.quote("Unhandled exception: %s" % err_message))
|
|
if res in [ID_YES, ID_NO]:
|
|
subprocess.Popen(['notepad.exe', error_log_path])
|
|
|
|
def prepareShutdown():
|
|
import atexit
|
|
atexit._run_exitfuncs()
|
|
|
|
# Close log files
|
|
if "main" in sys.modules:
|
|
logger = sys.modules["main"].logging.getLogger()
|
|
|
|
for handler in logger.handlers[:]:
|
|
handler.flush()
|
|
handler.close()
|
|
logger.removeHandler(handler)
|
|
|
|
import time
|
|
time.sleep(1) # Wait for files to close
|
|
|
|
def restart():
|
|
args = sys.argv[:]
|
|
|
|
sys.executable = sys.executable.replace(".pkg", "") # Frozen mac fix
|
|
|
|
if not getattr(sys, 'frozen', False):
|
|
args.insert(0, sys.executable)
|
|
|
|
# Don't open browser after restart
|
|
if "--open_browser" in args:
|
|
del args[args.index("--open_browser") + 1] # argument value
|
|
del args[args.index("--open_browser")] # argument key
|
|
|
|
if getattr(sys, 'frozen', False):
|
|
pos_first_arg = 1 # Only the executable
|
|
else:
|
|
pos_first_arg = 2 # Interpter, .py file path
|
|
|
|
args.insert(pos_first_arg, "--open_browser")
|
|
args.insert(pos_first_arg + 1, "False")
|
|
|
|
if sys.platform == 'win32':
|
|
args = ['"%s"' % arg for arg in args]
|
|
|
|
try:
|
|
print("Executing %s %s" % (sys.executable, args))
|
|
os.execv(sys.executable, args)
|
|
except Exception as err:
|
|
print("Execv error: %s" % err)
|
|
print("Bye.")
|
|
|
|
|
|
def start():
|
|
app_dir = os.path.dirname(os.path.abspath(__file__))
|
|
os.chdir(app_dir) # Change working dir to zeronet.py dir
|
|
sys.path.insert(0, os.path.join(app_dir, "src/lib")) # External liblary directory
|
|
sys.path.insert(0, os.path.join(app_dir, "src")) # Imports relative to src
|
|
|
|
if "--update" in sys.argv:
|
|
sys.argv.remove("--update")
|
|
print("Updating...")
|
|
import update
|
|
update.update()
|
|
else:
|
|
main()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
start()
|