import os from flask import jsonify, make_response, request, send_file from ki.auth import auth from ki.models import Language, Skill from app import app def models_to_list(models): models_list = [] for model in models: models_list.append(model.to_dict()) return models_list def handle_completion_request(model, key): query = model.query if "search" in request.args: query = query.filter(model.name.startswith(request.args.get("search"))) results = query.order_by(model.name) \ .limit(10) \ .all() api_results = models_to_list(results) response_data = {} response_data[key] = api_results return response_data def handle_icon_request(model, id, path): object = model.query.get(id) if object is None: return make_response({}, 404) icon_base_path = path + str(id) icon_svg_path = icon_base_path + ".svg" if os.path.exists(icon_svg_path): return send_file(icon_svg_path, mimetype="image/svg") icon_png_path = icon_base_path + ".png" if os.path.exists(icon_png_path): return send_file(icon_png_path, mimetype="image/png") unknown_svg_path = path + "unknown.svg" if os.path.exists(unknown_svg_path): return send_file(unknown_svg_path, mimetype="image/svg") unknown_png_path = path + "unknown.png" if os.path.exists(unknown_png_path): return send_file(unknown_png_path, mimetype="image/png") return make_response({"error": "icon not found"}, 404) @app.route("/") def hello_world(): return "KI" @app.route("/users/login", methods=["POST"]) def login(): username = request.json.get("username", "") password = request.json.get("password", "") token = auth(username, password) if token is None: return make_response({}, 403) return make_response({"token": token.token}) @app.route("/skills") def get_skills(): return handle_completion_request(Skill, "skills") @app.route("/skills//icon") def get_skill_icon(skill_id): skill_icons_path = app.config["KI_DATA_DIR"] + "/imgs/skill_icons/" return handle_icon_request(Skill, skill_id, skill_icons_path) @app.route("/languages") def get_languages(): return handle_completion_request(Language, "languages") @app.route("/languages//icon") def get_language_icon(language_id): language_flags_path = app.config["KI_DATA_DIR"] + "/imgs/flags/" return handle_icon_request(Language, language_id, language_flags_path)