Skip to content

Commit

Permalink
Use AUTH_SERVER_URL setting from flask.current_app not gn2.utility.tools
Browse files Browse the repository at this point in the history
  • Loading branch information
fredmanglis committed Mar 6, 2024
1 parent b14fefe commit b63b91c
Show file tree
Hide file tree
Showing 4 changed files with 13 additions and 20 deletions.
1 change: 0 additions & 1 deletion gn2/utility/tools.py
Expand Up @@ -271,7 +271,6 @@ def gn_version_repo_info(root_dir):
GN2_BASE_URL = get_setting('GN2_BASE_URL')
GN2_BRANCH_URL = get_setting('GN2_BRANCH_URL')
GN_SERVER_URL = get_setting('GN_SERVER_URL')
AUTH_SERVER_URL = get_setting('AUTH_SERVER_URL')
GN_PROXY_URL = get_setting('GN_PROXY_URL')
GN3_LOCAL_URL = get_setting('GN3_LOCAL_URL')
SERVER_PORT = get_setting_int('SERVER_PORT')
Expand Down
2 changes: 1 addition & 1 deletion gn2/wqflask/oauth2/checks.py
Expand Up @@ -27,7 +27,7 @@ def __with_token__(token):
client = OAuth2Session(
oauth2_clientid(), oauth2_clientsecret(), token=token)
resp = client.get(
urljoin(AUTH_SERVER_URL, "auth/user/"))
urljoin(authserver_uri(), "auth/user/"))
user_details = resp.json()
if not user_details.get("error", False):
return func(*args, **kwargs)
Expand Down
19 changes: 8 additions & 11 deletions gn2/wqflask/oauth2/data.py
Expand Up @@ -15,14 +15,13 @@
from gn2.jobs import jobs
from .ui import render_ui
from .request_utils import process_error
from .client import oauth2_get, oauth2_post
from .client import oauth2_get, oauth2_post, authserver_uri

data = Blueprint("data", __name__)

def __search_mrna__(query, template, **kwargs):
from gn2.utility.tools import AUTH_SERVER_URL
species_name = kwargs["species_name"]
search_uri = urljoin(AUTH_SERVER_URL, "auth/data/search")
search_uri = urljoin(authserver_uri(), "auth/data/search")
datasets = oauth2_get(
"auth/data/search",
json = {
Expand All @@ -45,9 +44,8 @@ def __selected_datasets__():
request.form.get("selected", []))

def __search_genotypes__(query, template, **kwargs):
from gn2.utility.tools import AUTH_SERVER_URL
species_name = kwargs["species_name"]
search_uri = urljoin(AUTH_SERVER_URL, "auth/data/search")
search_uri = urljoin(authserver_uri(), "auth/data/search")
datasets = oauth2_get(
"auth/data/search",
json = {
Expand All @@ -61,7 +59,7 @@ def __search_genotypes__(query, template, **kwargs):
return render_ui(template, search_uri=search_uri, **datasets, **kwargs)

def __search_phenotypes__(query, template, **kwargs):
from gn2.utility.tools import GN_SERVER_URL, AUTH_SERVER_URL
from gn2.utility.tools import GN_SERVER_URL
page = int(request.args.get("page", 1))
per_page = int(request.args.get("per_page", 50))
selected_traits = request.form.getlist("selected_traits")
Expand All @@ -71,10 +69,10 @@ def __search_success__(search_results):
template, traits=[], per_page=per_page, query=query,
selected_traits=selected_traits, search_results=search_results,
search_endpoint=urljoin(
AUTH_SERVER_URL, "auth/data/search"),
gn_server_url = AUTH_SERVER_URL,
authserver_uri(), "auth/data/search"),
gn_server_url = authserver_uri(),
results_endpoint=urljoin(
AUTH_SERVER_URL,
authserver_uri(),
f"auth/data/search/phenotype/{job_id}"),
**kwargs)
return oauth2_get("auth/data/search", json={
Expand Down Expand Up @@ -124,7 +122,6 @@ def __handle_error__(err):
@data.route("/phenotype/search", methods=["POST"])
def json_search_phenotypes() -> Response:
"""Search for phenotypes."""
from gn2.utility.tools import AUTH_SERVER_URL
form = request.json
def __handle_error__(err):
error = process_error(err)
Expand All @@ -138,7 +135,7 @@ def __handle_error__(err):
"query": form.get("query", ""),
"per_page": int(form.get("per_page", 50)),
"page": int(form.get("page", 1)),
"auth_server_uri": AUTH_SERVER_URL,
"auth_server_uri": authserver_uri(),
"selected_traits": form.get("selected_traits", [])
}).either(__handle_error__, jsonify)

Expand Down
11 changes: 4 additions & 7 deletions gn2/wqflask/oauth2/users.py
Expand Up @@ -11,7 +11,7 @@
from . import session
from .ui import render_ui
from .checks import require_oauth2, user_logged_in
from .client import oauth2_get, oauth2_post, oauth2_client
from .client import oauth2_get, oauth2_post, oauth2_client, authserver_uri
from .request_utils import (
user_details, request_error, process_error, with_flash_error)

Expand Down Expand Up @@ -65,15 +65,14 @@ def __success__(response):
@users.route("/login", methods=["GET", "POST"])
def login():
"""Route to allow users to sign up."""
from gn2.utility.tools import AUTH_SERVER_URL
next_endpoint=request.args.get("next", False)

if request.method == "POST":
form = request.form
client = oauth2_client()
try:
token = client.fetch_token(
urljoin(AUTH_SERVER_URL, "auth/token"),
urljoin(authserver_uri(), "auth/token"),
username=form.get("email_address"),
password=form.get("password"),
grant_type="password")
Expand Down Expand Up @@ -101,10 +100,9 @@ def login():

@users.route("/logout", methods=["GET", "POST"])
def logout():
from gn2.utility.tools import AUTH_SERVER_URL
if user_logged_in():
resp = oauth2_client().revoke_token(
urljoin(AUTH_SERVER_URL, "auth/revoke"))
urljoin(authserver_uri(), "auth/revoke"))
the_session = session.session_info()
if not bool(the_session["masquerading"]):
# Normal session - clear and go back.
Expand All @@ -124,7 +122,6 @@ def logout():

@users.route("/register", methods=["GET", "POST"])
def register_user():
from gn2.utility.tools import AUTH_SERVER_URL
if user_logged_in():
next_endpoint=request.args.get("next", "/")
flash(("You cannot register a new user while logged in. "
Expand All @@ -137,7 +134,7 @@ def register_user():

form = request.form
response = requests.post(
urljoin(AUTH_SERVER_URL, "auth/user/register"),
urljoin(authserver_uri(), "auth/user/register"),
data = {
"user_name": form.get("user_name"),
"email": form.get("email_address"),
Expand Down

0 comments on commit b63b91c

Please sign in to comment.