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

pep8: scripts

Browse Source
This commit is contained in:
William Ballenthin
2020-07-01 12:42:33 -06:00
parent d23ef48bb6
commit 23e70b4e85
3 changed files with 182 additions and 176 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
scripts/capafmt.py
View File

@@ -1,11 +1,11 @@
'''
"""
Reformat the given capa rule into a consistent style.
Use the -i flag to update the rule in-place.
Usage:
$ python capafmt.py -i foo.yml
'''
"""
import sys
import logging
@@ -14,22 +14,24 @@ import argparse
import capa.rules
logger = logging.getLogger('capafmt')
logger = logging.getLogger("capafmt")
def main(argv=None):
if argv is None:
argv = sys.argv[1:]
parser = argparse.ArgumentParser(description='Capa rule formatter.')
parser.add_argument('path', type=str,
help='Path to rule to format')
parser.add_argument('-i', '--in-place', action='store_true', dest='in_place',
help='Format the rule in place, otherwise, write formatted rule to STDOUT')
parser.add_argument('-v', '--verbose', action='store_true',
help='Enable debug logging')
parser.add_argument('-q', '--quiet', action='store_true',
help='Disable all output but errors')
parser = argparse.ArgumentParser(description="Capa rule formatter.")
parser.add_argument("path", type=str, help="Path to rule to format")
parser.add_argument(
"-i",
"--in-place",
action="store_true",
dest="in_place",
help="Format the rule in place, otherwise, write formatted rule to STDOUT",
)
parser.add_argument("-v", "--verbose", action="store_true", help="Enable debug logging")
parser.add_argument("-q", "--quiet", action="store_true", help="Disable all output but errors")
args = parser.parse_args(args=argv)
if args.verbose:
@@ -40,17 +42,17 @@ def main(argv=None):
level = logging.INFO
logging.basicConfig(level=level)
logging.getLogger('capafmt').setLevel(level)
logging.getLogger("capafmt").setLevel(level)
rule = capa.rules.Rule.from_yaml_file(args.path)
if args.in_place:
with open(args.path, 'wb') as f:
f.write(rule.to_yaml().encode('utf-8'))
with open(args.path, "wb") as f:
f.write(rule.to_yaml().encode("utf-8"))
else:
print(rule.to_yaml().rstrip("\n"))
return 0
if __name__ == '__main__':
sys.exit(main())
if __name__ == "__main__":
sys.exit(main())