maya-umbrella 0.5.0__py2.py3-none-any.whl → 0.6.0__py2.py3-none-any.whl

maya_umbrella/vaccine.py

@@ -1,238 +1,12 @@
-# Import built-in modules
-from collections import defaultdict
-import glob
-import logging
-import os
-import re
-
-# Import local modules
-from maya_umbrella.constants import FILE_VIRUS_SIGNATURES
-from maya_umbrella.filesystem import remove_virus_file_by_signature
-from maya_umbrella.filesystem import safe_remove_file
-from maya_umbrella.filesystem import safe_rmtree
-from maya_umbrella.maya_funs import check_reference_node_exists
-from maya_umbrella.maya_funs import cmds
-
-
-class MayaVirusCleaner(object):
-    _bad_files = []
-    _infected_files = []
-    _bad_nodes = []
-    _bad_script_nodes = []
-    _bad_script_jobs = []
-    _registered_callbacks = defaultdict(list)
-    _fix_funcs = []
-
-    def __init__(self, logger=None, auto_fix=True):
-        self.logger = logger or logging.getLogger(__name__)
-        self.auto_fix = auto_fix
-
-    @property
-    def user_app_dir(self):
-        """Return the user application directory."""
-        return cmds.internalVar(userAppDir=True)
-
-    @property
-    def maya_file(self):
-        """Return the current Maya file."""
-        return cmds.file(query=True, sceneName=True, shortName=True) or "empty/scene"
-
-    @property
-    def maya_install_root(self):
-        """Return the Maya installation root directory."""
-        return os.environ["MAYA_LOCATION"]
-
-    @property
-    def user_script_path(self):
-        """Return the user script directory."""
-        return cmds.internalVar(userScriptDir=True)
-
-    @property
-    def local_script_path(self):
-        """Return the local script directory."""
-        return os.path.join(self.user_app_dir, "scripts")
-
-    @property
-    def bad_files(self):
-        """Return a list of bad files."""
-        return [path for path in list(set(self._bad_files)) if os.path.exists(path)]
-
-    @property
-    def bad_nodes(self):
-        """Return a list of bad nodes."""
-        return list(set(self._bad_nodes))
-
-    @property
-    def bad_script_nodes(self):
-        """Return a list of bad script nodes."""
-        return list(set(self._bad_script_nodes))
-
-    @property
-    def bad_script_jobs(self):
-        """Return a list of bad script jobs."""
-        return list(set(self._bad_script_jobs))
-
-    @property
-    def infected_files(self):
-        return self._infected_files
-
-    def callback_remove_rename_temp_files(self, *args, **kwargs):
-        """Remove temporary files in the local script path."""
-        self.logger.info("Removing temporary files in %s", self.local_script_path)
-        for temp_file in glob.glob(os.path.join(self.local_script_path, "._*")):
-            safe_remove_file(temp_file)
-
-    @property
-    def registered_callbacks(self):
-        return self._registered_callbacks
-
-    def add_infected_files(self, files):
-        self._infected_files.extend(files)
-
-    def add_infected_file(self, file):
-        self._infected_files.append(file)
-
-    def add_bad_files(self, files):
-        self._bad_files.extend(files)
-
-    def add_bad_file(self, file):
-        self._bad_files.append(file)
-
-    def add_bad_nodes(self, nodes):
-        self._bad_nodes.extend(nodes)
-
-    def add_bad_node(self, node):
-        self._bad_nodes.append(node)
-
-    def add_bad_script_jobs(self, jobs):
-        self._bad_script_jobs.extend(jobs)
-
-    def add_bad_script_job(self, job):
-        self._bad_script_jobs.append(job)
-
-    def add_bad_script_nodes(self, nodes):
-        self._bad_script_nodes.extend(nodes)
-
-    def add_bad_script_node(self, node):
-        self._bad_script_nodes.append(node)
-
-    def register_callback(self, callback_name, callback):
-        """Register a callback to be executed before or after processing."""
-        self._registered_callbacks[callback_name].append(callback)
-
-    def add_after_open_callback(self, callback):
-        self.register_callback("after_open", callback)
-
-    def add_maya_initialized_callback(self, callback):
-        self.register_callback("maya_initialized", callback)
-
-    def add_after_import_callback(self, callback):
-        self.register_callback("after_import", callback)
-
-    def add_after_import_reference_callback(self, callback):
-        self.register_callback("after_import_reference", callback)
-
-    def add_after_load_reference_callback(self, callback):
-        self.register_callback("after_load_reference", callback)
-
-    def add_before_save_callback(self, callback):
-        self.register_callback("before_save", callback)
-
-    def add_before_import_callback(self, callback):
-        self.register_callback("before_import", callback)
-
-    def add_before_load_reference_callback(self, callback):
-        self.register_callback("before_load_reference", callback)
-
-    def add_before_import_reference_callback(self, callback):
-        self.register_callback("before_import_reference", callback)
-
-    def add_maya_exiting_callback(self, callback):
-        self.register_callback("maya_exiting", callback)
-
-    def setup_default_callbacks(self):
-        self.add_maya_initialized_callback(self.callback_remove_rename_temp_files)
-        self.add_maya_exiting_callback(self.callback_remove_rename_temp_files)
-
-    def add_fix_function(self, func):
-        self._fix_funcs.append(func)
-
-    def fix_script_jobs(self):
-        for script_job in self.bad_script_jobs:
-            script_num = int(re.findall(r"^(\d+):", script_job)[0])
-            self.logger.info("Kill script job %s", script_job)
-            cmds.scriptJob(kill=script_num, force=True)
-            self._bad_script_jobs.remove(script_job)
-
-    def fix_bad_files(self):
-        for file_ in self.bad_files:
-            if os.path.exists(file_):
-                if os.path.isfile(file_):
-                    self.logger.info("Removing %s", file_)
-                    safe_remove_file(file_)
-                    self._bad_files.remove(file_)
-                else:
-                    self.logger.info("Removing folder %s", file_)
-                    safe_rmtree(file_)
-                    self._bad_files.remove(file_)
-
-    def fix_bad_nodes(self):
-        for node in self.bad_nodes:
-            self.logger.info("Deleting %s", node)
-            is_referenced = check_reference_node_exists(node)
-            if is_referenced:
-                try:
-                    cmds.setAttr("{node}.before".format(node=node), "", type="string")
-                    cmds.setAttr("{node}.after".format(node=node), "", type="string")
-                    cmds.setAttr("{node}.scriptType".format(node=node), 0)
-                    self._bad_nodes.remove(node)
-                except Exception as e:
-                    self.logger.debug(e)
-            else:
-                try:
-                    cmds.lockNode(node, lock=False)
-                except ValueError:
-                    pass
-                try:
-                    cmds.delete(node)
-                except ValueError:
-                    pass
-                self._bad_nodes.remove(node)
-
-    def fix_infected_files(self):
-        for file_path in self.infected_files:
-            self.logger.info("Removing infected file: %s", file_path)
-            remove_virus_file_by_signature(file_path, FILE_VIRUS_SIGNATURES)
-            self._infected_files.remove(file_path)
-
-    def fix_all_issues(self):
-        """Fix all issues related to the Maya virus."""
-        self.logger.info("Starting Fixing all issues related to the Maya virus from %s.", self.maya_file)
-        self.fix_bad_files()
-        self.fix_infected_files()
-        self.fix_bad_nodes()
-        self.fix_script_jobs()
-        for func in self._fix_funcs:
-            func()
-        self.logger.info("Finished Fixing all issues related to the Maya virus from %s.", self.maya_file)
-
-    def report_all_issues(self):
-        """Report all issues related to the Maya virus."""
-        self.logger.info("Bad files: %s", self.bad_files)
-        self.logger.info("Bad nodes: %s", self.bad_nodes)
-        self.logger.info("Bad script jobs: %s", self.bad_script_jobs)
-        self.logger.info("Infected files: %s", self.infected_files)
-
-    def reset_all_issues(self):
-        """Reset all issues related to the Maya virus."""
-        self._bad_files = []
-        self._bad_nodes = []
-        self._bad_script_nodes = []
-        self._bad_script_jobs = []
-        self._infected_files = []
+class AbstractVaccine(object):
+    """Abstract base class for Vaccine classes.
 
+    Attributes:
+        virus_name (str): The name of the virus.
+        api (MayaVirusCleaner): The VaccineAPI instance.
+        logger (Logger): The logger instance.
+    """
 
