Skip to content

Commit

Permalink
fix ruff
Browse files Browse the repository at this point in the history
  • Loading branch information
mr-tz committed Jun 19, 2024
1 parent 740c739 commit 0c9d3d0
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 4 deletions.
2 changes: 1 addition & 1 deletion capa/features/extractors/vmray/call.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from capa.features.insn import API, Number
from capa.features.common import String, Feature
from capa.features.address import Address
from capa.features.extractors.vmray.models import Analysis, FunctionCall
from capa.features.extractors.vmray.models import Analysis
from capa.features.extractors.base_extractor import CallHandle, ThreadHandle, ProcessHandle

logger = logging.getLogger(__name__)
Expand Down
3 changes: 2 additions & 1 deletion capa/features/extractors/vmray/extractor.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,7 +99,7 @@ def from_zipfile(cls, zipfile_path: Path):
flog_xml = zipfile.read("logs/flog.xml", pwd=b"infected")
flog = Analysis.from_xml(flog_xml)

# debug(flog)
debug(flog.processes[1])
pprint(flog.processes[0])

return cls(VMRayAnalysis(sv2, flog))
Expand All @@ -108,5 +108,6 @@ def from_zipfile(cls, zipfile_path: Path):
if __name__ == "__main__":
# TODO(mr): for testing, removeme
import sys

input_path = Path(sys.argv[1])
VMRayExtractor.from_zipfile(input_path)
4 changes: 2 additions & 2 deletions capa/features/extractors/vmray/models.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,9 @@
# is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and limitations under the License.

from typing import Any, Dict, List, Union, Literal, Optional
from typing import Dict, List, Optional

from pydantic import Field, BaseModel
from pydantic import BaseModel
from pydantic_xml import BaseXmlModel, attr, element


Expand Down

0 comments on commit 0c9d3d0

Please sign in to comment.