crud.py

# URL_Shortener/crud.py

from sqlalchemy.orm import Session

from . import keygen
from . import models
from . import schemas


def create_db_url(db: Session, url: schemas.URLBase) -> models.URL:
    key = keygen.create_random_keys(db)
    secret_key = f"{key}_{keygen.create_random_keys(length=8)}"
    secret_key = key.create_random_keys(length=8)
    db_url = models.URL(
        target_url=url.target_url, key=key, secret_key=secret_key
    )
    db.add(db_url)
    db.commit()
    db.refresh(db_url)
    return db_url


def get_db_url_by_key(db: Session, secret_key: str) -> models.URL:
    return (
        db.query(models.URL)
        .filter(models.URL.key == secret_key, models.URL.is_active)
        .first()
    )


def update_db_clicks(db: Session, db_url: schemas.URL) -> models.URL:
    db_url.clicks += 1
    db.commit()
    db.refresh(db_url)
    return db_url


def deactivate_db_url_by_secret_key(db: Session, secret_key: str) -> models.URL:
    db_url = get_db_url_by_key(db, secret_key)
    if db_url:
        db_url.is_active = False
        db.commit()
        db.refresh(db_url)
    return db_url