Upgrade users.current to v2

README.md

@@ -42,16 +42,16 @@ Alternatively, if you have the [`pip`](https://pip.pypa.io/) package installed,
 pip3 install axiom-py
 ```
 
-Create an API token with the permissions you need in the
-[Axiom settings](https://cloud.axiom.co/profile) and export it as 
-`AXIOM_TOKEN`.
+If you use the [Axiom CLI](https://github.com/axiomhq/cli), run `eval $(axiom config export -f)` to configure your environment variables.
+
+Otherwise create a personal token in [the Axiom settings](https://cloud.axiom.co/profile) and export it as `AXIOM_TOKEN`. Set `AXIOM_ORG_ID` to the organization ID from the settings page of the organization you want to access.
 
 You can also configure the client using options passed to the client constructor:
 
 ```py
 import axiom_py
 
-client = axiom_py.Client("<api token>")
+client = axiom_py.Client("<api token>", "<org id>")
 ```
 
 Create and use a client like this:

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:

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