Merge pull request #64 from ipa-lab/restructure

Restructure

Author: andreashappe

.gitignore

@@ -7,3 +7,6 @@ __pycache__/
 *.sqlite3
 *.sqlite3-jounal
 *.sqlite
+src/hackingBuddyGPT.egg-info/
+build/
+dist/

pyproject.toml

@@ -0,0 +1,58 @@
+[build-system]
+requires = ["setuptools>=68"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "hackingBuddyGPT"
+authors = [
+	{ name = "Andreas Happe", email = "[email protected]" }
+]
+maintainers = [
+	{ name = "Andreas Happe", email = "[email protected]" },
+	{ name = "Juergen Cito", email = "[email protected]" }
+]
+description = "Helping Ethical Hackers use LLMs in 50 lines of code"
+readme = "README.md"
+keywords = ["hacking", "pen-testing", "LLM", "AI", "agent"]
+requires-python = ">=3.8"
+version = "0.3.0"
+license = { file = "LICENSE" }
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+	"Development Status :: 4 - Beta",
+]
+dependencies = [
+	'fabric == 3.2.2',
+	'Mako == 1.3.2',
+	'requests == 2.32.0',
+	'rich == 13.7.1',
+	'tiktoken == 0.6.0',
+	'instructor == 1.2.2',
+	'PyYAML == 6.0.1',
+	'python-dotenv == 1.0.1',
+	'pypsexec == 0.3.0'
+]
+
+[project.urls]
+Homepage = "https://www.hackingbuddy.ai"
+Documentation = "https://docs.hackingbuddy.ai"
+Repository = "https://github.com/ipa-lab/hackingBuddyGPT"
+"Bug Tracker" = "https://github.com/ipa-lab/hackingBuddyGPT/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.package-data]
+"hackingBuddyGPT.usecases.privesc.templates" = ["*.txt"]
+"hackingBuddyGPT.usecases.minimal" = ["*.txt"]
+
+[tool.pytest.ini_options]
+pythonpath = "src"
+addopts = [
+	"--import-mode=importlib",
+]
+
+[project.scripts]
+wintermute = "hackingBuddyGPT.cli.wintermute:main"

requirements.txt

@@ -1,11 +1,9 @@
 import base64
 from dataclasses import dataclass
-from typing import Literal, Optional, Dict
-
 import requests
+from typing import Literal, Optional, Dict
 
-from capabilities import Capability
-
+from . import Capability
 
 @dataclass
 class HTTPRequest(Capability):

src/hackingBuddyGPT/__init__.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass
 from typing import Tuple
 
-from utils import PSExecConnection
+from hackingBuddyGPT.utils import PSExecConnection
 from .capability import Capability
 
 

capabilities/__init__.py renamed to src/hackingBuddyGPT/capabilities/__init__.py

@@ -2,7 +2,7 @@
 from dataclasses import dataclass
 from typing import Tuple
 
-from utils import PSExecConnection
+from hackingBuddyGPT.utils import PSExecConnection
 from .capability import Capability
 
 

capabilities/capability.py renamed to src/hackingBuddyGPT/capabilities/capability.py

@@ -1,7 +1,7 @@
 from dataclasses import dataclass, field
 from typing import Tuple, List
 
-from capabilities import Capability
+from . import Capability
 
 
 @dataclass