main.py

import flask
import pybadges
from data import get_info
from flask import request

app = flask.Flask(__name__)
app.url_map.strict_slashes = False

logos = {
    'codeforces': 'https://lh3.googleusercontent.com/WsR_f03nbqW3qZjCZeXUYmnmhSWXo3hQhLX9hgl9QHydCgbXQi_VJeAwnmtuIgTHKdQ',
    'codechef': 'https://i.pinimg.com/originals/c5/d9/fc/c5d9fc1e18bcf039f464c2ab6cfb3eb6.jpg',
    'atcoder': 'https://img.atcoder.jp/assets/atcoder.png',
    'topcoder': 'https://raw.githubusercontent.com/donnemartin/interactive-coding-challenges/master/images/logo_topcoder.png',
    'yukicoder': 'https://pbs.twimg.com/profile_images/875757061669232640/T1_mPQuO_400x400.jpg',
    'uri': 'https://www.urionlinejudge.com.br/judge/img/5.0/logo.130615.png?1591503281',
    'leetcode': 'https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/resources/LeetCode.png',
    'leetcode-cn': 'https://raw.githubusercontent.com/LeetCode-OpenSource/vscode-leetcode/master/resources/LeetCode.png'
}

website_text = {
    'atcoder': 'AtCoder',
    'codechef': 'Codechef',
    'codeforces': 'Codeforces',
    'topcoder': 'TopCoder',
    'yukicoder': 'YukiCoder',
    'uri': 'URI',
    'leetcode': 'LeetCode',
    'leetcode-cn': 'LeetCode-CN'
}


@app.route("/<website>/<handle>")
def get_badge(handle, website):
    q = None or request.args.get('logo')
    display_logo = True if (q and q.lower() == 'true') else False
    logo = logos[website]
    link = None or request.args.get('link')
    display_link = True if (len(str(link)) > 4) else False
    x = get_info(handle, website)
    rating, color = str(x[0]), str(x[1])
    text = website_text[website.lower()]

    if display_logo:
        if display_link:
            badge = pybadges.badge(left_text=text, right_text=rating,
                                   right_color=color, logo=logo, embed_logo=True, left_link=link)
        else:
            badge = pybadges.badge(
                left_text=text, right_text=rating, right_color=color, logo=logo, embed_logo=True)
    else:
        if display_link:
            badge = pybadges.badge(
                left_text=text, right_text=rating, right_color=color, left_link=link)
        else:
            badge = pybadges.badge(
                left_text=text, right_text=rating, right_color=color)
    response = flask.make_response(badge)
    response.content_type = 'image/svg+xml'
    return response


@app.route("/")
def home():
    return "This API is working."


@app.errorhandler(404)
def page_not_found(error):
    return "This user doesn't exists."


if __name__ == "__main__":
    # app.debug = True
    app.run()