revi-archive/xygt
1
0
Fork 0
mirror of https://github.com/jackeilles/xygt.git synced 2024-11-23 23:32:38 +01:00
Code Issues Projects Releases Packages Wiki Activity

idk when but i broke url

Browse source
This commit is contained in:
Jack Eilles 2023-12-22 21:16:10 +00:00
parent c3214af880
commit b985ba69d6
2 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
app/routes.py
View file

@ -1,6 +1,6 @@
from app import app, worker
from config import Config, Errors
from flask import render_template, request, send_file
from flask import render_template, request, send_file, redirect
from werkzeug.datastructures import FileStorage
from werkzeug.utils import secure_filename
from io import BytesIO
@ -55,6 +55,9 @@ def index():
return result, status
elif 'url' in request.form:
url = request.form['url']
result, status = worker.shortURL(url, ip, userid, id, retention)
@app.route('/<id>')

4
app/worker.py
View file

@ -98,11 +98,11 @@ def shortenURL(url, ip, userid, id, retention):
def idInfo(id):
# Check files and url for the ID
if Config.files.find_one({"id": id}) is not None:
check = Config.files.find_one({"id": id}, {'_id': False}, {"ip": False})
check = Config.files.find_one({"id": id}, {'_id': False, "ip": False})
# "ip": False removes the IP from the returned data.
# If it's not there then check url
elif Config.url.find_one({"id": id}) is not None:
check = Config.url.find_one({"id": id}, {'_id': False}, {"ip": False})
check = Config.url.find_one({"id": id}, {'_id': False, "ip": False})
# Return the mongodb info about the file, removing IP if its present
return check