maya-umbrella 0.14.2__tar.gz → 0.15.0__tar.gz

{maya_umbrella-0.14.2 → maya_umbrella-0.15.0}/PKG-INFO

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: maya_umbrella
-Version: 0.14.2
+Version: 0.15.0
 Summary: A better Autodesk Maya antivirus tool detects and removes malicious.
-Home-page: https://github.com/loonghao/maya_umbrella
 License: MIT
+License-File: LICENSE
 Keywords: Autodesk Maya,python,Maya,dcc,antivirus,Security Tools,maya_umbrella
 Author: longhao
 Author-email: hal.long@outlook.com
@@ -19,11 +19,16 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Project-URL: Documentation, https://github.com/loonghao/maya_umbrella
+Project-URL: Homepage, https://github.com/loonghao/maya_umbrella
 Project-URL: Repository, https://github.com/loonghao/maya_umbrella
 Description-Content-Type: text/markdown
 
-![maya_umbrella.ing Banner](https://raw.githubusercontent.com/loonghao/maya_umbrella/main/resources/banner.png)
+![Maya Umbrella Banner](https://raw.githubusercontent.com/loonghao/maya_umbrella/main/resources/banner.png)
+
+[English](https://github.com/loonghao/maya_umbrella/blob/main/README.md) | [中文](https://github.com/loonghao/maya_umbrella/blob/main/README_zh.md)
 
 [![Python Version](https://img.shields.io/pypi/pyversions/maya-umbrella)](https://img.shields.io/pypi/pyversions/maya-umbrella)
 [![Nox](https://img.shields.io/badge/%F0%9F%A6%8A-Nox-D85E00.svg)](https://github.com/wntrblm/nox)
@@ -46,7 +51,7 @@ Description-Content-Type: text/markdown
 [![All Contributors](https://img.shields.io/badge/all_contributors-9-orange.svg?style=flat-square)](#contributors-)
 <!-- ALL-CONTRIBUTORS-BADGE:END -->
 
-A better Autodesk Maya antivirus tool detects and removes malicious.
+A better Autodesk Maya antivirus tool that detects and removes malicious code.
 
 This tool is designed to provide a robust solution for identifying and resolving any potential viruses within Autodesk
 Maya.
@@ -54,144 +59,107 @@ It ensures a secure and seamless user experience by proactively scanning for thr
 
 It can be provided as an API for seamless integration into your existing pipeline.
 
-# 安装
+# Installation
 
-## pip 安装
+## pip installation
 maya_umbrella is distributed as a standard pipy package, so we can install it via pip install.
-
-maya_umbrella是以标准pipy包去发布的，所以我们可以通过pip install去安装
 ```shell
 your/maya-root/mayapy -m pip install maya-umbrella
 ```
-## 一键安装
+## One-click installation
 Download the corresponding version of the zip package in the release, unzip it and double-click `install.bat` to install it.
 
-在release里面下载对应版本的zip包，解压后双击`install.bat`即可安装
-
-更新版本
+Update version
 ```shell
 your/maya-root/mayapy -m pip install maya-umbrella --upgrade
 ```
-卸载
+Uninstall
 ```shell
 your/maya-root/mayapy -m pip uninstall  maya-umbrella
 ```
 
-# 开发环境设置
+# Development Environment Setup
 
 Set up the development environment using a virtual environment,
 and it is recommended to use Python 3.8 or higher versions.
 
-通过虚拟环境去设置开发环境, 推荐python-3.8以上的版本
-
-通过pip安装相关开发依赖
+Install development dependencies via pip
 
 ```shell
 pip install -r requirements-dev.txt
 ```
 
-# 开发调试
+# Development Debugging
 
 
-## 在maya中测试
-With `nox -s maya -- <maya version>`, start maya.
-Nox will dynamically register a nox session based on your local installation of maya,
-e.g. if you have `maya-2018` installed locally, then you can start maya with a test environment.
-Then you can start maya with a test environment via
-
-通过`nox -s maya -- <maya version>`, 启动maya.
-nox会动态根据你本地安装得maya去注册nox session, 比如你本地安装了`maya-2018`，
-那么通过可以启动带有测试环境的maya
+## Testing in Maya
+With `nox -s maya -- <maya version>`, start Maya.
+Nox will dynamically register a nox session based on your local installation of Maya,
+e.g. if you have `maya-2018` installed locally, then you can start Maya with a test environment.
 
 ```shell
 nox -s maya -- 2018
 ```
 **Note: there are two `-` between maya and the version number**.
 
-Starting maya and executing the following code in the script editor will dynamically open the ma file from `<repo>/tests/virus/` to test it.
-
-**注意：maya 与 版本号之间有 俩个`-`**
-
-启动maya后在脚本编辑器中执行下面得代码，就会动态的从`<repo>/tests/virus/`里面去open ma文件去进行测试.
+After starting Maya, executing the following code in the script editor will dynamically open the ma file from `<repo>/tests/virus/` to test it.
 
 ```python
 import manual_test_in_maya
 
 manual_test_in_maya.start()
 ```
-It is also possible to execute the corresponding tests via pytest, which also requires a local installation of the corresponding maya
-
-也可以通过pytest去执行对应的测试，也需要本地安装了对应的maya
+It is also possible to execute the corresponding tests via pytest, which also requires a local installation of the corresponding Maya
 
 ```shell
 nox -s maya -- 2018 --test
 ```
 **Note: Command line crash may occur in versions below maya-2022 (PY2)**.
 
-**注意：在maya-2022 (PY2) 以下的版本可能会出现命令行crash的情况**
 
-
-## 增加新的疫苗
-Create a new py in `<repo>/maya_umbrella/vaccines/`, since many viruses don't have a specific name, we'll use `vaccine<id>.py`.
+## Adding New Vaccines
+Create a new py in `<repo>/maya_umbrella/vaccines/`. Since many viruses don't have a specific name, we'll use `vaccine<id>.py`.
 Inherit `from maya_umbrella.vaccine import AbstractVaccine` and call the class `Vaccine`, and then write the virus collection logic.
 
-在`<repo>/maya_umbrella/vaccines/` 新建一个py, 因为有很多病毒没有具体的名字代号，我们统一以`vaccine<id>.py`
-继承`from maya_umbrella.vaccine import AbstractVaccine`并且class名字叫`Vaccine`即可, 然后去写具体的收集病毒逻辑
-
-## 代码检查
+## Code Check
 
 We can use the encapsulated `nox` command to perform a code check.
 
-我们可以利用封装好的`nox`命令去执行进行代码检查
-
 ```shell
 nox -s lint
 ```
 
-进行代码整理
+Format code
 ```shell
 nox -s lint-fix
 ```
 
-# 生成安装包
+# Generate Installation Package
 Execute the following command to create a zip under <repo>/.zip, with `--version` the version number of the current tool.
 
 **Note: between `make-zip` and `--version` there are two `-`**.
 
-执行下面的命令可以在<repo>/.zip下面创建zip，参数 `--version` 当前工具的版本号
-
-**注意：`make-zip` 与 `--version`之间有 俩个`-`**
-
 ```shell
 nox -s make-zip -- --version 0.5.0
 ```
 
-# 环境变量
-We can use the following environment variables to modify some of the settings of maya_umbrella, 
-
+# Environment Variables
+We can use the following environment variables to modify some of the settings of maya_umbrella,
 so that companies with pipelines can better integrate them.
 
 Modify the log saving directory of maya umbrella, the default is the windows temp directory.
 
-我们可以通过下列环境变量去修改maya_umbrella的一些设置，方便有pipeline的公司可以更好的集成
-
-修改maya umbrella的日志保存目录，默认是windows temp目录
-
 ```shell
 MAYA_UMBRELLA_LOG_ROOT
 ```
 Change the name of the log file for maya umbrella, default is `maya_umbrella`.
 
-修改maya umbrella的日志文件名称, 默认是`maya_umbrella`
-
 ```shell
 MAYA_UMBRELLA_LOG_NAME
 ```
 
 Set the log level, the default is info, can be debug can see more log information.
 
-设置日志级别，默认是info, 可以是debug可以看到更多的日志信息.
-
 ```shell
 MAYA_UMBRELLA_LOG_LEVEL
 ```
@@ -202,53 +170,36 @@ For example:
 Your file path is `c:/your/path/file.ma`.
 
 Then the backup file path is `c:/your/path/_virus/file.ma`.
-
-修改杀毒后文件的备份文件夹名称， 默认是`_virus`
-比如:
-你文件路径是  `c:/your/path/file.ma`
-那么备份文件路径是 `c:/your/path/_virus/file.ma`
 ```shell
 MAYA_UMBRELLA_BACKUP_FOLDER_NAME
 ```
 The default display language, including logging printouts, etc.
-
-is set by default according to your current maya interface language, 
-
+is set by default according to your current maya interface language,
 but of course we can also set it via the following environment variables.
-
-默认的显示语言，包含日志打印输出等，默认是根据你当前maya的界面语言来设置的，当然我们也可以通过下面的环境变量去设置.
 ```shell
 MAYA_UMBRELLA_LANG
 ```
 Ignore saving to the backup folder,
-*please note that if you are not clear about the consequences of this please do not modify it easily*, 
-the default batch antivirus will automatically back up the source file to the current file's backup folder 
+*please note that if you are not clear about the consequences of this please do not modify it easily*,
+the default batch antivirus will automatically back up the source file to the current file's backup folder
 after the batch antivirus.
 
-忽略保存到备份文件夹，*请注意，如果你不清楚这个会导致的后果请不要轻易修改*，默认批量杀毒后会把源文件自动备份到当前文件的备份文件夹.
-
 ```shell
 MAYA_UMBRELLA_IGNORE_BACKUP
 ```
 If ignored please set to
-
-如果忽略请设置为
 ```shell
 SET MAYA_UMBRELLA_IGNORE_BACKUP=true
 ```
 
-For the portable version of Maya, 
+For the portable version of Maya,
 you can specify the Maya path by adding the `MAYA_LOCATION` environment variable.
 
-如果是便携版Maya，可以通过添加 `MAYA_LOCATION` 环境变量指定Maya路径
-
 ```shell
 SET MAYA_LOCATION=d:/your/path/maya_version/
 ```
 You can also specify a directory from the command line.
 
-也可以通过命令行指定目录
-
 ```shell
 nox -s maya -- 2018 --install-root /your/local/maya/root
 
@@ -258,8 +209,6 @@ nox -s maya -- 2018 --install-root /your/local/maya/root
 
 Get virus files that have not been repaired in the current scenario.
 
-获取当前场景没有被修复的病毒文件
-
 ```python
 from maya_umbrella import MayaVirusDefender
 
@@ -268,8 +217,6 @@ print(api.get_unfixed_references())
 ```
 
 Batch repair of files, via regular expressions.
-
-批量修复文件, 通过正则表达式
 ```python
 from maya_umbrella import MayaVirusScanner
 
@@ -278,7 +225,7 @@ print(api.scan_files_from_pattern("your/path/*.m[ab]"))
 
 ```
 
-# 案例
+# Examples
 
 If you want to quickly go through maya standalone and batch clean up maya files.
 
@@ -291,14 +238,6 @@ For example, if you have `2018` installed locally, Then `nox -s maya -- 2018 --s
 
 The following syntax starts a maya-2020 environment to dynamically check for viruses from the `c:/test` folder.
 
-如果你想要快速通过maya standalone去批量清理maya文件，
-可以`下载`或者`git clone`当前`main`分支的工程，
-根据上面指引设置好开发环境,
-通过`nox`命令去启动maya `standalone`环境，maya版本根据你当前本地安装的maya为准，
-比如你本地安装了`2018`,
-那么就是 `nox -s maya -- 2018 --standalone`
-下面的语法是启动一个maya-2020的环境去动态从`c:/test`文件夹下去查杀病毒
-
 ```shell
 nox -s maya -- 2018 --standalone --pattern c:/test/*.m[ab]
 ```

{maya_umbrella-0.14.2 → maya_umbrella-0.15.0}/README.md

@@ -1,4 +1,6 @@
-![maya_umbrella.ing Banner](https://raw.githubusercontent.com/loonghao/maya_umbrella/main/resources/banner.png)
+![Maya Umbrella Banner](https://raw.githubusercontent.com/loonghao/maya_umbrella/main/resources/banner.png)
+
+[English](https://github.com/loonghao/maya_umbrella/blob/main/README.md) | [中文](https://github.com/loonghao/maya_umbrella/blob/main/README_zh.md)
 
 [![Python Version](https://img.shields.io/pypi/pyversions/maya-umbrella)](https://img.shields.io/pypi/pyversions/maya-umbrella)
 [![Nox](https://img.shields.io/badge/%F0%9F%A6%8A-Nox-D85E00.svg)](https://github.com/wntrblm/nox)
@@ -21,7 +23,7 @@
 [![All Contributors](https://img.shields.io/badge/all_contributors-9-orange.svg?style=flat-square)](#contributors-)
 <!-- ALL-CONTRIBUTORS-BADGE:END -->
 
-A better Autodesk Maya antivirus tool detects and removes malicious.
+A better Autodesk Maya antivirus tool that detects and removes malicious code.
 
 This tool is designed to provide a robust solution for identifying and resolving any potential viruses within Autodesk
 Maya.
@@ -29,144 +31,107 @@ It ensures a secure and seamless user experience by proactively scanning for thr
 
 It can be provided as an API for seamless integration into your existing pipeline.
 
-# 安装
+# Installation
 
-## pip 安装
+## pip installation
 maya_umbrella is distributed as a standard pipy package, so we can install it via pip install.
-
-maya_umbrella是以标准pipy包去发布的，所以我们可以通过pip install去安装
 ```shell
 your/maya-root/mayapy -m pip install maya-umbrella
 ```
-## 一键安装
+## One-click installation
 Download the corresponding version of the zip package in the release, unzip it and double-click `install.bat` to install it.
 
-在release里面下载对应版本的zip包，解压后双击`install.bat`即可安装
-
-更新版本
+Update version
 ```shell
 your/maya-root/mayapy -m pip install maya-umbrella --upgrade
 ```
-卸载
+Uninstall
 ```shell
 your/maya-root/mayapy -m pip uninstall  maya-umbrella
 ```
 
-# 开发环境设置
+# Development Environment Setup
 
 Set up the development environment using a virtual environment,
 and it is recommended to use Python 3.8 or higher versions.
 
-通过虚拟环境去设置开发环境, 推荐python-3.8以上的版本
-
-通过pip安装相关开发依赖
+Install development dependencies via pip
 
 ```shell
 pip install -r requirements-dev.txt
 ```
 
-# 开发调试
+# Development Debugging
 
 
-## 在maya中测试
-With `nox -s maya -- <maya version>`, start maya.
-Nox will dynamically register a nox session based on your local installation of maya,
-e.g. if you have `maya-2018` installed locally, then you can start maya with a test environment.
-Then you can start maya with a test environment via
-
-通过`nox -s maya -- <maya version>`, 启动maya.
-nox会动态根据你本地安装得maya去注册nox session, 比如你本地安装了`maya-2018`，
-那么通过可以启动带有测试环境的maya
+## Testing in Maya
+With `nox -s maya -- <maya version>`, start Maya.
+Nox will dynamically register a nox session based on your local installation of Maya,
+e.g. if you have `maya-2018` installed locally, then you can start Maya with a test environment.
 
 ```shell
 nox -s maya -- 2018
 ```
 **Note: there are two `-` between maya and the version number**.
 
-Starting maya and executing the following code in the script editor will dynamically open the ma file from `<repo>/tests/virus/` to test it.
-
-**注意：maya 与 版本号之间有 俩个`-`**
-
-启动maya后在脚本编辑器中执行下面得代码，就会动态的从`<repo>/tests/virus/`里面去open ma文件去进行测试.
+After starting Maya, executing the following code in the script editor will dynamically open the ma file from `<repo>/tests/virus/` to test it.
 
 ```python
 import manual_test_in_maya
 
 manual_test_in_maya.start()
 ```
-It is also possible to execute the corresponding tests via pytest, which also requires a local installation of the corresponding maya
-
-也可以通过pytest去执行对应的测试，也需要本地安装了对应的maya
+It is also possible to execute the corresponding tests via pytest, which also requires a local installation of the corresponding Maya
 
 ```shell
 nox -s maya -- 2018 --test
 ```
 **Note: Command line crash may occur in versions below maya-2022 (PY2)**.
 
-**注意：在maya-2022 (PY2) 以下的版本可能会出现命令行crash的情况**
 
-
-## 增加新的疫苗
-Create a new py in `<repo>/maya_umbrella/vaccines/`, since many viruses don't have a specific name, we'll use `vaccine<id>.py`.
+## Adding New Vaccines
+Create a new py in `<repo>/maya_umbrella/vaccines/`. Since many viruses don't have a specific name, we'll use `vaccine<id>.py`.
 Inherit `from maya_umbrella.vaccine import AbstractVaccine` and call the class `Vaccine`, and then write the virus collection logic.
 
-在`<repo>/maya_umbrella/vaccines/` 新建一个py, 因为有很多病毒没有具体的名字代号，我们统一以`vaccine<id>.py`
-继承`from maya_umbrella.vaccine import AbstractVaccine`并且class名字叫`Vaccine`即可, 然后去写具体的收集病毒逻辑
-
-## 代码检查
+## Code Check
 
 We can use the encapsulated `nox` command to perform a code check.
 
-我们可以利用封装好的`nox`命令去执行进行代码检查
-
 ```shell
 nox -s lint
 ```
 
-进行代码整理
+Format code
 ```shell
 nox -s lint-fix
 ```
 
-# 生成安装包
+# Generate Installation Package
 Execute the following command to create a zip under <repo>/.zip, with `--version` the version number of the current tool.
 
 **Note: between `make-zip` and `--version` there are two `-`**.
 
-执行下面的命令可以在<repo>/.zip下面创建zip，参数 `--version` 当前工具的版本号
-
-**注意：`make-zip` 与 `--version`之间有 俩个`-`**
-
 ```shell
 nox -s make-zip -- --version 0.5.0
 ```
 
-# 环境变量
-We can use the following environment variables to modify some of the settings of maya_umbrella, 
-
+# Environment Variables
+We can use the following environment variables to modify some of the settings of maya_umbrella,
 so that companies with pipelines can better integrate them.
 
 Modify the log saving directory of maya umbrella, the default is the windows temp directory.
 
-我们可以通过下列环境变量去修改maya_umbrella的一些设置，方便有pipeline的公司可以更好的集成
-
-修改maya umbrella的日志保存目录，默认是windows temp目录
-
 ```shell
 MAYA_UMBRELLA_LOG_ROOT
 ```
 Change the name of the log file for maya umbrella, default is `maya_umbrella`.
 
-修改maya umbrella的日志文件名称, 默认是`maya_umbrella`
-
 ```shell
 MAYA_UMBRELLA_LOG_NAME
 ```
 
 Set the log level, the default is info, can be debug can see more log information.
 
-设置日志级别，默认是info, 可以是debug可以看到更多的日志信息.
-
 ```shell
 MAYA_UMBRELLA_LOG_LEVEL
 ```
@@ -177,53 +142,36 @@ For example:
 Your file path is `c:/your/path/file.ma`.
 
 Then the backup file path is `c:/your/path/_virus/file.ma`.
-
-修改杀毒后文件的备份文件夹名称， 默认是`_virus`
-比如:
-你文件路径是  `c:/your/path/file.ma`
-那么备份文件路径是 `c:/your/path/_virus/file.ma`
 ```shell
 MAYA_UMBRELLA_BACKUP_FOLDER_NAME
 ```
 The default display language, including logging printouts, etc.
-
-is set by default according to your current maya interface language, 
-
+is set by default according to your current maya interface language,
 but of course we can also set it via the following environment variables.
-
-默认的显示语言，包含日志打印输出等，默认是根据你当前maya的界面语言来设置的，当然我们也可以通过下面的环境变量去设置.
 ```shell
 MAYA_UMBRELLA_LANG
 ```
 Ignore saving to the backup folder,
-*please note that if you are not clear about the consequences of this please do not modify it easily*, 
-the default batch antivirus will automatically back up the source file to the current file's backup folder 
+*please note that if you are not clear about the consequences of this please do not modify it easily*,
+the default batch antivirus will automatically back up the source file to the current file's backup folder
 after the batch antivirus.
 
-忽略保存到备份文件夹，*请注意，如果你不清楚这个会导致的后果请不要轻易修改*，默认批量杀毒后会把源文件自动备份到当前文件的备份文件夹.
-
 ```shell
 MAYA_UMBRELLA_IGNORE_BACKUP
 ```
 If ignored please set to
-
-如果忽略请设置为
 ```shell
 SET MAYA_UMBRELLA_IGNORE_BACKUP=true
 ```
 
-For the portable version of Maya, 
+For the portable version of Maya,
 you can specify the Maya path by adding the `MAYA_LOCATION` environment variable.
 
-如果是便携版Maya，可以通过添加 `MAYA_LOCATION` 环境变量指定Maya路径
-
 ```shell
 SET MAYA_LOCATION=d:/your/path/maya_version/
 ```
 You can also specify a directory from the command line.
 
-也可以通过命令行指定目录
-
 ```shell
 nox -s maya -- 2018 --install-root /your/local/maya/root
 
@@ -233,8 +181,6 @@ nox -s maya -- 2018 --install-root /your/local/maya/root
 
 Get virus files that have not been repaired in the current scenario.
 
-获取当前场景没有被修复的病毒文件
-
 ```python
 from maya_umbrella import MayaVirusDefender
 
@@ -243,8 +189,6 @@ print(api.get_unfixed_references())
 ```
 
 Batch repair of files, via regular expressions.
-
-批量修复文件, 通过正则表达式
 ```python
 from maya_umbrella import MayaVirusScanner
 
@@ -253,7 +197,7 @@ print(api.scan_files_from_pattern("your/path/*.m[ab]"))
 
 ```
 
-# 案例
+# Examples
 
 If you want to quickly go through maya standalone and batch clean up maya files.
 
@@ -266,14 +210,6 @@ For example, if you have `2018` installed locally, Then `nox -s maya -- 2018 --s
 
 The following syntax starts a maya-2020 environment to dynamically check for viruses from the `c:/test` folder.
 
-如果你想要快速通过maya standalone去批量清理maya文件，
-可以`下载`或者`git clone`当前`main`分支的工程，
-根据上面指引设置好开发环境,
-通过`nox`命令去启动maya `standalone`环境，maya版本根据你当前本地安装的maya为准，
-比如你本地安装了`2018`,
-那么就是 `nox -s maya -- 2018 --standalone`
-下面的语法是启动一个maya-2020的环境去动态从`c:/test`文件夹下去查杀病毒
-
 ```shell
 nox -s maya -- 2018 --standalone --pattern c:/test/*.m[ab]
 ```

maya_umbrella-0.15.0/maya_umbrella/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.15.0"

{maya_umbrella-0.14.2 → maya_umbrella-0.15.0}/maya_umbrella/signatures.py

@@ -9,16 +9,28 @@ virus20240430_sig1 = VirusSignature("virus20240430", "python(.*);.+exec.+(pyCode
 # https://regex101.com/r/2D14UA/1
 virus20240430_sig2 = VirusSignature("virus20240430", r"^\['.+']")
 
+# maya_secure_system virus signatures
+maya_secure_system_sig1 = VirusSignature("maya_secure_system", "import maya_secure_system")
+maya_secure_system_sig2 = VirusSignature("maya_secure_system", r"maya_secure_system\.MayaSecureSystem\(\)\.startup\(\)")
+
 JOB_SCRIPTS_VIRUS_SIGNATURES = [
     "petri_dish_path.+cmds.internalVar.+",
     "userSetup",
     "fuckVirus",
     virus20240430_sig1.signature,
     virus20240430_sig2.signature,
+    maya_secure_system_sig1.signature,
+    maya_secure_system_sig2.signature,
 ]
 
 FILE_VIRUS_SIGNATURES = [
     "import vaccine",
     "cmds.evalDeferred.*leukocyte.+",
     virus20240430_sig1.signature,
+    maya_secure_system_sig1.signature,
+]
+
+MAYA_SECURE_SYSTEM_VIRUS_SIGNATURES = [
+    maya_secure_system_sig1.signature,
+    maya_secure_system_sig2.signature,
 ]

maya_umbrella-0.15.0/maya_umbrella/vaccines/vaccine4.py

@@ -0,0 +1,55 @@
+# Import built-in modules
+import os
+
+# Import local modules
+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.signatures import MAYA_SECURE_SYSTEM_VIRUS_SIGNATURES
+from maya_umbrella.vaccine import AbstractVaccine
+
+
+class Vaccine(AbstractVaccine):
+    """A class for handling the maya_secure_system virus."""
+
+    virus_name = "maya_secure_system"
+
+    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
+            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, MAYA_SECURE_SYSTEM_VIRUS_SIGNATURES):
+                    self.report_issue(script_node)
+                    self.api.add_infected_node(script_node)
+
+    def collect_issues(self):
+        """Collect all issues related to the virus."""
+        # Add malicious files that need to be deleted
+        self.api.add_malicious_files(
+            [
+                os.path.join(self.api.local_script_path, "maya_secure_system.py"),
+                os.path.join(self.api.local_script_path, "maya_secure_system.pyc"),
+            ],
+        )
+        self.collect_infected_user_setup_py()
+        self.collect_infected_nodes()
+
+    def collect_infected_user_setup_py(self):
+        """Collect all bad userSetup.py files related to the virus."""
+        user_setup_py_files = [
+            os.path.join(self.api.local_script_path, "userSetup.py"),
+            os.path.join(self.api.user_script_path, "userSetup.py"),
+        ]
+
+        for user_setup_py in user_setup_py_files:
+            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-0.14.2 → maya_umbrella-0.15.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "maya_umbrella"
-version = "0.14.2"
+version = "0.15.0"
 description = "A better Autodesk Maya antivirus tool detects and removes malicious."
 homepage = "https://github.com/loonghao/maya_umbrella"
 repository = "https://github.com/loonghao/maya_umbrella"
@@ -36,7 +36,7 @@ nox = { version = "^2024.3.2", python = ">=3.8.1,<3.11" }
 
 [tool.commitizen]
 name = "cz_conventional_commits"
-version = "0.14.2"
+version = "0.15.0"
 tag_format = "v$version"
 version_files = [
     "pyproject.toml:version",

maya_umbrella-0.14.2/maya_umbrella/__version__.py

@@ -1 +0,0 @@
-__version__ = "0.14.2"