mirror of
https://github.com/Xevion/v1.xevion.dev.git
synced 2025-12-06 15:16:59 -06:00
100 lines
3.5 KiB
Python
100 lines
3.5 KiB
Python
from app import app, db, login
|
|
from app.models import Search
|
|
from app.custom import require_role
|
|
from flask_login import current_user, login_user, logout_user, login_required
|
|
from flask import request, render_template
|
|
import requests
|
|
import xmltodict
|
|
import base64
|
|
import json
|
|
|
|
@app.route('/hidden/history')
|
|
@login_required
|
|
@require_role(roles=['Hidden', 'Admin'])
|
|
def hidden_history():
|
|
return render_template('hidden_history.html')
|
|
|
|
|
|
@app.route('/hidden/help')
|
|
@login_required
|
|
@require_role(roles=['Hidden'])
|
|
def hidden_help():
|
|
return render_template('hidden_help.html')
|
|
|
|
# Parses strings to test for "boolean-ness"
|
|
def boolparse(string, default=False):
|
|
trues = ['true', '1']
|
|
if string is None:
|
|
return default
|
|
elif string.lower() in trues:
|
|
return True
|
|
return False
|
|
|
|
@app.route('/hidden/')
|
|
@login_required
|
|
@require_role(roles=['Hidden'])
|
|
def hidden():
|
|
# Handled within request
|
|
tags = request.args.get('tags') or 'trap'
|
|
try:
|
|
page = int(request.args.get('page') or 1)
|
|
except (TypeError, ValueError):
|
|
return '\"page\" parameter must be Integer.<br>Invalid \"page\" parameter: \"{}\"'.format(request.args.get('page'))
|
|
# Handled within building
|
|
try:
|
|
count = int(request.args.get('count') or 50)
|
|
except (TypeError, ValueError):
|
|
return '\"count\" parameter must be Integer.<br>Invalid \"count\": \"{}\"'.format(request.args.get('count'))
|
|
base64 = boolparse(request.args.get('base64'))
|
|
# Handled within Jinja template
|
|
showfull = boolparse(request.args.get('showfull'))
|
|
showtags = boolparse(request.args.get('showtags'))
|
|
# Request, Parse & Build Data
|
|
data = build_data(tags, page-1, count, base64, showfull)
|
|
# Handling for limiters
|
|
if base64:
|
|
if showfull:
|
|
count = min(25, count)
|
|
else:
|
|
count = min(50, count)
|
|
search = Search(user_id=current_user.id, exact_url=str(request.url), query_args=json.dumps(request.args.to_dict()))
|
|
db.session.add(search)
|
|
db.session.commit()
|
|
return render_template('hidden.html', title='Gelbooru Browser', data=data, tags=tags, page=page, count=count, base64=base64, showfull=showfull, showtags=showtags)
|
|
|
|
def base64ify(url):
|
|
return base64.b64encode(requests.get(url).content).decode()
|
|
|
|
gelbooru_api_url = "https://gelbooru.com/index.php?page=dapi&s=post&q=index&tags={}&pid={}&limit={}"
|
|
gelbooru_view_url = "https://gelbooru.com/index.php?page=post&s=view&id={}"
|
|
|
|
def build_data(tags, page, count, base64, showfull):
|
|
# URL Building & Request
|
|
temp = gelbooru_api_url.format(tags, page, count)
|
|
response = requests.get(temp).text
|
|
# XML Parsing & Data Building
|
|
parse = xmltodict.parse(response)
|
|
build = []
|
|
|
|
try:
|
|
parse['posts']['post']
|
|
except KeyError:
|
|
return build
|
|
|
|
for index, element in enumerate(parse['posts']['post'][:count]):
|
|
temp = {
|
|
'index' : str(index + 1),
|
|
'real_url' : element['@file_url'],
|
|
'sample_url' : element['@preview_url'],
|
|
# strips tags, ensures no empty tags (may be unnecessary)
|
|
'tags' : list(filter(lambda tag : tag != '', [tag.strip() for tag in element['@tags'].split(' ')])),
|
|
'view' : gelbooru_view_url.format(element['@id'])
|
|
}
|
|
if base64:
|
|
if not showfull:
|
|
temp['base64'] = base64ify(temp['sample_url'])
|
|
else:
|
|
temp['base64'] = base64ify(temp['real_url'])
|
|
|
|
build.append(temp)
|
|
return build |