From 93f9ef792cd68180f5274e09cce0a055bc607f1b Mon Sep 17 00:00:00 2001 From: Ajay Dangi Date: Thu, 11 Jan 2024 21:21:18 +0530 Subject: [PATCH] Added link customization --- src/link_liberate/main.py | 10 ++++++---- src/link_liberate/templates/liberate.html | 1 + 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/src/link_liberate/main.py b/src/link_liberate/main.py index 4d98346..62cdbcd 100644 --- a/src/link_liberate/main.py +++ b/src/link_liberate/main.py @@ -54,13 +54,15 @@ async def web(request: Request) -> Response: @app.post("/liberate", response_class=HTMLResponse) @limiter.limit("100/minute") async def web_post( - request: Request, content: Annotated[str, Form()], db: Session = Depends(get_db) + request: Request, content: Annotated[str, Form()], customString: Annotated[str, Form()] = None, db: Session = Depends(get_db) ) -> PlainTextResponse: try: link: str = make_proper_url(content) - uuid: str = generate_uuid() - if db.query(LiberatedLink).filter(LiberatedLink.uuid == uuid).first(): - uuid = generate_uuid() + uuid: str = customString + if uuid is None: + uuid: str = generate_uuid() + if db.query(LiberatedLink).filter(LiberatedLink.uuid == uuid).first(): + uuid = generate_uuid() new_liberated_link = LiberatedLink(uuid=uuid, link=link) db.add(new_liberated_link) db.commit() diff --git a/src/link_liberate/templates/liberate.html b/src/link_liberate/templates/liberate.html index 60419af..bd638a5 100644 --- a/src/link_liberate/templates/liberate.html +++ b/src/link_liberate/templates/liberate.html @@ -22,6 +22,7 @@

🔗LinkLiberate🪽

+