-class AbstractVaccine(object):
     virus_name = None
 
     def __init__(self, api, logger):
@@ -247,7 +21,17 @@ class AbstractVaccine(object):
         self.logger = logger
 
     def collect_issues(self):
+        """Collect issues related to the virus.
+
+        Raises:
+            NotImplementedError: This method must be implemented in the derived classes.
+        """
         raise NotImplementedError
 
     def report_issue(self, name):
-        self.logger.warning("%s: Infected by Malware!", name)
+        """Report an issue related to the virus.
+
+        Args:
+            name (str): The name of the issue.
+        """
+        self.logger.warning(self.api.translator.translate("report_issue", name=name))

maya_umbrella/vaccines/vaccine1.py

@@ -7,9 +7,10 @@ from maya_umbrella.vaccine import AbstractVaccine
 
 class Vaccine(AbstractVaccine):
     """A class for handling the PuTianTongQi virus."""
+    virus_name = "PutTianTongQi"
 
     def collect_issues(self):
-        self.api.add_bad_files(
+        self.api.add_malicious_files(
             [
                 os.path.join(self.api.local_script_path, "fuckVirus.py"),
                 os.path.join(self.api.local_script_path, "fuckVirus.pyc"),

maya_umbrella/vaccines/vaccine2.py

@@ -7,6 +7,7 @@ from maya_umbrella.filesystem import check_virus_by_signature
 from maya_umbrella.filesystem import check_virus_file_by_signature
 from maya_umbrella.maya_funs import check_reference_node_exists
 from maya_umbrella.maya_funs import cmds
+from maya_umbrella.maya_funs import get_attr_value
 from maya_umbrella.vaccine import AbstractVaccine
 
 
@@ -15,38 +16,37 @@ class Vaccine(AbstractVaccine):
 
     virus_name = "zei jian kang"
 
-    def collect_bad_nodes(self):
+    def collect_infected_nodes(self):
         """Collect all bad nodes related to the virus."""
         for script_node in cmds.ls(type="script"):
             if check_reference_node_exists(script_node):
                 continue
-            script_before_string = cmds.getAttr("{}.before".format(script_node))
-            script_after_string = cmds.getAttr("{}.after".format(script_node))
-            for script_string in [script_before_string, script_after_string]:
+            for attr_name in ("before", "after"):
+                script_string = get_attr_value(script_node, attr_name)
                 if not script_string:
                     continue
                 if check_virus_by_signature(script_string, JOB_SCRIPTS_VIRUS_SIGNATURES):
                     self.report_issue(script_node)
-                    self.api.add_bad_node(script_node)
+                    self.api.add_infected_node(script_node)
 
     def collect_issues(self):
         """Collect all issues related to the virus."""
-        self.api.add_bad_files(
+        self.api.add_malicious_files(
             [
                 os.path.join(self.api.local_script_path, "vaccine.py"),
                 os.path.join(self.api.local_script_path, "vaccine.pyc"),
             ],
         )
-        self.collect_bad_usersetup_py()
-        self.collect_bad_nodes()
+        self.collect_infected_user_setup_py()
+        self.collect_infected_nodes()
 
-    def collect_bad_usersetup_py(self):
+    def collect_infected_user_setup_py(self):
         """Collect all bad userSetup.py files related to the virus."""
-        for usersetup_py in [
+        for user_setup_py in [
             os.path.join(self.api.local_script_path, "userSetup.py"),
             os.path.join(self.api.user_script_path, "userSetup.py"),
         ]:
-            if os.path.exists(usersetup_py):
-                if check_virus_file_by_signature(usersetup_py):
-                    self.report_issue(usersetup_py)
-                    self.api.add_infected_file(usersetup_py)
+            if os.path.exists(user_setup_py):
+                if check_virus_file_by_signature(user_setup_py):
+                    self.report_issue(user_setup_py)
+                    self.api.add_infected_file(user_setup_py)

maya_umbrella/vaccines/vaccine3.py

@@ -1,10 +1,14 @@
 # Import built-in modules
 import glob
-import os.path
+import os
 
 # Import local modules
+from maya_umbrella.constants import JOB_SCRIPTS_VIRUS_SIGNATURES
+from maya_umbrella.filesystem import check_virus_by_signature
 from maya_umbrella.filesystem import check_virus_file_by_signature
 from maya_umbrella.maya_funs import cmds
+from maya_umbrella.maya_funs import get_attr_value
+from maya_umbrella.maya_funs import get_reference_file_by_node
 from maya_umbrella.maya_funs import is_maya_standalone
 from maya_umbrella.vaccine import AbstractVaccine
 
@@ -12,23 +16,27 @@ from maya_umbrella.vaccine import AbstractVaccine
 class Vaccine(AbstractVaccine):
     """A class for handling the virus2024429 virus."""
 
-    virus_name = "virus2024429"
+    virus_name = "Virus2024429"
 
-    def collect_bad_nodes(self):
+    def collect_infected_nodes(self):
         """Collect all bad nodes related to the virus."""
         for script_node in cmds.ls(type="script"):
-            if cmds.referenceQuery(script_node, isNodeReferenced=True):
-                continue
-            # check uifiguration
-            if cmds.objExists("{}.KGMScriptProtector".format(script_node)):
-                self.report_issue(script_node)
-                self.api.add_bad_node(script_node)
             # check vaccine
             if "_gene" in script_node:
                 self.report_issue(script_node)
-                self.api.add_bad_node(script_node)
+                self.api.add_infected_node(script_node)
+                self.api.add_infected_reference_file(get_reference_file_by_node(script_node))
+            if "uifiguration" in script_node:
+                for attr_name in ("before", "notes"):
+                    script_string = get_attr_value(script_node, attr_name)
+                    if not script_string:
+                        continue
+                if check_virus_by_signature(script_string, JOB_SCRIPTS_VIRUS_SIGNATURES):
+                    self.report_issue(script_node)
+                    self.api.add_infected_node(script_node)
+                    self.api.add_infected_reference_file(get_reference_file_by_node(script_node))
 
-    def collect_bad_mel_files(self):
+    def collect_infected_mel_files(self):
         """Collect all bad MEL files related to the virus."""
         # check usersetup.mel
         # C:/Users/hallong/Documents/maya/scripts/usersetup.mel
@@ -51,7 +59,7 @@ class Vaccine(AbstractVaccine):
         for script_job in cmds.scriptJob(listJobs=True):
             for virus in virus_gene:
                 if virus in script_job:
-                    self.api.add_bad_script_job(script_job)
+                    self.api.add_infected_script_job(script_job)
 
     def fix_bad_hik_files(self):
         """Fix all bad HIK files related to the virus."""
@@ -63,10 +71,10 @@ class Vaccine(AbstractVaccine):
 
     def collect_issues(self):
         """Collect all issues related to the virus."""
-        self.api.add_bad_file(os.path.join(os.getenv("APPDATA"), "syssst"))
-        self.collect_bad_mel_files()
-        self.collect_bad_nodes()
+        self.api.add_malicious_file(os.path.join(os.getenv("APPDATA"), "syssst"))
+        self.collect_infected_mel_files()
+        self.collect_infected_nodes()
         # This only works for Maya Gui model.
         if not is_maya_standalone():
             self.collect_script_jobs()
-        self.api.add_fix_function(self.fix_bad_hik_files)
+        self.api.add_additionally_fix_function(self.fix_bad_hik_files)

{maya_umbrella-0.5.0.dist-info → maya_umbrella-0.6.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: maya_umbrella
-Version: 0.5.0
+Version: 0.6.0
 Summary: Check and fix maya virus.
 Home-page: https://github.com/loonghao/maya_umbrella
 License: MIT
@@ -43,18 +43,32 @@ Description-Content-Type: text/markdown
 [![maya-2018](https://img.shields.io/badge/maya-2018-green)](https://img.shields.io/badge/maya-2018-green)
 
 <!-- ALL-CONTRIBUTORS-BADGE:START - Do not remove or modify this section -->
-[![All Contributors](https://img.shields.io/badge/all_contributors-4-orange.svg?style=flat-square)](#contributors-)
+[![All Contributors](https://img.shields.io/badge/all_contributors-5-orange.svg?style=flat-square)](#contributors-)
 <!-- ALL-CONTRIBUTORS-BADGE:END -->
 
-This tool is designed to provide a robust solution for identifying and resolving any potential viruses within Autodesk Maya. 
+This tool is designed to provide a robust solution for identifying and resolving any potential viruses within Autodesk
+Maya.
 It ensures a secure and seamless user experience by proactively scanning for threats and effectively neutralizing them.
 
 It can be provided as an API for seamless integration into your existing pipeline.
 
+# 安装
+maya_umbrella是以标准pipy包去发布的，所以我们可以通过pip install去安装
+```shell
+your/maya-root/mayapy -m pip install maya-umbrella
+```
+更新版本
+```shell
+your/maya-root/mayapy -m pip install maya-umbrella --upgrade
+```
+卸载
+```shell
+your/maya-root/mayapy -m pip uninstall  maya-umbrella
+```
 
 # 开发环境设置
 
-Set up the development environment using a virtual environment, 
+Set up the development environment using a virtual environment,
 and it is recommended to use Python 3.8 or higher versions.
 
 通过虚拟环境去设置开发环境, 推荐python-3.8以上的版本
@@ -95,6 +109,14 @@ manual_test_in_maya.start()
 nox -s ruff_check
 ```
 
+# 生成安装包
+
+执行下面的命令可以在<repo>/.zip下面创建zip，参数 `--version` 当前工具的版本号
+
+```shell
+nox -s make-zip -- --version 0.5.0
+```
+
 # 环境变量
 
 我们可以通过下列环境变量去修改maya_umbrella的一些设置，方便有pipeline的公司可以更好的集成
@@ -115,6 +137,45 @@ MAYA_UMBRELLA_LOG_NAME
 
 ```shell
 MAYA_UMBRELLA_LOG_LEVEL
+```
+修改杀毒后文件的备份文件夹名称， 默认是`_virus`
+比如:
+你文件路径是  `c:/your/path/file.ma`
+那么备份文件路径是 `c:/your/path/_maya_umbrella/file.ma`
+```shell
+MAYA_UMBRELLA_BACKUP_FOLDER_NAME
+```
+默认的显示语言，包含日志打印输出等，默认是根据你当前maya的界面语言来设置的，当然我们也可以通过下面的环境变量去设置
+```shell
+MAYA_UMBRELLA_LANG
+```
+
+忽略保存到备份文件夹，*请注意，如果你不清楚这个会导致的后果请不要轻易修改*，默认批量杀毒后会把源文件自动备份到当前文件的备份文件夹.
+```shell
+MAYA_UMBRELLA_IGNORE_BACKUP
+```
+如果忽略请设置为
+```shell
+SET MAYA_UMBRELLA_IGNORE_BACKUP=true
+```
+
+# API
+获取当前场景没有被修复的病毒文件
+
+```python
+from maya_umbrella import MayaVirusDefender
+
+api = MayaVirusDefender()
+print(api.get_unfixed_references())
+```
+
+批量修复文件
+```python
+from maya_umbrella import MayaVirusScanner
+
+api = MayaVirusScanner()
+print(api.scan_files_from_pattern("your/path/*.m[ab]"))
+
 ```
 
 ## Contributors ✨
@@ -131,6 +192,7 @@ Thanks goes to these wonderful people ([emoji key](https://allcontributors.org/d
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/hotwinter0"><img src="https://avatars.githubusercontent.com/u/106237305?v=4?s=100" width="100px;" alt="hotwinter0"/><br /><sub><b>hotwinter0</b></sub></a><br /><a href="https://github.com/loonghao/maya_umbrella/commits?author=hotwinter0" title="Tests">⚠️</a> <a href="https://github.com/loonghao/maya_umbrella/commits?author=hotwinter0" title="Code">💻</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/lingyunfx"><img src="https://avatars.githubusercontent.com/u/73666629?v=4?s=100" width="100px;" alt="lingyunfx"/><br /><sub><b>lingyunfx</b></sub></a><br /><a href="https://github.com/loonghao/maya_umbrella/commits?author=lingyunfx" title="Tests">⚠️</a></td>
       <td align="center" valign="top" width="14.28%"><a href="https://github.com/yjjjj"><img src="https://avatars.githubusercontent.com/u/12741735?v=4?s=100" width="100px;" alt="yjjjj"/><br /><sub><b>yjjjj</b></sub></a><br /><a href="https://github.com/loonghao/maya_umbrella/commits?author=yjjjj" title="Tests">⚠️</a> <a href="https://github.com/loonghao/maya_umbrella/commits?author=yjjjj" title="Code">💻</a></td>
+      <td align="center" valign="top" width="14.28%"><a href="https://github.com/uncleschen"><img src="https://avatars.githubusercontent.com/u/37014389?v=4?s=100" width="100px;" alt="Unclechen"/><br /><sub><b>Unclechen</b></sub></a><br /><a href="https://github.com/loonghao/maya_umbrella/commits?author=uncleschen" title="Tests">⚠️</a></td>
     </tr>
   </tbody>
 </table>

maya_umbrella-0.6.0.dist-info/RECORD

@@ -0,0 +1,28 @@
+maya_umbrella/__init__.py,sha256=uHgbJ-AL3NOfDNB55_7a4_c-OAuP4X8eQ_Yi1udaA0I,401
+maya_umbrella/__version__.py,sha256=cID1jLnC_vj48GgMN6Yb1FA3JsQ95zNmCHmRYE8TFhY,22
+maya_umbrella/cleaner.py,sha256=5EC3w4qRkIsr9AIZneEV8ljhxaOw385Gj2C2KTrzyJg,5044
+maya_umbrella/collector.py,sha256=rAFmvY8Isdle89ezn2-H36hSJd77iBvPBLRPzruCycA,13111
+maya_umbrella/constants.py,sha256=2sl0dL0U82mwu-Msda9B8-USxr89Wu2yu6SYiuaaJZk,539
+maya_umbrella/defender.py,sha256=K5EERl9bwd0IY_ilvnNYXYaQpIe-DUGep-buEb3qLV8,5601
+maya_umbrella/filesystem.py,sha256=CY3MlOXi_fJ8NTITvgotyPiZHyzM4uZQTgMmOYD25T0,8333
+maya_umbrella/hooks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+maya_umbrella/hooks/delete_turtle.py,sha256=OPFRFH1iwonHvETndrP87MZQlJLhxpe564AJE3KyJY8,761
+maya_umbrella/hooks/delete_unknown_plugin_node.py,sha256=xbJvihfjZBhF946ugVqj2IpY_VjOBZt_8KO5omDrJhw,1281
+maya_umbrella/hooks/fix_model_panel.py,sha256=dLuMOz5uQ1nqAboNWMCx-Bi_gHM3FQNTlGxPG5wRYEs,556
+maya_umbrella/hooks/fix_on_model_change_3dc.py,sha256=o4WEQPcHNzaTMXdNnHZWWNCYlHfLxcSFYXR4YW0ZLwk,484
+maya_umbrella/i18n.py,sha256=NCC6isPdchv3yRBS2F4Cguu1bVyi0VoMMcI4YOIxBtU,2683
+maya_umbrella/locales/en_US.json,sha256=LW2gPgO2YJIYR5UfcIBuxW_DS7rf4gkjeVuADqs1V5s,962
+maya_umbrella/locales/zh_CN.json,sha256=UsuRN2yaxxc9LE-7NQkbMqrkjXjNhNegS0QRkQLSTiE,1006
+maya_umbrella/log.py,sha256=IJweFEBTThL6_mW86jAAPKyWNvI79CrMifs5vO3t6ks,1338
+maya_umbrella/maya_funs.py,sha256=WDvREjzLfEsPND0RU-2ErbWgsU6MsILqVBW219p8sZo,3394
+maya_umbrella/scanner.py,sha256=zqo7_Znu1OLI7WrqPBgU1aAxlp19UKVI24s2E-xC6Z8,3710
+maya_umbrella/signatures.py,sha256=RkEXChch7tdqA9Yq6YQ8AMnK2ra6ih0dfnpeZy5FkYs,354
+maya_umbrella/vaccine.py,sha256=aBW6pdT4tD4OMBPZ-d3E4_n16Rylz-2gb7JWzMZVPK0,1022
+maya_umbrella/vaccines/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+maya_umbrella/vaccines/vaccine1.py,sha256=WLo1uJElTLSjVCf5CBtRNU4HKs63my5mkHiGqTfnNEE,489
+maya_umbrella/vaccines/vaccine2.py,sha256=yvHpnrQ6Jts9_zCy8WRJcbxz0CVSRAPkn2vF552atQs,2123
+maya_umbrella/vaccines/vaccine3.py,sha256=d89Ocj1AmsZt9YJjggjDJqAWXcckGLMl-6u8jXeAPjg,3493
+maya_umbrella-0.6.0.dist-info/LICENSE,sha256=tJf0Pz8q_65AjEkm3872K1cl4jGil28vJO5Ko_LhUqc,1060
+maya_umbrella-0.6.0.dist-info/METADATA,sha256=JYAKRBCSJIq-vbrfRfehI97VuZPzAyyTo9drpkgofI0,9178
+maya_umbrella-0.6.0.dist-info/WHEEL,sha256=IrRNNNJ-uuL1ggO5qMvT1GGhQVdQU54d6ZpYqEZfEWo,92
+maya_umbrella-0.6.0.dist-info/RECORD,,

maya_umbrella/core.py

@@ -1,120 +0,0 @@
-# Import built-in modules
-import logging
-
-# Import local modules
-from maya_umbrella.filesystem import get_hooks
-from maya_umbrella.filesystem import get_vaccines
-from maya_umbrella.filesystem import load_hook
-from maya_umbrella.log import setup_logger
-from maya_umbrella.maya_funs import is_maya_standalone
-from maya_umbrella.maya_funs import om
-from maya_umbrella.vaccine import MayaVirusCleaner
-
-
-class MayaVirusDefender(object):
-    callback_ids = []
-    remove_callbacks = []
-    _bad_files = []
-    _vaccines = []
-    callback_maps = {
-        "after_open": om.MSceneMessage.kAfterOpen,
-        "maya_initialized": om.MSceneMessage.kMayaInitialized,
-        "after_import": om.MSceneMessage.kAfterImport,
-        "after_import_reference": om.MSceneMessage.kAfterImportReference,
-        "after_load_reference": om.MSceneMessage.kAfterLoadReference,
-        "before_save": om.MSceneMessage.kBeforeSave,
-        "before_import": om.MSceneMessage.kBeforeImport,
-        "before_load_reference": om.MSceneMessage.kBeforeLoadReference,
-        "before_import_reference": om.MSceneMessage.kBeforeImportReference,
-        "maya_exiting": om.MSceneMessage.kMayaExiting,
-    }
-
-    def __init__(self, auto_fix=True):
-        """Initialize the MayaVirusDefender.
-
-        Args:
-            auto_fix (bool): Whether to automatically fix issues.
-        """
-        logger = logging.getLogger(__name__)
-        self.auto_fix = auto_fix
-        self.logger = setup_logger(logger)
-        self.virus_cleaner = MayaVirusCleaner(self.logger)
-        self.load_vaccines()
-
-    def load_vaccines(self):
-        """Load all vaccines."""
-        for vaccine in get_vaccines():
-            vaccine_class = load_hook(vaccine).Vaccine
-            try:
-                self._vaccines.append(vaccine_class(api=self.virus_cleaner, logger=self.logger))
-            except Exception as e:
-                self.logger.error("Error loading vaccine: %s", e)
-
-    @property
-    def vaccines(self):
-        """Get all loaded vaccines.
-
-        Returns:
-            list: A list of loaded vaccines.
-        """
-        return self._vaccines
-
-    def run_hooks(self):
-        """Run all hooks, only works in non-batch mode."""
-        if not is_maya_standalone():
-            for hook_file in get_hooks():
-                self.logger.debug("run_hook: %s", hook_file)
-                try:
-                    load_hook(hook_file).hook(virus_cleaner=self.virus_cleaner)
-                except Exception as e:
-                    self.logger.error("Error running hook: %s", e)
-
-    def collect(self):
-        """Collect all issues related to the Maya virus."""
-        for vaccine in self.vaccines:
-            vaccine.collect_issues()
-
-    def reset(self):
-        """Reset internal buffer."""
-        self.virus_cleaner.reset_all_issues()
-
-    def fix(self):
-        """Fix all issues related to the Maya virus."""
-        self.virus_cleaner.fix_all_issues()
-
-    def report(self):
-        """Report all issues related to the Maya virus."""
-        self.reset()
-        self.collect()
-        self.virus_cleaner.report_all_issues()
-
-    def setup(self):
-        """Set up the MayaVirusDefender."""
-        self.virus_cleaner.setup_default_callbacks()
-        for name, callbacks in self.virus_cleaner.registered_callbacks.items():
-            maya_callback = self.callback_maps[name]
-            self.logger.debug("%s setup.", name)
-            for func in callbacks:
-                self.callback_ids.append(om.MSceneMessage.addCallback(maya_callback, func))
-        for name, callbacks in self.callback_maps.items():
-            self.logger.debug("setup callback %s.", name)
-            self.callback_ids.append(om.MSceneMessage.addCallback(callbacks, self._callback))
-
-    def _callback(self, *args, **kwargs):
-        """Callback function for MayaVirusDefender.
-
-        Args:
-            *args: Variable length argument list.
-            **kwargs: Arbitrary keyword arguments.
-        """
-        if self.auto_fix:
-            self.reset()
-            self.collect()
-            self.fix()
-            self.run_hooks()
-        else:
-            self.report()
-
-    def start(self):
-        """Start the MayaVirusDefender."""
-        self._callback()