From 9ba4c4eee41b21842ae54d09763c47753134ef17 Mon Sep 17 00:00:00 2001
From: Jonas Leder <jonas@jonasled.de>
Date: Sat, 9 Nov 2019 18:25:26 +0100
Subject: [PATCH] added delete function
---
main.py | 27 ++++++++++++++++++++++++---
1 file changed, 24 insertions(+), 3 deletions(-)
diff --git a/main.py b/main.py
index e1dbb96..6a1a900 100644
--- a/main.py
+++ b/main.py
@@ -135,7 +135,7 @@ def grecaptcha_verify(request): #This function is used to verify the google reca
@app.route('/', methods=['GET'])
def home_get():
try:
- loginbar = "Hello " + request.cookies.get('username') + ' (<a href="/user/logout" style="color:white">logout</a>)'
+ loginbar = "Hello " + request.cookies.get('username') + ' (<a href="/user/links" style="color:white">your links</a>, <a href="/user/logout" style="color:white">logout</a>)'
except:
loginbar = '<a href="/user/login" style="color:white">login</a>'
@@ -149,7 +149,7 @@ def home_post():
try:
userID = request.cookies.get('userID')
- loginbar = "Hello " + request.cookies.get('username') + ' (<a href="/user/logout" style="color:white">logout</a>)'
+ loginbar = "Hello " + request.cookies.get('username') + ' (<a href="/user/links" style="color:white">your links</a>, <a href="/user/logout" style="color:white">logout</a>)'
except:
userID = "null"
loginbar = '<a href="/user/login" style="color:white">login</a>'
@@ -248,13 +248,34 @@ def ownLinks():
res = cursor.execute('SELECT LONG_URL, SHORT_URL FROM WEB_URL WHERE USERNAME=?', [userID])
response = '<table id="t01">\n<tr>\n<th>Long URL</th>\n<th>Short URL</th>\n<th>Action</th>\n</tr>\n'
try:
- for entries in res.fetchall():
+ entriesList = res.fetchall()
+ for entries in entriesList:
response = response + "<tr>\n<td>" + entries[0] + "</td>\n<td>" + entries[1] + '</td>\n<td><a id="red" href="/user/delete?link=' + escape(entries[1]) + '">delete</a></tr>\n'
+
+ if(len(entriesList) == 0): response = 'you have no shorten links. <a href="/">back</a>'
except:
abort(500)
response = response + "</table>"
return render_template('editEntries.html', content=response, loginbar=loginbar)
+
+@app.route('/user/delete')
+def delete():
+ try:
+ userID = request.cookies.get('userID')
+ loginbar = "Hello " + request.cookies.get('username') + ' (<a href="/user/logout" style="color:white">logout</a>)'
+ except:
+ return redirect("/user/login")
+ linkToDelete = request.args.get('link')
+
+ with sqlite3.connect('db/urls.db') as conn: #Get the original URL from the database
+ cursor = conn.cursor()
+ try:
+ cursor.execute('DELETE FROM WEB_URL WHERE SHORT_URL=? AND USERNAME=?', [linkToDelete, userID])
+ return redirect('/user/links')
+ except:
+ abort(500)
+
if __name__ == '__main__':
table_check()# This code checks whether database table is created or not
--
GitLab