mirror of
https://github.com/Rainyy21/URL_shortener.git
synced 2026-10-11 00:30:13 -04:00
add database and test that it work
This commit is contained in:
+7
-11
@@ -1,18 +1,14 @@
|
||||
from pydantic import Secret
|
||||
import sqlalchemy.orm import Session
|
||||
from 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
|
||||
)
|
||||
|
||||
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()
|
||||
|
||||
def get_url(db: Session, short_code: str):
|
||||
return db.query(models.URL).filter(models.URL.short_code == short_code).first()
|
||||
|
||||
Reference in New Issue
Block a user