def post_api_key(
api_key: rest_models.BaseApiKey,
dao: Dao = Depends(get_dao),
auth: authorization.Rules = Depends(get_rules),
):
auth.assert_create_api_key_roles(api_key.roles)
user_id = auth.assert_user()
key = generate_random_key(32)
dao.create_api_key(user_id, api_key, key)
user_role_keys, custom_role_keys = dao.get_api_keys_with_members(
user_id, key)
if len(user_role_keys) > 0:
key = user_role_keys[0]
return rest_models.ApiKey(
key=key.key,
description=key.description,
time_created=key.time_created,
expire_at=key.expire_at,
roles=None,
)
else:
key = custom_role_keys[0][0]
package_member = custom_role_keys[0][1]
channel_member = custom_role_keys[0][2]
roles = []
if package_member:
roles.append(
CPRole(
channel=package_member.channel_name,
package=package_member.package_name,
role=package_member.role,
))
if channel_member:
roles.append(
CPRole(
channel=channel_member.channel_name,
package=None,
role=channel_member.role,
))
return rest_models.ApiKey(
key=key.key,
description=key.description,
time_created=key.time_created,
expire_at=key.expire_at,
roles=roles,
)
def get_api_keys(dao: Dao = Depends(get_dao),
auth: authorization.Rules = Depends(get_rules)):
"""Get API keys for current user"""
user_id = auth.assert_user()
user_role_keys, custom_role_keys = dao.get_api_keys_with_members(user_id)
api_keys = []
for key in user_role_keys:
api_keys.append(
rest_models.ApiKey(
key=key.key,
description=key.description,
time_created=key.time_created,
expire_at=key.expire_at,
roles=None,
))
from itertools import groupby
grouped_by_key = groupby(custom_role_keys, key=lambda k: k[0])
for group_key, group_items in grouped_by_key:
roles = []
for _, package_member, channel_member in group_items:
if package_member:
roles.append(
CPRole(
channel=package_member.channel_name,
package=package_member.package_name,
role=package_member.role,
))
if channel_member:
roles.append(
CPRole(
channel=channel_member.channel_name,
package=None,
role=channel_member.role,
))
api_keys.append(
rest_models.ApiKey(
key=group_key.key,
description=group_key.description,
time_created=group_key.time_created,
expire_at=group_key.expire_at,
roles=roles,
))
return api_keys
def get_api_keys(dao: Dao = Depends(get_dao),
auth: authorization.Rules = Depends(get_rules)):
"""Get API keys for current user"""
user_id = auth.assert_user()
api_key_list = dao.get_package_api_keys(user_id)
api_channel_key_list = dao.get_channel_api_keys(user_id)
from itertools import groupby
return [
rest_models.ApiKey(
key=api_key.key,
description=api_key.description,
roles=[
rest_models.CPRole(
channel=member.channel_name,
package=member.package_name if hasattr(
member, 'package_name') else None,
role=member.role,
) for member, api_key in member_key_list
],
) for api_key, member_key_list in groupby(
[*api_key_list, *api_channel_key_list],
lambda member_api_key: member_api_key[1],
)
]
def post_api_key(
api_key: rest_models.BaseApiKey,
dao: Dao = Depends(get_dao),
auth: authorization.Rules = Depends(get_rules),
):
auth.assert_create_api_key_roles(api_key.roles)
user_id = auth.assert_user()
key = secrets.token_urlsafe(32)
dao.create_api_key(user_id, api_key, key)
return rest_models.ApiKey(description=api_key.description,
roles=api_key.roles,
key=key)
