emojos/emojos.py

81 lines
2.8 KiB
Python

# Copyright (c) 2018 iliana weller
#
# This program is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option) any
# later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
# FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>.
import botocore.session
import operator
import requests
import urllib.parse
from flask import Flask, redirect, render_template, request, url_for
app = Flask(__name__)
@app.route('/<domain>')
def emojo(domain):
if request.args.get('filter_all','') == 'on':
filter_all = True
else:
filter_all = False
if request.args.get('filter_animated','') == 'on':
filter_animated = True
else:
filter_animated = False
try:
url = urllib.parse.urlunsplit(
('https', domain, '/api/v1/custom_emojis', '', ''))
if filter_all:
emojo = sorted(requests.get(url).json(),
key=operator.itemgetter('shortcode'))
else:
emojo = sorted(filter(lambda x: x.get('visible_in_picker', True),
requests.get(url).json()),
key=operator.itemgetter('shortcode'))
return render_template('emojo.html', domain=domain, emojo=emojo, filter_animated=filter_animated)
except requests.exceptions.RequestException as e:
return render_template('oh_no.html', domain=domain)
@app.route('/favicon.ico')
@app.route('/robots.txt')
def no_content():
return ('', 204)
@app.route('/code')
def code():
context = request.environ.get('lambda.context')
session = botocore.session.get_session()
# region name is detected from lambda environment
client = session.create_client('lambda')
code = client.get_function(FunctionName=context.function_name,
Qualifier=context.function_version)
return redirect(code['Code']['Location'], code=303)
@app.route('/', methods=('GET', 'POST'))
def index():
if request.method == 'POST':
if 'instance' in request.form:
filter_all = request.form.get('filter_all')
filter_animated = request.form.get('filter_animated')
return redirect(url_for('emojo', domain=request.form['instance'],
filter_all=filter_all, filter_animated=filter_animated))
else:
return redirect(url_for('index'))
else:
return render_template('index.html')