import os import json from flask import Flask, render_template, send_from_directory, request, \ url_for, abort, g, redirect, jsonify, session, flash from flask_login import login_required from fatcat_web import app, api, auth_api from fatcat_web.auth import handle_token_login, handle_logout, load_user, handle_ia_xauth from fatcat_client.rest import ApiException from fatcat_web.search import do_search ### Views ################################################################### @app.route('/container//history', methods=['GET']) def container_history(ident): try: entity = api.get_container(ident) history = api.get_container_history(ident) except ApiException as ae: abort(ae.status) #print(history) return render_template('entity_history.html', page_title=entity.name, entity_type="container", entity=entity, history=history) @app.route('/container//edit', methods=['GET']) def container_edit_view(ident): try: entity = api.get_container(ident) except ApiException as ae: abort(ae.status) return render_template('entity_edit.html') #@app.route('/container//edit', methods=['POST']) #def container_edit(ident): # raise NotImplemented() # params = dict() # for k in request.form: # if k.startswith('container_'): # params[k[10:]] = request.form[k] # edit = api.update_container(params=params) # return redirect("/container/{}".format(edit.ident)) # # else: # #return render_template('container_edit.html') @app.route('/container/create', methods=['GET']) def container_create_view(): return render_template('container_create.html') @app.route('/container/create', methods=['POST']) def container_create(): raise NotImplementedError params = dict() for k in request.form: if k.startswith('container_'): params[k[10:]] = request.form[k] container = None edit = api.create_container(container, params=params) return redirect("/container/{}".format(edit.ident)) @app.route('/container/lookup', methods=['GET']) def container_lookup(): extid = None for key in ('issnl', 'wikidata_qid'): if request.args.get(key): extid = key break if extid is None: abort(400) try: resp = api.lookup_container(**{extid: request.args.get(extid)}) except ApiException as ae: abort(ae.status) return redirect('/container/{}'.format(resp.ident)) @app.route('/container/', methods=['GET']) def container_view(ident): try: entity = api.get_container(ident) except ApiException as ae: abort(ae.status) if entity.state == "redirect": return redirect('/container/{}'.format(entity.redirect)) if entity.state == "deleted": return render_template('deleted_entity.html', entity=entity) return render_template('container_view.html', container=entity) @app.route('/creator//history', methods=['GET']) def creator_history(ident): try: entity = api.get_creator(ident) history = api.get_creator_history(ident) except ApiException as ae: abort(ae.status) return render_template('entity_history.html', page_title=entity.display_name, entity_type="creator", entity=entity, history=history) @app.route('/creator//edit', methods=['GET']) def creator_edit_view(ident): try: entity = api.get_creator(ident) except ApiException as ae: abort(ae.status) return render_template('entity_edit.html') @app.route('/creator/lookup', methods=['GET']) def creator_lookup(): for key in ('orcid', 'wikidata_qid'): if request.args.get(key): extid = key break if extid is None: abort(400) try: resp = api.lookup_creator(**{extid: request.args.get(extid)}) except ApiException as ae: abort(ae.status) return redirect('/creator/{}'.format(resp.ident)) @app.route('/creator/', methods=['GET']) def creator_view(ident): try: entity = api.get_creator(ident) releases = api.get_creator_releases(ident) except ApiException as ae: abort(ae.status) if entity.state == "redirect": return redirect('/creator/{}'.format(entity.redirect)) if entity.state == "deleted": return render_template('deleted_entity.html', entity=entity) return render_template('creator_view.html', creator=entity, releases=releases) @app.route('/file//history', methods=['GET']) def file_history(ident): try: entity = api.get_file(ident) history = api.get_file_history(ident) except ApiException as ae: abort(ae.status) return render_template('entity_history.html', page_title=None, entity_type="file", entity=entity, history=history) @app.route('/file//edit', methods=['GET']) def file_edit_view(ident): try: entity = api.get_file(ident) except ApiException as ae: abort(ae.status) return render_template('entity_edit.html') @app.route('/file/lookup', methods=['GET']) def file_lookup(): for key in ('md5', 'sha1', 'sha256'): if request.args.get(key): extid = key break if extid is None: abort(400) try: resp = api.lookup_file(**{extid: request.args.get(extid)}) except ApiException as ae: abort(ae.status) return redirect('/file/{}'.format(resp.ident)) @app.route('/file/', methods=['GET']) def file_view(ident): try: entity = api.get_file(ident) except ApiException as ae: abort(ae.status) if entity.state == "redirect": return redirect('/file/{}'.format(entity.redirect)) if entity.state == "deleted": return render_template('deleted_entity.html', entity=entity) return render_template('file_view.html', file=entity) @app.route('/release/lookup', methods=['GET']) def release_lookup(): for key in ('doi', 'wikidata_qid', 'pmid', 'pmcid', 'isbn13', 'core_id'): if request.args.get(key): extid = key break if extid is None: abort(400) try: resp = api.lookup_release(**{extid: request.args.get(extid)}) except ApiException as ae: abort(ae.status) return redirect('/release/{}'.format(resp.ident)) @app.route('/release/create', methods=['GET']) def release_create_view(): return render_template('release_create.html') @app.route('/release/create', methods=['POST']) def release_create(): raise NotImplementedError params = dict() for k in request.form: if k.startswith('release_'): params[k[10:]] = request.form[k] release = None edit = api.create_release(release, params=params) return redirect("/release/{}".format(edit.ident)) @app.route('/release//history', methods=['GET']) def release_history(ident): try: entity = api.get_release(ident) history = api.get_release_history(ident) except ApiException as ae: abort(ae.status) return render_template('entity_history.html', page_title=entity.title, entity_type="release", entity=entity, history=history) @app.route('/release//edit', methods=['GET']) def release_edit_view(ident): try: entity = api.get_release(ident) except ApiException as ae: abort(ae.status) return render_template('entity_edit.html') @app.route('/release/', methods=['GET']) def release_view(ident): try: entity = api.get_release(ident) files = api.get_release_files(ident) container = None if entity.container_id is not None: container = api.get_container(entity.container_id) except ApiException as ae: abort(ae.status) if entity.state == "redirect": return redirect('/release/{}'.format(entity.redirect)) if entity.state == "deleted": return render_template('deleted_entity.html', entity=entity) authors = [c for c in entity.contribs if c.role in ('author', None)] authors = sorted(authors, key=lambda c: c.index) for fe in files: # crudely filter out exact duplicates kept = [] for u in fe.urls: if not u in kept: kept.append(u) fe.urls = [u for u in kept if not '/web/None/' in u.url] return render_template('release_view.html', release=entity, authors=authors, files=files, container=container) @app.route('/work/create', methods=['GET']) def work_create_view(): return abort(404) @app.route('/work//history', methods=['GET']) def work_history(ident): try: entity = api.get_work(ident) history = api.get_work_history(ident) except ApiException as ae: abort(ae.status) return render_template('entity_history.html', page_title=None, entity_type="work", entity=entity, history=history) @app.route('/work//edit', methods=['GET']) def work_edit_view(ident): try: entity = api.get_work(ident) except ApiException as ae: abort(ae.status) return render_template('entity_edit.html') @app.route('/work/', methods=['GET']) def work_view(ident): try: entity = api.get_work(ident) releases = api.get_work_releases(ident) except ApiException as ae: abort(ae.status) if entity.state == "redirect": return redirect('/work/{}'.format(entity.redirect)) if entity.state == "deleted": return render_template('deleted_entity.html', entity=entity) return render_template('work_view.html', work=entity, releases=releases) @app.route('/editgroup/', methods=['GET']) def editgroup_view(ident): try: entity = api.get_editgroup(str(ident)) except ApiException as ae: abort(ae.status) return render_template('editgroup_view.html', editgroup=entity) @app.route('/editor/', methods=['GET']) def editor_view(ident): try: entity = api.get_editor(ident) except ApiException as ae: abort(ae.status) return render_template('editor_view.html', editor=entity) @app.route('/editor//editgroups', methods=['GET']) def editor_changelog(ident): try: editor = api.get_editor(ident) editgroups = api.get_editor_editgroups(ident, limit=50) except ApiException as ae: abort(ae.status) return render_template('editor_editgroups.html', editor=editor, editgroups=editgroups) # Not implemented #@app.route('/editor//wip', methods=['GET']) #def editor_wip(ident): # raise NotImplementedError # try: # editor = api.get_editor(ident) # entries = api.get_editor_wip(ident) # except ApiException as ae: # abort(ae.status) # return render_template('editor_changelog.html', editor=editor, # entries=entries) @app.route('/changelog', methods=['GET']) def changelog_view(): try: entries = api.get_changelog(limit=request.args.get('limit')) except ApiException as ae: abort(ae.status) return render_template('changelog.html', entries=entries) @app.route('/changelog/', methods=['GET']) def changelog_entry_view(index): try: entry = api.get_changelog_entry(int(index)) except ApiException as ae: abort(ae.status) return render_template('changelog_view.html', entry=entry, editgroup=entry.editgroup) ### Search ################################################################## @app.route('/release/search', methods=['GET', 'POST']) def search(): limit = 20 query = request.args.get('q') fulltext_only = bool(request.args.get('fulltext_only')) # Convert raw DOIs to DOI queries if query is not None: oldquery = query.split() for word in oldquery: if word.startswith("10.") and word.count("/") >= 1: query = query.replace(word, 'doi:"{}"'.format(word)) if 'q' in request.args.keys(): # always do files for HTML found = do_search(query, limit=limit, fulltext_only=fulltext_only) return render_template('release_search.html', found=found, query=query, fulltext_only=fulltext_only) else: return render_template('release_search.html', query=query, fulltext_only=fulltext_only) ### Auth #################################################################### @app.route('/auth/login') def login(): # show the user a list of login options return render_template('auth_login.html') @app.route('/auth/ia/login', methods=['GET', 'POST']) def ia_xauth_login(): if 'email' in request.form: # if a login attempt... return handle_ia_xauth(request.form.get('email'), request.form.get('password')) # else show form return render_template('auth_ia_login.html') @app.route('/auth/token_login', methods=['GET', 'POST']) def token_login(): # show the user a list of login options if 'token' in request.args: return handle_token_login(request.args.get('token')) if 'token' in request.form: return handle_token_login(request.form.get('token')) return render_template('auth_token_login.html') @app.route('/auth/change_username', methods=['POST']) @login_required def change_username(): # show the user a list of login options if not 'username' in request.form: abort(400) # on behalf of user... user_api = auth_api(session['api_token']) editor = user_api.get_editor(session['editor']['editor_id']) editor.username = request.form['username'] editor = user_api.update_editor(editor.editor_id, editor) # update our session session['editor'] = editor.to_dict() load_user(editor.editor_id) flash("Username updated successfully") return redirect('/auth/account') @app.route('/auth/logout') def logout(): handle_logout() return render_template('auth_logout.html') @app.route('/auth/account') @login_required def auth_account(): editor = api.get_editor(session['editor']['editor_id']) session['editor'] = editor.to_dict() load_user(editor.editor_id) return render_template('auth_account.html') ### Static Routes ########################################################### @app.errorhandler(404) def page_not_found(e): return render_template('404.html'), 404 @app.route('/', methods=['GET']) def homepage(): return render_template('home.html') @app.route('/about', methods=['GET']) def aboutpage(): return render_template('about.html') @app.route('/search', methods=['GET']) def search_redirect(): return redirect("/release/search") @app.route('/robots.txt', methods=['GET']) def robots(): return send_from_directory(os.path.join(app.root_path, 'static'), 'robots.txt', mimetype='text/plain') @app.route('/static/fatcat.jpg', methods=['GET']) def fatcat_photo(): return send_from_directory(os.path.join(app.root_path, 'static'), 'fatcat.jpg', mimetype='image/jpeg') @app.route('/health', methods=['GET']) def health(): return jsonify({'ok': True})