diff --git a/app/func/del_link_records.py b/app/func/del_link_records.py new file mode 100644 index 0000000..fbbce0e --- /dev/null +++ b/app/func/del_link_records.py @@ -0,0 +1,20 @@ +import sqlalchemy + +from db import engine + +""" +Delete all of the IP log records that are associated with a specific link +""" +def del_link_records(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 'Link does not exist', 200 + + if owner == link_owner: + with engine.begin() as conn: + conn.execute(sqlalchemy.text('DELETE FROM records WHERE link = :link'), [{'link': link}]) + return 'Link records have been deleted', 200 + else: + return 'You are not the owner of this link', 401 \ No newline at end of file diff --git a/app/routes.py b/app/routes.py index 4a181cb..b05a778 100644 --- a/app/routes.py +++ b/app/routes.py @@ -10,6 +10,7 @@ from func.log import log from func.delete_link import delete_link from func.renew_link import renew_link from func.link_records import link_records +from func.del_link_records import del_link_records app = flask.Flask(__name__) @@ -88,4 +89,10 @@ def records_link(link): if response[0].startswith('Timestamp'): return response[0], response[1] else: - return flask.jsonify(msg=response[0]), response[1] \ No newline at end of file + return flask.jsonify(msg=response[0]), response[1] + + +@app.route('//records', methods=['POST']) +def records_delete(link): + response = del_link_records(link, auth.current_user()) + return flask.jsonify(msg=response[0]), response[1] \ No newline at end of file