Secure a session#
The atoti-plus plugin is required to follow this how to.
By default, no authentication is required to query an atoti session and the users have access to all the data in the session.
This shows how to configure:
An authentication mechanism to secure access to the session.
Restrictions to control the data each user is allowed to see.
Configuring the authentication mechanism#
atoti supports multiple authentication mechanisms.
Here we will use OpenID Connect through Auth0:
[1]:
import os
import atoti as tt
session = tt.Session(
authentication=tt.OidcConfig(
provider_id="auth0",
# The connection details are read from environment variables for improved security.
# See https://12factor.net/config.
issuer_url=os.environ["AUTH0_ISSUER"],
client_id=os.environ["AUTH0_CLIENT_ID"],
client_secret=os.environ["AUTH0_CLIENT_SECRET"],
scopes=["email", "profile", "username", "roles"],
name_claim="email",
roles_claims=["https://activeviam:eu:auth0:com/roles"],
# Remove the following property to let end users authenticate using their browser.
access_token_format="opaque",
)
)
sales_table = session.read_csv(
"../../getting_started/tutorial/data/sales.csv", keys=["Sale ID"]
)
shops_table = session.read_csv(
"../../getting_started/tutorial/data/shops.csv", keys=["Shop ID"]
)
sales_table.join(shops_table, mapping={"Shop": "Shop ID"})
cube = session.create_cube(sales_table)
cube.schema
[1]:
The users configured in Auth0 are:
paris_manager@atoti.io with the roles Paris and atoti user.
france_manager@atoti.io with the roles France and atoti user.
global_manager@atoti.io with only the role atoti user.
Here we are using an Open ID Connect authentication provider, however the configuration of the rest of the session is similar if you are using ldap or kerberos authentication.
Managing security configurations on the session#
The UserServiceClient is used to all the dynamic security configuration on the session.
[2]:
from atoti_plus import UserServiceClient
user_service_client = UserServiceClient.from_session(session)
Roles and Restrictions#
Let’s create the roles required so that:
paris_manager@atoti.io only has access to Paris data.
france_manager@atoti.io only has access to France data.
global_manager@atoti.io has access to everything.
Since by default users have access to all the data we only need to create roles and restrictions for the regional managers:
[3]:
role_paris = user_service_client.create_role(
"ROLE_PARIS", restrictions={("Shops", "City"): ["Paris"]}
)
role_france = user_service_client.create_role(
"ROLE_FRANCE", restrictions={("Shops", "Country"): ["France"]}
)
We will update the role mapping to create a mapping between the roles our users have in Auth0, and the roles we want them to have in the application.
[4]:
user_service_client.oidc.role_mapping.update(
{
"Paris": [role_paris.name],
"France": [role_france.name],
"atoti user": ["ROLE_USER"],
}
)
Connecting to the session#
First, we can check that paris_manager@atoti.io can only see data for shops in Paris:
Same thing if we use a query session authenticated as this user:
[5]:
def query_city_and_quantity_sum(*, impersonated_username: str):
auth = tt.OAuth2ResourceOwnerPasswordAuthentication(
username=impersonated_username,
# For simplicity's sake, in this how-to, all the demo users share the same password.
password=os.environ["AUTH0_PASSWORD"],
issuer_url=os.environ["AUTH0_ISSUER"],
client_id=os.environ["AUTH0_CLIENT_ID"],
client_secret=os.environ["AUTH0_CLIENT_SECRET"],
)
query_session = tt.QuerySession(f"http://localhost:{session.port}", auth=auth)
query_cube = query_session.cubes[next(iter(session.cubes))]
return query_cube.query(
query_cube.measures["Quantity.SUM"],
levels=[query_cube.levels["City"]],
include_totals=True,
)
[6]:
query_city_and_quantity_sum(impersonated_username="paris_manager@atoti.io")
[6]:
Quantity.SUM | |
---|---|
City | |
Total | 603.00 |
Paris | 603.00 |
When we connect to the application as france_manager@atoti.io, we can only see data for shops in France:
[7]:
query_city_and_quantity_sum(impersonated_username="france_manager@atoti.io")
[7]:
Quantity.SUM | |
---|---|
City | |
Total | 3,027.00 |
Lyon | 609.00 |
Marseille | 603.00 |
Nice | 609.00 |
Paris | 603.00 |
Saint-Étienne | 603.00 |
And finally, when we connect as global_manager@atoti.io, we can see data for shops everywhere:
[8]:
query_city_and_quantity_sum(impersonated_username="global_manager@atoti.io")
[8]:
Quantity.SUM | |
---|---|
City | |
Total | 8,077.00 |
Chicago | 603.00 |
Houston | 606.00 |
Los Angeles | 606.00 |
Lyon | 609.00 |
Marseille | 603.00 |
New York | 808.00 |
Nice | 609.00 |
Paris | 603.00 |
Saint-Étienne | 603.00 |
San Antonio | 606.00 |
San Diego | 606.00 |
San Francisco | 612.00 |
San Jose | 603.00 |
Technical users#
Technical users (also called service accounts) are often required for plugging external services to the application (e.g. to collect metrics with Prometheus). In order to ease the connection of these tools, atoti automatically enables Basic Authentication on the session when another authentication mechanism is configured. Let’s create a technical user and query our application using the session’s REST API.
[9]:
technical_user_name = "Technical user"
technical_user_password = "change me"
technical_user = user_service_client.basic.create_user(
technical_user_name, password=technical_user_password
)
We create a simple endpoint to test the technical user’s credentials with:
[10]:
from dataclasses import asdict
@session.endpoint("whoami", method="GET")
def whoami(request, user, session):
return asdict(user)
Now we can query this endpoint by authenticating as the technical user:
[11]:
import requests
response = requests.get(
f"http://localhost:{session.port}/atoti/pyapi/whoami",
auth=(technical_user_name, technical_user_password),
)
response.json()
[11]:
{'name': 'Technical user', 'roles': ['ROLE_USER']}
To finish up, we can check that our endpoint requires authentication:
[12]:
response = requests.get(f"http://localhost:{session.port}/atoti/pyapi/whoami")
response.status_code
[12]:
403