Add activities command to cli

piket_client/cli.py

@@ -1,8 +1,10 @@
 import click
 from piket_client.model import (
+    AardbeiActivity,
     ServerStatus,
     NetworkError,
     Consumption,
+    AardbeiPeopleDiff,
     Person,
     Settlement,
     ConsumptionType,
@@ -217,5 +219,112 @@ def print_error(msg: str) -> None:
     click.echo(click.style(msg, fg="red", bold=True), err=True)
 
 
+@cli.group()
+@click.option("--token", required=True)
+@click.option("--endpoint", default="http://localhost:3000")
+@click.pass_context
+def aardbei(ctx, token: str, endpoint: str) -> None:
+    ctx.ensure_object(dict)
+    ctx.obj["AardbeiToken"] = token
+    ctx.obj["AardbeiEndpoint"] = endpoint
+
+
+@aardbei.group("activities")
+def aardbei_activities() -> None:
+    pass
+
+
+@aardbei_activities.command("list")
+@click.pass_context
+def aardbei_list_activities(ctx) -> None:
+    acts = AardbeiActivity.get_available(
+        token=ctx.obj["AardbeiToken"], endpoint=ctx.obj["AardbeiEndpoint"]
+    )
+
+    if isinstance(acts, NetworkError):
+        print_error(f"Could not get activities: {acts.value}")
+        return
+
+    table = PrettyTable()
+    table.field_names = ["ID", "Name"]
+    table.align = "l"
+
+    for a in acts:
+        table.add_row([a.aardbei_id, a.name])
+
+    print(table)
+
+
+@aardbei_activities.command("apply")
+@click.argument("activity_id", type=click.INT)
+@click.pass_context
+def aardbei_apply_activity(ctx, activity_id: int) -> None:
+    result = AardbeiActivity.apply_activity(
+        token=ctx.obj["AardbeiToken"],
+        endpoint=ctx.obj["AardbeiEndpoint"],
+        activity_id=activity_id,
+    )
+
+    if isinstance(result, NetworkError):
+        print_error("Failed to apply activity: {result.value}")
+        return
+
+    print_ok(f"Activity applied. There are now {result} active people.")
+
+
+@aardbei.group("people")
+def aardbei_people() -> None:
+    pass
+
+
+@aardbei_people.command("diff")
+@click.pass_context
+def aardbei_diff_people(ctx) -> None:
+    diff = AardbeiPeopleDiff.get_diff(
+        token=ctx.obj["AardbeiToken"], endpoint=ctx.obj["AardbeiEndpoint"]
+    )
+
+    if isinstance(diff, NetworkError):
+        print_error(f"Could not get differences: {diff.value}")
+        return
+
+    if diff.num_changes == 0:
+        print_ok("There are no changes to apply.")
+        return
+
+    click.echo(f"There are {diff.num_changes} pending changes:")
+    show_diff(diff)
+
+
+@aardbei_people.command("sync")
+@click.pass_context
+def aardbei_sync_people(ctx) -> None:
+    diff = AardbeiPeopleDiff.sync(
+        token=ctx.obj["AardbeiToken"], endpoint=ctx.obj["AardbeiEndpoint"]
+    )
+
+    if isinstance(diff, NetworkError):
+        print_error(f"Could not apply differences: {diff.value}")
+        return
+
+    if diff.num_changes == 0:
+        print_ok("There were no changes to apply.")
+        return
+
+    print_ok(f"Applied {diff.num_changes} pending changes:")
+    show_diff(diff)
+
+
+def show_diff(diff: AardbeiPeopleDiff) -> None:
+    for name in diff.new_people:
+        click.echo(f" - Create local Person for {name}")
+
+    for name in diff.link_existing:
+        click.echo(f" - Link local and remote people for {name}")
+
+    for name in diff.altered_name:
+        click.echo(f" - Process name change for {name}")
+
+
 if __name__ == "__main__":
     cli()

piket_client/model.py

@@ -229,6 +229,7 @@ class Person(NamedTuple):
             return NetworkError.HttpFailure
 
         except ValueError as e:
+            LOG.exception(e)
             return NetworkError.InvalidData
 
     @classmethod
@@ -546,3 +547,124 @@ class Settlement(NamedTuple):
         data["settlement"]["count_info"] = data["count_info"]
 
         return cls.from_dict(data["settlement"])
+
+
+@dataclass(frozen=True)
+class AardbeiActivity:
+    aardbei_id: int
+    name: str
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> AardbeiActivity:
+        return cls(data["activity"]["id"], data["activity"]["name"])
+
+    @classmethod
+    def get_available(
+        cls, token: str, endpoint: str
+    ) -> Union[List[AardbeiActivity], NetworkError]:
+        try:
+            req = requests.post(
+                urljoin(SERVER_URL, "/aardbei/get_activities"),
+                json={"endpoint": endpoint, "token": token},
+            )
+
+            req.raise_for_status()
+            return [cls.from_dict(x) for x in req.json()["activities"]]
+
+        except requests.ConnectionError as e:
+            LOG.exception(e)
+            return NetworkError.ConnectionFailure
+
+        except requests.HTTPError as e:
+            LOG.exception(e)
+            return NetworkError.HttpFailure
+
+        except ValueError as e:
+            LOG.exception(e)
+            return NetworkError.InvalidData
+
+    @classmethod
+    def apply_activity(
+        cls, token: str, endpoint: str, activity_id: int
+    ) -> Union[int, NetworkError]:
+        try:
+            req = requests.post(
+                urljoin(SERVER_URL, "/aardbei/apply_activity"),
+                json={"activity_id": activity_id, "token": token, "endpoint": endpoint},
+            )
+            req.raise_for_status()
+            data = req.json()
+
+            return data["activity"]["response_counts"]["present"]
+
+        except requests.ConnectionError as e:
+            LOG.exception(e)
+            return NetworkError.ConnectionFailure
+
+        except requests.HTTPError as e:
+            LOG.exception(e)
+            return NetworkError.HttpFailure
+
+        except ValueError as e:
+            LOG.exception(e)
+            return NetworkError.InvalidData
+
+
+@dataclass(frozen=True)
+class AardbeiPeopleDiff:
+    altered_name: List[str]
+    link_existing: List[str]
+    new_people: List[str]
+    num_changes: int
+
+    @classmethod
+    def from_dict(cls, data: Dict[str, Any]) -> AardbeiPeopleDiff:
+        return cls(**data)
+
+    @classmethod
+    def get_diff(cls, token: str, endpoint: str) -> Union[AardbeiPeopleDiff, NetworkError]:
+        try:
+            req = requests.post(
+                urljoin(SERVER_URL, "/aardbei/diff_people"),
+                json={"endpoint": endpoint, "token": token},
+            )
+            req.raise_for_status()
+            data = req.json()
+
+            return cls.from_dict(data)
+
+        except requests.ConnectionError as e:
+            LOG.exception(e)
+            return NetworkError.ConnectionFailure
+
+        except requests.HTTPError as e:
+            LOG.exception(e)
+            return NetworkError.HttpFailure
+
+        except ValueError as e:
+            LOG.exception(e)
+            return NetworkError.InvalidData
+
+    @classmethod
+    def sync(cls, token: str, endpoint: str) -> Union[AardbeiPeopleDiff, NetworkError]:
+        try:
+            req = requests.post(
+                urljoin(SERVER_URL, "/aardbei/sync_people"),
+                json={"endpoint": endpoint, "token": token},
+            )
+            req.raise_for_status()
+            data = req.json()
+
+            return cls.from_dict(data)
+
+        except requests.ConnectionError as e:
+            LOG.exception(e)
+            return NetworkError.ConnectionFailure
+
+        except requests.HTTPError as e:
+            LOG.exception(e)
+            return NetworkError.HttpFailure
+
+        except ValueError as e:
+            LOG.exception(e)
+            return NetworkError.InvalidData