linklogger/app/func/link/delete.py

20 lines
584 B
Python

import sqlalchemy
from db import engine
"""
Delete the specified link from the users associated links
"""
def delete_link(link, owner):
with engine.begin() as conn:
try:
link_owner = conn.execute(sqlalchemy.text('SELECT owner FROM links WHERE link = :link'), [{'link': link}]).fetchone()[0]
except TypeError:
return 404
if owner == link_owner:
with engine.begin() as conn:
conn.execute(sqlalchemy.text('DELETE FROM links WHERE link = :link'), [{'link': link}])
return link
else:
return 401