fix: type checks

frappe · Jan 23, 2025 · 92d40ee · 92d40ee
1 parent 5e6278a
commit 92d40ee
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/press/api/bench.py b/press/api/bench.py
@@ -1036,7 +1036,7 @@ def fail_and_redeploy(name: str, dc_name: str):
 
 
 @frappe.whitelist(allow_guest=True)
-def confirm_bench_transfer(key):
+def confirm_bench_transfer(key: str):
 	if team_change := frappe.db.get_value("Team Change", {"key": key}):
 		team_change = frappe.get_doc("Team Change", team_change)
 		team_change.transfer_completed = True

diff --git a/press/api/marketplace.py b/press/api/marketplace.py
@@ -1196,7 +1196,7 @@ def get_discount_percent(plan, discount=0.0):
 
 
 @frappe.whitelist(allow_guest=True)
-def login_via_token(token, team, site):
+def login_via_token(token: str, team: str, site: str):
 	if not token or not isinstance(token, str):
 		frappe.throw("Invalid Token")
 

diff --git a/press/api/site.py b/press/api/site.py
@@ -1981,7 +1981,7 @@ def send_change_team_request(name, team_mail_id, reason):
 
 
 @frappe.whitelist(allow_guest=True)
-def confirm_site_transfer(key):
+def confirm_site_transfer(key: str):
 	if team_change := frappe.db.get_value("Team Change", {"key": key}):
 		team_change = frappe.get_doc("Team Change", team_change)
 		team_change.transfer_completed = True