-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
60 lines (49 loc) · 2.01 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from flask import Flask, render_template, request, redirect, url_for, flash, abort, session, jsonify
import json
import os.path
from werkzeug.utils import secure_filename
app = Flask(__name__)
app.secret_key = 'h432hi5ohi3h5i5hi3o2hi'
@app.route('/')
def home():
return render_template('home.html', codes=session.keys())
@app.route('/your-url', methods=['GET','POST'])
def your_url():
if request.method == 'POST':
urls = {}
if os.path.exists('urls.json'):
with open('urls.json') as urls_file:
urls = json.load(urls_file)
if request.form['code'] in urls.keys():
flash('That short name has already been taken. Please select another name.')
return redirect(url_for('home'))
if 'url' in request.form.keys():
urls[request.form['code']] = {'url':request.form['url']}
else:
f = request.files['file']
full_name = request.form['code'] + secure_filename(f.filename)
f.save('C:/Users/prabh/pybox/url-shortener/static/user_files/' + full_name)
urls[request.form['code']] = {'file':full_name}
with open('urls.json','w') as url_file:
json.dump(urls, url_file)
session[request.form['code']] = True
return render_template('your_url.html', code=request.form['code'])
else:
return redirect(url_for('home'))
@app.route('/<string:code>')
def redirect_to_url(code):
if os.path.exists('urls.json'):
with open('urls.json') as urls_file:
urls = json.load(urls_file)
if code in urls.keys():
if 'url' in urls[code].keys():
return redirect(urls[code]['url'])
else:
return redirect(url_for('static', filename='user_files/' + urls[code]['file']))
return abort(404)
@app.errorhandler(404)
def page_not_found(error):
return render_template('page_not_found.html'), 404
@app.route('/api')
def session_api():
return jsonify(list(session.keys()))