From 588c46173e28abb50d7b176ff09b383e0b7b0a09 Mon Sep 17 00:00:00 2001
From: Melody Horn <melody@boringcactus.com>
Date: Mon, 29 Mar 2021 19:10:24 -0600
Subject: allow overriding package names in specific repos

---
 app.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'app.py')

diff --git a/app.py b/app.py
index 248bb8d..05e8449 100644
--- a/app.py
+++ b/app.py
@@ -1,4 +1,4 @@
-from flask import Flask, make_response, render_template
+from flask import Flask, make_response, render_template, request
 from jinja2 import select_autoescape
 
 import repos
@@ -8,12 +8,13 @@ app.jinja_options['autoescape'] = select_autoescape(default=True)
 
 @app.route('/')
 def hello_world():
-    return '<a href="/badge/rust.svg">sample badge for Rust</a>'
+    return '<a href="/badge/nushell.svg?crates-io=nu">sample badge for Rust</a>'
 
 
 @app.route('/badge/<package>.svg')
 def badge(package: str):
-    versions = repos.get_versions(package)
+    args = request.args
+    versions = repos.get_versions(package, args)
     newest_version = max(versions.values())
     rendered = render_template('badge.svg.jinja', versions=versions, newest_version=newest_version)
     response = make_response(rendered)
-- 
cgit v1.2.3