Skip to content
Snippets Groups Projects
deletelink.py 972 B
Newer Older
  • Learn to ignore specific revisions
  • import sqlite3
    from flask import redirect, abort
    
    def deleteLink(request, s):
        try:
            userID = s.loads(request.cookies.get('userID')) #get the userid from the cookie
            loginbar = "Hello " + s.loads(request.cookies.get('username')) + ' (<a href="/user/logout" style="color:white">logout</a>)' # generate the login form
            loginbar = loginbar + "" #to hide the unused variable message
        except:
            return redirect("/user/login") # if user is not logged in redirect him to the login page
        linkToDelete = request.args.get('link') #get the link, which the user want's to delete from the parameter in the url.
    
        with sqlite3.connect('db/urls.db') as conn:
            cursor = conn.cursor()
            try:
                cursor.execute('DELETE FROM WEB_URL WHERE SHORT_URL=? AND USERNAME=?', [linkToDelete, userID]) #Delete the entrie
                return redirect('/user/links') #redirect the user back to the table.
            except:
                abort(500)