Skip to content

Commit d0d4b3d

Browse files
author
r0BIT
committed
refactor: organize folder structure and add missing __init__.py files
- Add __init__.py to parsers/, smb/, utils/ packages - Move engine_async.py to engine/async_runner.py - Update imports in cli.py, test_engine_async.py, engine/__init__.py - Delete empty connectors/tests/ directory - All 223 tests passing
1 parent 9251ab5 commit d0d4b3d

File tree

7 files changed

+39
-5
lines changed

7 files changed

+39
-5
lines changed

taskhound/cli.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
from .auth import AuthContext
77
from .config import build_parser, validate_args
88
from .engine import process_offline_directory, process_target
9-
from .engine_async import (
9+
from .engine.async_runner import (
1010
AsyncConfig,
1111
AsyncTaskHound,
1212
aggregate_results,

taskhound/engine/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
# these at the engine module level (e.g., @patch("taskhound.engine.smb_connect"))
88
from ..smb.connection import smb_connect
99
from ..smb.tasks import crawl_tasks
10+
from .async_runner import AsyncConfig, AsyncTaskHound, TargetResult, aggregate_results
1011
from .helpers import sort_tasks_by_priority
1112
from .offline import process_offline_directory
1213
from .online import process_target
@@ -17,4 +18,9 @@
1718
"sort_tasks_by_priority",
1819
"smb_connect",
1920
"crawl_tasks",
21+
# Async/parallel processing
22+
"AsyncConfig",
23+
"AsyncTaskHound",
24+
"TargetResult",
25+
"aggregate_results",
2026
]
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,9 +28,9 @@
2828
TimeRemainingColumn,
2929
)
3030

31-
from .laps import LAPSFailure
32-
from .models.task import TaskRow
33-
from .utils.console import (
31+
from ..laps import LAPSFailure
32+
from ..models.task import TaskRow
33+
from ..utils.console import (
3434
console,
3535
good,
3636
info,

taskhound/parsers/__init__.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
"""Parsers for TaskHound.
2+
3+
Modules:
4+
highvalue: High-value target identification and password analysis
5+
task_xml: Windows Task Scheduler XML parsing
6+
"""

taskhound/smb/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
"""SMB connectivity and task enumeration for TaskHound.
2+
3+
Modules:
4+
connection: SMB connection management
5+
credguard: Credential Guard detection
6+
task_rpc: Task Scheduler RPC operations
7+
tasks: Task enumeration and crawling
8+
"""

taskhound/utils/__init__.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
"""Utility modules for TaskHound.
2+
3+
Modules:
4+
bh_api: BloodHound API client
5+
bh_auth: BloodHound authentication
6+
cache_manager: Session and result caching
7+
console: Rich console output
8+
date_parser: Date/time parsing utilities
9+
helpers: General helper functions
10+
ldap: LDAP connection utilities
11+
logging: Logging configuration
12+
network: Network utilities
13+
sid_resolver: SID resolution utilities
14+
"""

tests/test_engine_async.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import time
55
from typing import List, Optional, Tuple
66

7-
from taskhound.engine_async import (
7+
from taskhound.engine.async_runner import (
88
AsyncConfig,
99
AsyncTaskHound,
1010
TargetResult,

0 commit comments

Comments
 (0)