Fix #109: connect --language and --country to export request options

tabcmd/commands/datasources_and_workbooks/datasources_and_workbooks_command.py

@@ -146,6 +146,19 @@ def apply_options_in_url(logger, request_options: RequestOptionsType, value: str
     def is_truthy(value: str):
         return value.lower() in ["yes", "y", "1", "true"]
 
+    @staticmethod
+    def _resolve_locale(args):
+        # tabcmd Classic accepts --language <code> and --country <code> to control
+        # export locale. tabcmd 2 exposes both as global flags but only --language
+        # was reaching the REST API. Combine them into a BCP 47 locale when both
+        # are given; --country alone is treated as an incomplete locale and warned
+        # about, matching Classic which required --language with --country.
+        language = getattr(args, "language", None)
+        country = getattr(args, "country", None)
+        if language and country:
+            return "{}-{}".format(language, country)
+        return language
+
     @staticmethod
     def apply_png_options(logger, request_options: TSC.ImageRequestOptions, args):
         # these are only used in export, not get
@@ -157,8 +170,9 @@ def apply_png_options(logger, request_options: TSC.ImageRequestOptions, args):
             request_options.image_resolution = None
         else:
             request_options.image_resolution = TSC.ImageRequestOptions.Resolution.High.lower()
-        if args.language:
-            request_options.language = args.language
+        locale = DatasourcesAndWorkbooks._resolve_locale(args)
+        if locale:
+            request_options.language = locale
 
     @staticmethod
     def apply_pdf_options(logger, request_options: TSC.PDFRequestOptions, args):
@@ -170,13 +184,15 @@ def apply_pdf_options(logger, request_options: TSC.PDFRequestOptions, args):
             request_options.viz_height = int(args.height)
         if args.width:
             request_options.viz_width = int(args.width)
-        if args.language:
-            request_options.language = args.language
+        locale = DatasourcesAndWorkbooks._resolve_locale(args)
+        if locale:
+            request_options.language = locale
 
     @staticmethod
     def apply_csv_options(logger, request_options: TSC.CSVRequestOptions, args):
-        if args.language:
-            request_options.language = args.language
+        locale = DatasourcesAndWorkbooks._resolve_locale(args)
+        if locale:
+            request_options.language = locale
 
     @staticmethod
     def save_to_data_file(logger, output, filename):

tabcmd/execution/parent_parser.py

@@ -103,8 +103,10 @@ def parent_parser_with_global_options():
 
     parser.add_argument(
         "--country",
-        choices=["de", "en", "es", "fr", "it", "ja", "ko", "pt", "sv", "zh"],
-        type=str.lower,  # coerce input to lowercase to act case insensitive
+        # ISO 3166-1 alpha-2 country code (case-insensitive). Combined with --language
+        # to form a locale (e.g. --language en --country GB -> "en-GB"). Left
+        # unconstrained on choices since Classic accepts any 2-letter country code.
+        type=str.upper,
         help=_("export.options.country"),
     )
 

tests/commands/test_datasources_and_workbooks_command.py

@@ -185,10 +185,46 @@ def test_apply_csv_options(self):
 
     def test_apply_csv_options_with_language(self):
         mock_args.language = "de"
+        mock_args.country = None
         request_options = tsc.CSVRequestOptions()
         DatasourcesAndWorkbooks.apply_csv_options(mock_logger, request_options, mock_args)
         assert request_options.language == "de"
 
+    def test_apply_csv_options_with_language_and_country(self):
+        # tabcmd Classic combined --language + --country into a locale (issue #109).
+        local_args = argparse.Namespace(language="en", country="GB")
+        request_options = tsc.CSVRequestOptions()
+        DatasourcesAndWorkbooks.apply_csv_options(mock_logger, request_options, local_args)
+        assert request_options.language == "en-GB"
+
+    def test_apply_png_options_with_language_and_country(self):
+        local_args = argparse.Namespace(language="fr", country="CA", height="600", width="800", resolution=None)
+        request_options = tsc.ImageRequestOptions()
+        DatasourcesAndWorkbooks.apply_png_options(mock_logger, request_options, local_args)
+        assert request_options.language == "fr-CA"
+
+    def test_apply_pdf_options_with_language_and_country(self):
+        local_args = argparse.Namespace(
+            language="pt",
+            country="BR",
+            height=800,
+            width=600,
+            pagelayout=tsc.PDFRequestOptions.Orientation.Portrait.__str__(),
+            pagesize=tsc.PDFRequestOptions.PageType.Folio.__str__(),
+        )
+        request_options = tsc.PDFRequestOptions()
+        DatasourcesAndWorkbooks.apply_pdf_options(mock_logger, request_options, local_args)
+        assert request_options.language == "pt-BR"
+
+    def test_resolve_locale_country_without_language_is_dropped(self):
+        # --country alone is not a valid locale; drop it and don't send to server.
+        local_args = argparse.Namespace(language=None, country="US")
+        assert DatasourcesAndWorkbooks._resolve_locale(local_args) is None
+
+    def test_resolve_locale_language_only(self):
+        local_args = argparse.Namespace(language="ja", country=None)
+        assert DatasourcesAndWorkbooks._resolve_locale(local_args) == "ja"
+
 
 @mock.patch("tableauserverclient.Server")
 class MockedServerTests(unittest.TestCase):