diff --git a/httpz-scanner/__init__.py b/httpz-scanner/__init__.py index 452b9d1..242571b 100644 --- a/httpz-scanner/__init__.py +++ b/httpz-scanner/__init__.py @@ -6,4 +6,4 @@ from .scanner import HTTPZScanner from .colors import Colors -__version__ = '1.0.4' \ No newline at end of file +__version__ = '1.0.5' \ No newline at end of file diff --git a/httpz/__main__.py b/httpz/__main__.py index 0519ecb..8f68508 100644 --- a/httpz/__main__.py +++ b/httpz/__main__.py @@ -1 +1,18 @@ - \ No newline at end of file +#!/usr/bin/env python3 +# HTTPZ Web Scanner - Developed by acidvegas in Python (https://github.com/acidvegas/httpz) +# httpz/__main__.py + +import asyncio +import sys +from .cli import main + +def cli(): + """Entry point for the command line interface""" + try: + asyncio.run(main()) + except KeyboardInterrupt: + sys.exit(1) + +# This allows both 'python -m httpz' and the 'httpz' command to work +if __name__ == '__main__': + cli() \ No newline at end of file diff --git a/setup.py b/setup.py index ec7252e..19a3010 100644 --- a/setup.py +++ b/setup.py @@ -9,7 +9,7 @@ with open('README.md', 'r', encoding='utf-8') as f: setup( name='httpz-scanner', - version='1.0.4', + version='1.0.5', author='acidvegas', author_email='acid.vegas@acid.vegas', description='Hyper-fast HTTP Scraping Tool', @@ -41,7 +41,7 @@ setup( ], entry_points={ 'console_scripts': [ - 'httpz=httpz.__main__:main', # This creates the 'httpz' command + 'httpz=httpz.__main__:cli', # This creates the 'httpz' command ], }, ) \ No newline at end of file