import ctypes import logging import os import shutil import sys import threading import time import traceback import webbrowser import requests from io import BytesIO from threading import Thread from zipfile import ZipFile from uuid import uuid1 from hashlib import md5 from win32com.client import Dispatch from win32comext.shell import shell, shellcon import fishy import winshell from fishy import web def not_implemented(): logging.error("Not Implemented") def empty_function(): pass def wait_until(func): while not func(): time.sleep(0.1) def sign(x): return -1 if x < 0 else 1 def open_web(website): """ Opens a website on browser, uses multi-threading so that current thread doesnt get blocked :param website: url """ logging.debug("opening web, please wait...") Thread(target=lambda: webbrowser.open(website, new=2)).start() def initialize_uid(): from .config import config if config.get("uid") is not None: return new_uid = web.register_user() if new_uid is not None: config.set("uid", new_uid) else: logging.error("Couldn't register uid, some features might not work") def _create_new_uid(): """ Creates a unique id for user """ return md5(str(uuid1()).encode()).hexdigest() def install_thread_excepthook(): """ Workaround for sys.excepthook thread bug https://bugs.python.org/issue1230540 (https://sourceforge.net/tracker/?func=detail&atid=105470&aid=1230540&group_id=5470). Call once from __main__ before creating any threads. If using psyco, call psycho.cannotcompile(threading.Thread.run) since this replaces a new-style class method. """ run_old = threading.Thread.run # noinspection PyBroadException def run(*args, **kwargs): try: run_old(*args, **kwargs) except (KeyboardInterrupt, SystemExit): raise except Exception: sys.excepthook(*sys.exc_info()) threading.Thread.run = run def unhandled_exception_logging(*exc_info): text = "".join(traceback.format_exception(*exc_info)) logging.error("Unhandled exception: %s", text) def manifest_file(rel_path): """ returns a file from the manifest files, used to get the files which are installed along with the scripts :param rel_path: relative path from `__init__.py` :return: abs path of the file """ return os.path.join(os.path.dirname(fishy.__file__), rel_path) def create_shortcut_first(): from .config import config if not config.get("shortcut_created", False): create_shortcut(False) config.set("shortcut_created", True) # noinspection PyBroadException def create_shortcut(anti_ghosting: bool): """ creates a new shortcut on desktop """ try: desktop = winshell.desktop() path = os.path.join(desktop, "Fishybot ESO.lnk") shell = Dispatch('WScript.Shell') shortcut = shell.CreateShortCut(path) if anti_ghosting: shortcut.TargetPath = r"C:\Windows\System32\cmd.exe" python_dir = os.path.join(os.path.dirname(sys.executable), "pythonw.exe") shortcut.Arguments = f"/C start /affinity 1 /low {python_dir} -m fishy" else: shortcut.TargetPath = os.path.join(os.path.dirname(sys.executable), "python.exe") shortcut.Arguments = "-m fishy" shortcut.IconLocation = manifest_file("icon.ico") shortcut.save() logging.info("Shortcut created") except Exception: traceback.print_exc() logging.error("Couldn't create shortcut") def get_addondir(): # noinspection PyUnresolvedReferences from win32com.shell import shell, shellcon documents = shell.SHGetFolderPath(0, shellcon.CSIDL_PERSONAL, None, 0) return os.path.join(documents, "Elder Scrolls Online", "live", "Addons") def addon_exists(name, url=None, v=None): return os.path.exists(os.path.join(get_addondir(), name)) def get_addonversion(name, url=None, v=None): if addon_exists(name): txt = name + ".txt" try: with open(os.path.join(get_addondir(), name, txt)) as f: for line in f: if "AddOnVersion" in line: return int(line.split(' ')[2]) except Exception: pass return 0 # noinspection PyBroadException def install_addon(name, url, v=None): try: r = requests.get(url, stream=True) z = ZipFile(BytesIO(r.content)) z.extractall(path=get_addondir()) logging.info("Add-On "+name+" installed successfully!\nPlease make sure to enable \"Allow outdated addons\" in ESO") return 0 except Exception as ex: logging.error("Could not install Add-On "+name+", try doing it manually") return 1 def remove_addon(name, url=None, v=None): try: shutil.rmtree(os.path.join(get_addondir(), name)) logging.info("Add-On "+name+" removed!") except FileNotFoundError: pass except PermissionError as ex: logging.error("Fishy has no permission to remove "+name+" Add-On") return 1 return 0 def get_documents(): return shell.SHGetFolderPath(0, shellcon.CSIDL_PERSONAL, None, 0) def restart(): os.execl(sys.executable, *([sys.executable] + sys.argv)) def log_raise(msg): logging.error(msg) raise Exception(msg) def update(): from .config import config config.delete("dont_ask_update") restart() # noinspection PyProtectedMember,PyUnresolvedReferences def _get_id(thread): # returns id of the respective thread if hasattr(thread, '_thread_id'): return thread._thread_id for id, thread in threading._active.items(): if thread is thread: return id def kill_thread(thread): thread_id = _get_id(thread) res = ctypes.pythonapi.PyThreadState_SetAsyncExc(thread_id, ctypes.py_object(SystemExit)) if res > 1: ctypes.pythonapi.PyThreadState_SetAsyncExc(thread_id, 0) print('Exception raise failure')