roboherd 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

roboherd/__main__.py

@@ -11,12 +11,13 @@ from roboherd.herd.manager import HerdManager
 from roboherd.util import create_connection
 from roboherd.register import register as run_register
 from roboherd.validators import validators
+from roboherd.version import __version__
 
 logging.basicConfig(level=logging.INFO)
 logging.captureWarnings(True)
 
 
-@click.group()
+@click.group(invoke_without_command=True)
 @click.option(
     "--connection_string",
     default=None,
@@ -28,9 +29,22 @@ logging.captureWarnings(True)
     help="Base url to create cows with",
 )
 @click.option("--config_file", default="roboherd.toml", help="Configuration file")
+@click.option(
+    "--version", is_flag=True, default=False, help="display version then exit"
+)
 @click.pass_context
-def main(ctx, connection_string, base_url, config_file):
+def main(
+    ctx: click.Context,
+    connection_string: str,
+    base_url: str,
+    config_file: str,
+    version: bool,
+):
     """Configuration is usually loaded from the config_file. These options can be overwritten by passing as a command line argument."""
+    if version:
+        print(f"roboherd version: {__version__}")
+        exit(0)
+
     settings = dynaconf.Dynaconf(
         settings_files=[config_file],
         envvar_prefix="ROBOHERD",
@@ -44,17 +58,39 @@ def main(ctx, connection_string, base_url, config_file):
     if connection_string:
         ctx.obj["connection_string"] = connection_string
     else:
-        ctx.obj["connection_string"] = settings.connection_string
+        ctx.obj["connection_string"] = settings.connection_string  # type: ignore
 
     if base_url:
         ctx.obj["base_url"] = base_url
     else:
-        ctx.obj["base_url"] = settings.base_url
+        ctx.obj["base_url"] = settings.base_url  # type: ignore
+
+    print("Please specify a command")
+
+
+@main.command()
+@click.option("--fail", is_flag=True, default=False, help="Fail if actors do not exist")
+@click.pass_context
+def check(ctx: click.Context, fail: bool):
+    """Checks that the connection is configured correctly"""
+
+    create_connection(ctx)
+
+    herd = RoboHerd(base_url=ctx.obj["base_url"])
+
+    settings = ctx.obj["settings"]
+
+    if settings.get("cow"):
+        herd.manager = HerdManager.from_settings(settings)
+        asyncio.run(herd.check(ctx.obj["connection"], raise_if_cows_to_create=fail))
+    else:
+        click.echo("No cows specified")
+        exit(1)
 
 
 @main.command()
 @click.pass_context
-def run(ctx):
+def run(ctx: click.Context):
     """Runs the roboherd by connecting to the server."""
 
     create_connection(ctx)
@@ -91,7 +127,7 @@ def watch(ctx):
     prompt=True,
 )
 @click.option("--fediverse", help="Fediverse handle", prompt=True)
-def register(ctx, name, password, fediverse):
+def register(ctx: click.Context, name: str, password: str, fediverse: str):
     """Registers a new account on dev.bovine.social. All three options are required. If not provided, you will be prompted for them."""
 
     if os.path.exists(ctx.obj["config_file"]):

roboherd/herd/__init__.py

@@ -1,7 +1,6 @@
 import asyncio
 import logging
 
-from typing import List, Tuple
 from dataclasses import dataclass, field
 
 from roboherd.cow import RoboCow, CronEntry
@@ -20,7 +19,7 @@ class RoboHerd:
     base_url: str = "http://abel"
 
     manager: HerdManager = field(default_factory=HerdManager)
-    cows: List[RoboCow] = field(default_factory=list)
+    cows: list[RoboCow] = field(default_factory=list)  # type: ignore
 
     async def run(self, connection: Almabtrieb):
         async with connection:
@@ -28,12 +27,26 @@ class RoboHerd:
             await self.startup(connection)
             await self.process(connection)
 
+    async def check(
+        self, connection: Almabtrieb, raise_if_cows_to_create: bool = False
+    ):
+        async with connection:
+            if not connection.information:
+                raise Exception("Could not get information from server")
+            self.validate(connection)
+            cows_to_create = self.manager.cows_to_create(connection.information.actors)
+
+            if len(cows_to_create) > 0:
+                print("Missing cows: " + ", ".join([x.name for x in cows_to_create]))
+
+                if raise_if_cows_to_create:
+                    raise Exception("Missing cows")
+
     async def startup(self, connection: Almabtrieb):
         if not connection.information:
             raise Exception("Could not get information from server")
 
         self.cows = self.manager.existing_cows(connection.information.actors)
-
         cows_to_create = self.manager.cows_to_create(connection.information.actors)
 
         for cow_config in cows_to_create:
@@ -65,8 +78,10 @@ class RoboHerd:
 
             connection.add_on_disconnect(scheduler.stop)
 
-    def validate(self, connection):
+    def validate(self, connection: Almabtrieb):
         result = connection.information
