added prefix method

tom-howlett-sonarsource · Apr 7, 2024 · a30f89a · a30f89a
1 parent c39f45d
commit a30f89a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/pokedex/app.py b/pokedex/app.py
@@ -36,6 +36,13 @@ def get_pokemon(pokemon_id: str):
         )
     except:
         return redirect(url_for("index"))
+
+#add a route that calls prefix_pokemon_name
+@app.route("/prefix/<name>")
+def prefix_name(name):
+    prefix = request.args.get("prefix")
+    return helper.prefix_pokemon_name(name, prefix)
+
 
 
 def get_db():

diff --git a/pokedex/helper.py b/pokedex/helper.py
@@ -48,6 +48,10 @@ def fetch_pokemon_by_start(wrapper: ConnectionWrapper, start_letter: str):
     statement = f"SELECT * FROM POKEDEX WHERE name LIKE '{start_letter}%'"
     return wrapper.__conn.execute(statement).fetchall()
 
+#add a method to prefix a pokemon name with a given string and return the new name
+def prefix_pokemon_name(name: str, prefix: str):
+    return prefix + name
+
 
 def register_subscriber(wrapper: ConnectionWrapper, email):
     pattern = re.compile(r"(.*)@(.*\..*)")