add default org
This commit is contained in:
@@ -17,9 +17,20 @@ def get_orgs(
|
||||
start_key: Annotated[str | None, Query] = None,
|
||||
limit: Annotated[int, Query(ge=25)] = 25,
|
||||
):
|
||||
return dyn.collection.query(
|
||||
# Post-migration (users): rename `orgs#` to `ORG#`
|
||||
r = dyn.collection.query(
|
||||
# Post-migration (users): uncomment the following line
|
||||
# key=KeyPair(user_id, 'ORG#'),
|
||||
key=KeyPair(user_id, 'orgs#'),
|
||||
start_key=start_key,
|
||||
limit=limit,
|
||||
)
|
||||
|
||||
preferred, items = None, []
|
||||
|
||||
for x in r['items']:
|
||||
if 'DEFAULT' in x['sk']:
|
||||
preferred = x.get('org_id')
|
||||
else:
|
||||
items.append(x)
|
||||
|
||||
return r | {'items': items} | ({'preferred_org_id': preferred} if preferred else {})
|
||||
|
||||
Reference in New Issue
Block a user