+        if result is None:
+            raise ValueError("information not retrieved from connection")
 
         logger.info("Got base urls: %s", ",".join(result.base_urls))
 
@@ -78,7 +93,7 @@ class RoboHerd:
             )
             raise ValueError("Incorrectly configured base url")
 
-    def cron_entries(self) -> List[Tuple[RoboCow, CronEntry]]:
+    def cron_entries(self) -> list[tuple[RoboCow, CronEntry]]:
         """Returns the cron entries of all cows"""
 
         result = []
@@ -88,7 +103,7 @@ class RoboHerd:
 
         return result
 
-    def incoming_handlers(self) -> List[RoboCow]:
+    def incoming_handlers(self) -> list[RoboCow]:
         result = []
         for cow in self.cows:
             if cow.internals.handlers.has_handlers:

roboherd/herd/manager/__init__.py

@@ -1,6 +1,5 @@
 import logging
 
-from typing import List
 from dataclasses import dataclass, field
 
 from almabtrieb.model import ActorInformation
@@ -21,7 +20,7 @@ class HerdManager:
     def from_settings(settings):
         return HerdManager(herd_config=HerdConfig.from_settings(settings))
 
-    def existing_cows(self, actors: List[ActorInformation]) -> List[RoboCow]:
+    def existing_cows(self, actors: list[ActorInformation]) -> list[RoboCow]:
         existing_cows = []
 
         for info in actors:

roboherd/herd/test_herd.py

@@ -6,11 +6,11 @@ from roboherd.examples.dev_null import bot as dev_null
 
 from roboherd.cow import CronEntry
 
-from . import RoboHerd
+from . import RoboHerd, RoboCow
 
 
 @pytest.mark.parametrize("cow, length", [(moocow, 0), (bot, 1)])
-def test_cron_entries(cow, length):
+def test_cron_entries(cow: RoboCow, length: int):
     manager = RoboHerd(cows=[cow])
 
     assert len(manager.cron_entries()) == length

roboherd/register.py

@@ -2,7 +2,9 @@ import aiohttp
 import tomli_w
 
 
-def create_config(name, password, domain="dev.bovine.social"):
+def create_config(
+    name: str, password: str, domain: str = "dev.bovine.social"
+) -> dict[str, bool | str]:
     """
 
     ```
@@ -21,7 +23,7 @@ def create_config(name, password, domain="dev.bovine.social"):
     }
 
 
-async def register(config_file, name, password, fediverse):
+async def register(config_file: str, name: str, password: str, fediverse: str):
     async with aiohttp.ClientSession() as session:
         result = await session.post(
             "https://dev.bovine.social/register",

roboherd/util.py

@@ -8,7 +8,7 @@ from almabtrieb import Almabtrieb
 logger = logging.getLogger(__name__)
 
 
-def parse_connection_string(connection_string: str) -> dict:
+def parse_connection_string(connection_string: str) -> dict[str, str | int | None]:
     """
     Parse a connection string into a dictionary of connection parameters.
 
@@ -43,7 +43,7 @@ def parse_connection_string(connection_string: str) -> dict:
     }
 
 
-def create_connection(ctx):
+def create_connection(ctx: click.Context):
     connection_string = ctx.obj["connection_string"]
     base_url = ctx.obj["base_url"]
 

roboherd/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.11"

