diff --git a/main.py b/main.py index d4b4677..154b71e 100755 --- a/main.py +++ b/main.py @@ -50,29 +50,14 @@ def exception_handler(e: werkzeug.exceptions.InternalServerError) -> tuple[str, @app.route("/") def start() -> str: """Start form.""" - enwp = flask.request.args.get("enwp") - if not enwp: - return flask.render_template("combined.html") - enwp = enwp.strip() - if not enwp: + wikipedia_url = flask.request.args.get("wikipedia") + if not wikipedia_url: return flask.render_template("combined.html") - input_is = "url" if enwp.startswith(enwiki) else "title" + start = wikipedia_url.find(enwiki) + len(enwiki) + wiki_part2 = unquote(wikipedia_url[start:]) - wikipedia_url: str - wiki_part1: str - wiki_part2: str - if input_is == "url": - start = enwp.find(enwiki) + len(enwiki) - wiki_part2 = unquote(enwp[start:]) - name = wiki_part2 - wiki_part1 = enwp[:start] - wikipedia_url = enwp - else: - name = enwp - wiki_part1 = enwiki - wiki_part2 = name.replace(" ", "_") - wikipedia_url = wiki_part1 + wiki_part2 + name = wiki_part2 if "_(" in name: name = name[: name.find("_(")] @@ -83,9 +68,11 @@ def start() -> str: return flask.render_template( "combined.html", name=name, - enwp=enwp, + wikipedia_url=wikipedia_url, ) + wiki_part1 = wikipedia_url[:start] + if "/in/" in flickr_url: flickr_url = flickr_url[: flickr_url.find("/in/")] @@ -103,7 +90,6 @@ def start() -> str: msg = flask.render_template( "message.jinja", flickr_url=flickr_url, - enwp=enwp, wikipedia_url=wikipedia_url, name=name, wiki_part1=wiki_part1, @@ -117,7 +103,7 @@ def start() -> str: return flask.render_template( "combined.html", name=name, - enwp=enwp, + wikipedia_url=wikipedia_url, flickr_url=flickr_url, subject=subject, lines=lines, diff --git a/templates/combined.html b/templates/combined.html index 7609539..d458e7e 100644 --- a/templates/combined.html +++ b/templates/combined.html @@ -8,8 +8,8 @@