From 35e4f208108daef82662c0f8f66066ba7f9797c0 Mon Sep 17 00:00:00 2001
From: Arne Bahlo <arne@axiom.co>
Date: Thu, 12 Sep 2024 12:17:38 +0200
Subject: [PATCH] feat: Upgrade users.current to v2

---
 examples/client.py    |  3 +++
 src/axiom_py/users.py | 12 +++++++++---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/examples/client.py b/examples/client.py
index 3bd321c..2aaa444 100644
--- a/examples/client.py
+++ b/examples/client.py
@@ -5,6 +5,9 @@ def main():
     client = Client()
     dataset_name = "my-dataset"
 
+    # Get current user
+    print(client.users.current())
+
     # List datasets
     res = client.datasets.get_list()
     for dataset in res:
diff --git a/src/axiom_py/users.py b/src/axiom_py/users.py
index fe5ea5a..5cef077 100644
--- a/src/axiom_py/users.py
+++ b/src/axiom_py/users.py
@@ -1,16 +1,22 @@
-from typing import List
 from .util import Util
 from dataclasses import dataclass
 from requests import Session
 
 
+@dataclass
+class Role:
+    id: str
+    name: str
+
+
 @dataclass
 class User:
     """An authenticated axiom user."""
 
     id: str
     name: str
-    emails: List[str]
+    email: str
+    role: Role
 
 
 class UsersClient:
@@ -25,6 +31,6 @@ def current(self) -> User:
 
         See https://axiom.co/docs/restapi/endpoints/getCurrentUser
         """
-        res = self.session.get("/v1/user")
+        res = self.session.get("/v2/user")
         user = Util.from_dict(User, res.json())
         return user