Changing output format in saving to the file methods

Author: jpatacz

skillcorner/client.py

@@ -302,6 +302,15 @@ def __new__(cls, classname, supers, cls_dict):
             if write_output_format_file_format_binding:
                 if 'df' in write_output_format_file_format_binding.keys():
                     write_output_format_file_format_binding['csv'] = write_output_format_file_format_binding.pop('df')
+                if 'python_structured_data' in write_output_format_file_format_binding.keys():
+                    if 'json' in write_output_format_file_format_binding['python_structured_data']:
+                        write_output_format_file_format_binding['json'] = write_output_format_file_format_binding.pop('python_structured_data')
+                    elif 'jsonl' in write_output_format_file_format_binding['python_structured_data']:
+                        write_output_format_file_format_binding['jsonl'] = write_output_format_file_format_binding.pop('python_structured_data')
+                if 'bytes' in write_output_format_file_format_binding.keys():
+                    write_output_format_file_format_binding.pop('bytes')
+                    write_output_format_file_format_binding['txt'] = ['fifa-data']
+                    write_output_format_file_format_binding['xml'] = ['fifa-xml']
             setattr(skcr_client,
                     get_and_save_method_name,
                     _freeze_args(skcr_client._get_and_write_data,
@@ -317,8 +326,11 @@ def __new__(cls, classname, supers, cls_dict):
                                      docstring.split(" in the ")[1]
             if default_output_format:
                 get_and_save_docstring = get_and_save_docstring.replace('df', 'csv')
-                get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'json')
                 get_and_save_docstring = get_and_save_docstring.replace('bytes', 'xml, txt')
+                if 'jsonl' in write_output_format_file_format_binding.keys():
+                    get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'jsonl')
+                elif 'json' in write_output_format_file_format_binding.keys():
+                    get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'json')
             cls_dict[get_and_save_method_name.strip("_")].__doc__ = get_and_save_docstring
 
         for key, value in METHOD_URL_ID_BINDING.items():
@@ -370,6 +382,15 @@ def __new__(cls, classname, supers, cls_dict):
             if write_output_format_file_format_binding:
                 if 'df' in write_output_format_file_format_binding.keys():
                     write_output_format_file_format_binding['csv'] = write_output_format_file_format_binding.pop('df')
+                if 'python_structured_data' in write_output_format_file_format_binding.keys():
+                    if 'json' in write_output_format_file_format_binding['python_structured_data']:
+                        write_output_format_file_format_binding['json'] = write_output_format_file_format_binding.pop('python_structured_data')
+                    elif 'jsonl' in write_output_format_file_format_binding['python_structured_data']:
+                        write_output_format_file_format_binding['jsonl'] = write_output_format_file_format_binding.pop('python_structured_data')
+                if 'bytes' in write_output_format_file_format_binding.keys():
+                    write_output_format_file_format_binding.pop('bytes')
+                    write_output_format_file_format_binding['txt'] = ['fifa-data']
+                    write_output_format_file_format_binding['xml'] = ['fifa-xml']
             setattr(skcr_client,
                     get_and_save_method_name,
                     _freeze_args(skcr_client._get_and_write_data_with_id,
@@ -390,8 +411,11 @@ def __new__(cls, classname, supers, cls_dict):
                                      docstring.split(" in the ")[1]
             if default_output_format:
                 get_and_save_docstring = get_and_save_docstring.replace('df', 'csv')
-                get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'json')
                 get_and_save_docstring = get_and_save_docstring.replace('bytes', "xml', 'txt")
+                if 'jsonl' in write_output_format_file_format_binding.keys():
+                    get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'jsonl')
+                elif 'json' in write_output_format_file_format_binding.keys():
+                    get_and_save_docstring = get_and_save_docstring.replace('python_structured_data', 'json')
 
             cls_dict[get_and_save_method_name.strip("_")].__doc__ = get_and_save_docstring