Add typer equivalents for rattail commands

rattail_mailchimp/commands.py

@@ -2,7 +2,7 @@
 ################################################################################
 #
 #  Rattail -- Retail Software Framework
-#  Copyright © 2010-2021 Lance Edgar
+#  Copyright © 2010-2024 Lance Edgar
 #
 #  This file is part of Rattail.
 #
@@ -24,11 +24,31 @@
 Rattail Commands for MailChimp integration
 """
 
-from rattail import commands
+import typer
-from rattail.util import load_object
+
+from rattail.commands import rattail_typer, ImportSubcommand
+from rattail.commands.typer import importer_command, typer_get_runas_user
+from rattail.commands.importing import ImportCommandHandler
 
 
-class ImportMailChimp(commands.ImportSubcommand):
+@rattail_typer.command()
+@importer_command
+def import_mailchimp(
+        ctx: typer.Context,
+        **kwargs
+):
+    """
+    Import data to Rattail, from MailChimp API
+    """
+    config = ctx.parent.rattail_config
+    progress = ctx.parent.rattail_progress
+    handler = ImportCommandHandler(
+        config, import_handler_key='to_rattail.from_mailchimp.import')
+    kwargs['user'] = typer_get_runas_user(ctx)
+    handler.run(kwargs, progress=progress)
+
+
+class ImportMailChimp(ImportSubcommand):
     """
     Import data to Rattail, from MailChimp API
     """
@@ -43,4 +63,4 @@ class ImportMailChimp(commands.ImportSubcommand):
         else:
             # just use default, for sake of cmd line help
             spec = self.default_handler_spec
-        return load_object(spec)
+        return self.app.load_object(spec)

setup.cfg

@@ -38,5 +38,8 @@ zip_safe = False
 rattail.subcommands =
         import-mailchimp = rattail_mailchimp.commands:ImportMailChimp
 
+rattail.typer_imports =
+        rattail_mailchimp = rattail_mailchimp.commands
+
 rattail.emails =
         rattail_mailchimp = rattail_mailchimp.emails