mirror of
https://github.com/Rainyy21/URL_shortener.git
synced 2026-10-10 23:40:14 -04:00
Rainy21
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
from fastapi import FastAPI, HTTPException, Depends
|
|
from fastapi.responses import RedirectResponse
|
|
|
|
# the sqlalchemy
|
|
from sqlalchemy.orm import Session
|
|
import string
|
|
import random
|
|
|
|
from db.database import SessionLocal, engine
|
|
from db import models, crud, schemas
|
|
|
|
models.Base.metadata.create_call(bind=engine)
|
|
app = FastAPI()
|
|
|
|
|
|
def get_db():
|
|
db = SessionLocal()
|
|
try:
|
|
yield db
|
|
finally:
|
|
db.close()
|
|
|
|
|
|
def generate_short_code(length=6):
|
|
chars = string.ascii_letters + string.digits
|
|
return "".join(random.choice(chars) for _ in range(length))
|
|
|
|
|
|
@app.post("/shorten", response_model=schemas.URLResponse)
|
|
def shorten_url(request: schemas.URLCreate, db: Session = Depends(get_db)):
|
|
short_code = generate_short_code()
|
|
|
|
crud.create_url(db, short_code, request.long_url)
|
|
|
|
return {"short_url": f"http://localhost:8000/{short_code}"}
|
|
|
|
|
|
@app.get("/{short_code}")
|
|
def redirect_url(short_code: str, db: Session = Depends(get_db)):
|
|
db_url = crud.get_url(db, short_code)
|
|
|
|
if not db_url:
|
|
raise HTTPException(status_code=404, detail="URL not found")
|
|
return {"long_url": db_url.long_url}
|