add database but need to test it

This commit is contained in:
Rainy21
2026-03-10 23:45:53 -04:00
parent dcf185ddfb
commit 46bc71f16a
6 changed files with 76 additions and 14 deletions
+18
View File
@@ -0,0 +1,18 @@
from pydantic import Secret
import sqlalchemy.orm import Session
from . import models
def create_url(db:Session, short_code:str , long_url:str):
db_url = models.URL(
short_code=short_code,
long_url=long_url
)
db.add(db_url)
db.commit()
db.refresh(db_url)
return db_url
def get_url(db:Session , short_code:str):
return db.query(models.URL).filter(
models.URL.short_code == short_code
).first()