PyCharm grand repo wide reformat

This commit is contained in:
Xevion
2020-03-08 20:21:18 -05:00
parent 6aeb41249c
commit 66c2ff228c
17 changed files with 277 additions and 187 deletions

View File

@@ -1,13 +1,16 @@
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
import requests
import xmltodict
from flask import request, render_template
from flask_login import current_user, login_required
from app import app, db
from app.custom import require_role
from app.models import Search
@app.route('/hidden/history')
@login_required
@require_role(roles=['Hidden', 'Admin'])
@@ -21,6 +24,7 @@ def hidden_history():
def hidden_help():
return render_template('hidden_help.html')
# Parses strings to test for "boolean-ness"
def boolparse(string, default=False):
trues = ['true', '1']
@@ -30,6 +34,7 @@ def boolparse(string, default=False):
return True
return False
@app.route('/hidden/')
@login_required
@require_role(roles=['Hidden'])
@@ -39,7 +44,8 @@ def hidden():
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'))
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)
@@ -50,7 +56,7 @@ def hidden():
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)
data = build_data(tags, page - 1, count, base64, showfull)
# Handling for limiters
if base64:
if showfull:
@@ -60,14 +66,18 @@ def hidden():
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)
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)
@@ -75,7 +85,7 @@ def build_data(tags, page, count, base64, showfull):
# XML Parsing & Data Building
parse = xmltodict.parse(response)
build = []
try:
parse['posts']['post']
except KeyError:
@@ -83,13 +93,13 @@ def build_data(tags, page, count, base64, showfull):
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'])
}
'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'])
@@ -97,4 +107,4 @@ def build_data(tags, page, count, base64, showfull):
temp['base64'] = base64ify(temp['real_url'])
build.append(temp)
return build
return build