gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-22 07:10:29 -08:00
Code Issues Packages Projects Releases Wiki Activity

main: display spinner while generating viv workspace

Browse Source
This commit is contained in:
William Ballenthin
2020-08-14 09:38:08 -06:00
parent 6e525a93d7
commit de2c3c9800
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
capa/main.py
View File

@@ -18,6 +18,7 @@ import datetime
import textwrap import textwrap
import collections import collections
import halo
import tqdm import tqdm
import colorama import colorama
@@ -272,13 +273,14 @@ def get_workspace(path, format, should_save=True):
def get_extractor_py2(path, format): def get_extractor_py2(path, format):
import capa.features.extractors.viv import capa.features.extractors.viv
vw = get_workspace(path, format, should_save=False) with halo.Halo(text="analyzing program", spinner="simpleDots"):
vw = get_workspace(path, format, should_save=False)
try: try:
vw.saveWorkspace() vw.saveWorkspace()
except IOError: except IOError:
# see #168 for discussion around how to handle non-writable directories # see #168 for discussion around how to handle non-writable directories
logger.info("source directory is not writable, won't save intermediate workspace") logger.info("source directory is not writable, won't save intermediate workspace")
return capa.features.extractors.viv.VivisectFeatureExtractor(vw, path) return capa.features.extractors.viv.VivisectFeatureExtractor(vw, path)

3
setup.py
View File

@@ -11,7 +11,8 @@ import sys
import setuptools import setuptools
requirements = ["six", "tqdm", "pyyaml", "tabulate", "colorama", "termcolor", "ruamel.yaml", "wcwidth"] # halo==0.0.30 is the last version to support py2.7
requirements = ["six", "tqdm", "pyyaml", "tabulate", "colorama", "termcolor", "ruamel.yaml", "wcwidth", "halo==0.0.30"]
if sys.version_info >= (3, 0): if sys.version_info >= (3, 0):
# py3 # py3