mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
add new init command for generate boilerplate in flat or src layout
This commit is contained in:
@@ -1,12 +1,12 @@
|
|||||||
from typer import Typer
|
from typer import Typer
|
||||||
|
|
||||||
from .commands import run_handler
|
from .commands import run_handler, init_handler
|
||||||
|
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
app = Typer()
|
app = Typer()
|
||||||
app.command("run")(run_handler)
|
app.command("run")(run_handler)
|
||||||
app.command("init")(run_handler)
|
app.command("init")(init_handler)
|
||||||
app()
|
app()
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
|||||||
@@ -1 +1,2 @@
|
|||||||
from .run import run_handler
|
from .run import run_handler as run_handler
|
||||||
|
from .init import init_handler as init_handler
|
||||||
@@ -0,0 +1,108 @@
|
|||||||
|
__all__ = ["init_handler"]
|
||||||
|
|
||||||
|
from pathlib import Path
|
||||||
|
from typing import Literal
|
||||||
|
|
||||||
|
GITIGNORE_CONTENT = """
|
||||||
|
__pycache__/
|
||||||
|
*.py[cod]
|
||||||
|
.env
|
||||||
|
.venv/
|
||||||
|
env/
|
||||||
|
"""
|
||||||
|
|
||||||
|
FLAT_MAIN_TEMPLATE = """
|
||||||
|
from argenta import Orchestrator, App
|
||||||
|
|
||||||
|
from handlers import router
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
app = App()
|
||||||
|
app.include_router(router)
|
||||||
|
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
"""
|
||||||
|
|
||||||
|
FLAT_HANDLERS_TEMPLATE = """
|
||||||
|
from argenta import Router, Response
|
||||||
|
|
||||||
|
router = Router("Hello command")
|
||||||
|
|
||||||
|
@router.command("hello")
|
||||||
|
def start_handler(response: Response):
|
||||||
|
print("Hello world!")
|
||||||
|
"""
|
||||||
|
|
||||||
|
SRC_MAIN_TEMPLATE = """
|
||||||
|
from argenta import Orchestrator, App
|
||||||
|
|
||||||
|
from .routers import router
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
app = App()
|
||||||
|
app.include_router(router)
|
||||||
|
|
||||||
|
orchestrator = Orchestrator()
|
||||||
|
orchestrator.start_polling(app)
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
|
"""
|
||||||
|
|
||||||
|
SRC_ROUTERS_TEMPLATE = """
|
||||||
|
from argenta import Router
|
||||||
|
from .handlers.hello_world_handler import hello_handler
|
||||||
|
|
||||||
|
router = Router()
|
||||||
|
|
||||||
|
router.command('hello')(hello_handler)
|
||||||
|
"""
|
||||||
|
|
||||||
|
SRC_HANDLER_TEMPLATE = """
|
||||||
|
from argenta import Response
|
||||||
|
|
||||||
|
|
||||||
|
def hello_handler(response: Response) -> None:
|
||||||
|
print("Hello world!")
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
|
||||||
|
def create_file(path: Path, content: str) -> None:
|
||||||
|
if not path.exists():
|
||||||
|
path.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
path.write_text(content.strip(), encoding="utf-8")
|
||||||
|
print(f"Created: {path}")
|
||||||
|
else:
|
||||||
|
print(f"Skipped: {path} (already exists)")
|
||||||
|
|
||||||
|
|
||||||
|
def init_handler(with_arch: Literal["flat", "src"] = "flat") -> None:
|
||||||
|
cwd = Path.cwd()
|
||||||
|
project_name = cwd.name.lower().replace(" ", "_")
|
||||||
|
|
||||||
|
create_file(cwd / ".gitignore", GITIGNORE_CONTENT)
|
||||||
|
|
||||||
|
if with_arch == "flat":
|
||||||
|
create_file(cwd / "main.py", FLAT_MAIN_TEMPLATE)
|
||||||
|
create_file(cwd / "handlers.py", FLAT_HANDLERS_TEMPLATE)
|
||||||
|
|
||||||
|
elif with_arch == "src":
|
||||||
|
base_pkg = cwd / "src" / project_name / "application"
|
||||||
|
|
||||||
|
create_file(base_pkg / "__main__.py", SRC_MAIN_TEMPLATE)
|
||||||
|
create_file(base_pkg / "routers.py", SRC_ROUTERS_TEMPLATE)
|
||||||
|
create_file(base_pkg / "handlers" / "hello_world_handler.py", SRC_HANDLER_TEMPLATE)
|
||||||
|
|
||||||
|
create_file(cwd / "src" / "__init__.py", "")
|
||||||
|
create_file(cwd / "src" / project_name / "__init__.py", "")
|
||||||
|
create_file(base_pkg / "__init__.py", "")
|
||||||
|
create_file(base_pkg / "handlers" / "__init__.py", "")
|
||||||
|
|
||||||
|
print("\nInitialization complete.")
|
||||||
@@ -7,7 +7,7 @@ __all__ = [
|
|||||||
"HandlerFunc",
|
"HandlerFunc",
|
||||||
]
|
]
|
||||||
|
|
||||||
from typing import Any, Protocol, TypeVar
|
from typing import Any, Protocol, TypeVar, Callable
|
||||||
|
|
||||||
from argenta.response import Response
|
from argenta.response import Response
|
||||||
|
|
||||||
@@ -39,6 +39,4 @@ class DescriptionMessageGenerator(Protocol):
|
|||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
|
|
||||||
class HandlerFunc(Protocol):
|
type HandlerFunc = Callable[..., Any]
|
||||||
def __call__(self, response: Response, /, *args: Any, **kwargs: Any) -> None:
|
|
||||||
raise NotImplementedError
|
|
||||||
|
|||||||
Reference in New Issue
Block a user