diff --git a/setup.py b/setup.py index 9485b71..de2df60 100644 --- a/setup.py +++ b/setup.py @@ -3,7 +3,7 @@ setup( name='xanalyzer', - version='0.2.1', + version=open('xanalyzer/VERSION', 'r').read(), packages=find_packages(), package_data={ 'xanalyzer': [ diff --git a/xanalyzer/VERSION b/xanalyzer/VERSION new file mode 100644 index 0000000..7dff5b8 --- /dev/null +++ b/xanalyzer/VERSION @@ -0,0 +1 @@ +0.2.1 \ No newline at end of file diff --git a/xanalyzer/config.py b/xanalyzer/config.py index 0905b91..e9aac0d 100644 --- a/xanalyzer/config.py +++ b/xanalyzer/config.py @@ -10,6 +10,7 @@ class Config: # UserDB.TXT文件默认是GBK编码, 需要简单处理一下, 去掉乱码, 新的UserDB.TXT加了新的规则, peutils解析不了 peid_signature_path = home_dir / 'data' / 'UserDB.TXT' tools_info_path = home_dir / 'data' / 'tools_info.json' + VERSION = open(home_dir / 'VERSION', 'r').read().strip() conf = {} diff --git a/xanalyzer/main.py b/xanalyzer/main.py index 1b05096..8234869 100644 --- a/xanalyzer/main.py +++ b/xanalyzer/main.py @@ -18,9 +18,14 @@ def main(): group = parser.add_mutually_exclusive_group(required=True) group.add_argument('-f', '--file', nargs='+', help='analyze one or more files, can be a folder path') group.add_argument('-u', '--url', help='analyze the url') + group.add_argument('--version', action='store_true', help='print version info') parser.add_argument('-s', '--save', action='store_true', help='save log and data') args = parser.parse_args() + if args.version: + print(Config.VERSION) + return + Config.init(args.save) init_log()