{roboherd-0.1.9.dist-info → roboherd-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: roboherd
-Version: 0.1.9
+Version: 0.1.11
 Summary: A Fediverse bot framework
 Project-URL: Documentation, https://bovine.codeberg.page/roboherd/
 Project-URL: Repository, https://codeberg.org/bovine/roboherd

{roboherd-0.1.9.dist-info → roboherd-0.1.11.dist-info}/RECORD

@@ -1,9 +1,10 @@
 roboherd/__init__.py,sha256=E89YtYyL77iJDJJkHS4Pn0JdTbmkmQS7Cp_XPqpRgx8,533
-roboherd/__main__.py,sha256=DMsxzSdO_tN0RazYcJicF1eZZzWT7NAYg3jXi7kEuU8,2963
-roboherd/register.py,sha256=Gqa5aT2supVJMj7z21btMYRma3_WW0oK5gjZftr048s,976
+roboherd/__main__.py,sha256=Bix6QLAXXs9zh3G8UOPHbq33QDhOCn4bNB7P0z1_9N0,3979
+roboherd/register.py,sha256=jnnHv-epeyXHcmmulxJoHqT7uWmb_efFnCbdxojKOBM,1044
 roboherd/test_validators.py,sha256=UiFgJkutmXBLjGpdLP2IBYCEf75quwnRHX7Z51G8Xqo,222
-roboherd/util.py,sha256=nrFRtyfvMQ8oXoGDYxmCAXcIN14dWBAPGv0cI_B_TP0,1710
+roboherd/util.py,sha256=gG8Scht1xRfauUOdL7GrLf9e3UO-j88X2wEdc1Ymm3o,1748
 roboherd/validators.py,sha256=2mc43ZGwFazp4f3B9J4RxZCU4Y_ErSNotib8MnYVtmY,140
+roboherd/version.py,sha256=nllDrH0jyChMuuYrK0CC55iTBKUNTUjejtcwxyUF2EQ,23
 roboherd/annotations/__init__.py,sha256=BrP5UWiVnh8FwmHc6wEY3G69BStidGUpF8uP4Ea5-dg,1140
 roboherd/annotations/bovine.py,sha256=cFLUSFzTulikNjZb1lLq5v4tGjL1i7_dR3H6wimegCg,2322
 roboherd/annotations/common.py,sha256=n5rm8WTgsg116RczKdzm_VoPg1mBk_IMf2V7l8UgWKQ,1351
@@ -25,20 +26,20 @@ roboherd/examples/moocow.py,sha256=2bZ8oZOpysuf7l3DP3zM4YiaALQHFbWpbP3h_Xlok6E,9
 roboherd/examples/number.py,sha256=c4zALxrKR5POuAUTmzAF4wHVgXMPfrP736N7dgn4xzo,1845
 roboherd/examples/rooster.py,sha256=ItPBpxhz72agkzFf13nv6csYVUJCZRabtUaGyKwUKRg,501
 roboherd/examples/scarecrow.py,sha256=jUnBaDJ2w9LQNIJAAfA53oJyRvKPJoKHtvHbeW-Pex0,626
-roboherd/herd/__init__.py,sha256=AtmpzoIbJwX82cx-9GdYMYQ5guj2vWb1dVYNGc1PReo,3126
+roboherd/herd/__init__.py,sha256=_JC0GK1sXSeY5nIwZ9Sd7XWuG6ZteHRcK3N8q5Awt3k,3825
 roboherd/herd/builder.py,sha256=MSVPRF0Jsxure9kdyCoYJHQ7nYilGAD0_uQaGQ-rQyE,619
 roboherd/herd/processor.py,sha256=4wZUp4xND9ZhxLIedUN7Yu3jxKWuuFqJent7wRiaAE0,1153
 roboherd/herd/scheduler.py,sha256=4zAPJkWs1ZGHlwTweHdhYb5xrHnpiSTpTEQ64drA-oE,1735
-roboherd/herd/test_herd.py,sha256=sQkzGCWdFveLklhaOJUybtl7odO-QOSDdd-_gan1py8,845
+roboherd/herd/test_herd.py,sha256=9Gf_Yh2a58EsSq3KYbR6IJBqM9JOueO6unzPwe-ciJ4,868
 roboherd/herd/test_scheduler.py,sha256=wLisqRMSl734P_rjbqMNH5WTQKepwihgr7ZC32nEj80,424
 roboherd/herd/types.py,sha256=_EidQbglm0jpsKX1EsL6U2qm_J5wCPhwUi6Avac22Ow,210
-roboherd/herd/manager/__init__.py,sha256=NqOJsp1CdAobjARJGmzvU1ceTW7j2bt0FdRbpM8iFUw,1464
+roboherd/herd/manager/__init__.py,sha256=7GcH0AZwLjNX7kLK-CXTvnaM3ZMWdstlTEqAY3IstZA,1440
 roboherd/herd/manager/config.py,sha256=8m-313_wn_-b4ivPSoPyyTQh0ONNLcd-Zz3pgWh8p1c,2429
 roboherd/herd/manager/load.py,sha256=BoeBID2UGP--sIKwITABQkQv2lMc9Y8pyp7_nleu2bw,351
 roboherd/herd/manager/test_config.py,sha256=I2EP7nEUdBVO8Wqot7SRhzp5UZFr5oGIuFS7cNB2iFk,1745
 roboherd/herd/manager/test_load.py,sha256=zyu5LIChMfTnxu_tYK63-bSOHYn1K1zUlbDY5DkE3GY,514
 roboherd/herd/manager/test_manager.py,sha256=9pSMaH7zmN-zagYCIBpQcV3Q0sBT7XZSCvsmLVC0rOI,1047
-roboherd-0.1.9.dist-info/METADATA,sha256=OPELZHKFq-g1XS1XqUzyvhE6ygMyTutw4YshGFpW3_8,2205
-roboherd-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-roboherd-0.1.9.dist-info/entry_points.txt,sha256=WebdVUmh8Ot-FupKJY6Du8LuFbmezt9yoy2UICqV3bE,52
-roboherd-0.1.9.dist-info/RECORD,,
+roboherd-0.1.11.dist-info/METADATA,sha256=7l6P2qnsbNozntECa2Otgggvx9TeMV_l5dmhpap8iIY,2206
+roboherd-0.1.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+roboherd-0.1.11.dist-info/entry_points.txt,sha256=WebdVUmh8Ot-FupKJY6Du8LuFbmezt9yoy2UICqV3bE,52
+roboherd-0.1.11.dist-info/RECORD,,