o 5ti_*@svdZddlZddlmZddlmZmZddlZddlm Z ddl m Z m Z ddl mZdd lmZGd d d ZdS) zA .. codeauthor:: Tsuyoshi Hombashi N)chain)AnyFinal)logger) FormatAttr TableFormat)WriterNotFoundError)AbstractTableWriterc@speZdZdZedededefddZedededefdd Z ede efd d Z ede efd d Z dS)TableWriterFactoryz2 A factory class of table writer classes. file_extensionkwargsreturnc Kstj|d}t|r|}n|}|d}tD]"}||jvr#q|j t j @r*qt d|jd|jd i|Std|dddd d |g) a` Create a table writer class instance from a file extension. Supported file extensions are as follows: ================== =================================== Extension Writer Class ================== =================================== ``".adoc"`` :py:class:`~.AsciiDocTableWriter` ``".asciidoc"`` :py:class:`~.AsciiDocTableWriter` ``".asc"`` :py:class:`~.AsciiDocTableWriter` ``".css"`` :py:class:`~.CssTableWriter` ``".csv"`` :py:class:`~.CsvTableWriter` ``".htm"`` :py:class:`~.HtmlTableWriter` ``".html"`` :py:class:`~.HtmlTableWriter` ``".js"`` :py:class:`~.JavaScriptTableWriter` ``".json"`` :py:class:`~.JsonTableWriter` ``".jsonl"`` :py:class:`~.JsonLinesTableWriter` ``".ltsv"`` :py:class:`~.LtsvTableWriter` ``".ldjson"`` :py:class:`~.JsonLinesTableWriter` ``".md"`` :py:class:`~.MarkdownTableWriter` ``".ndjson"`` :py:class:`~.JsonLinesTableWriter` ``".py"`` :py:class:`~.PythonCodeTableWriter` ``".rst"`` :py:class:`~.RstGridTableWriter` ``".tsv"`` :py:class:`~.TsvTableWriter` ``".xls"`` :py:class:`~.ExcelXlsTableWriter` ``".xlsx"`` :py:class:`~.ExcelXlsxTableWriter` ``".sqlite"`` :py:class:`~.SqliteTableWriter` ``".sqlite3"`` :py:class:`~.SqliteTableWriter` ``".tsv"`` :py:class:`~.TsvTableWriter` ``".toml"`` :py:class:`~.TomlTableWriter` ``".yml"`` :py:class:`~.YamlTableWriter` ================== =================================== :param str file_extension: File extension string (case insensitive). :param kwargs: Keyword arguments that pass to a writer class constructor. :return: Writer instance that coincides with the ``file_extension``. :rtype: :py:class:`~pytablewriter.writer._table_writer.TableWriterInterface` :raises pytablewriter.WriterNotFoundError: |WriterNotFoundError_desc| the file extension. r. create a instance sz (unknown file extension).z#acceptable file extensions are: {}., N)ospathsplitexttypepyis_null_stringlstriplowerrfile_extensionsformat_attributer SECONDARY_EXTrdebug writer_classr joinformatget_extensions)clsr r ext table_formatrrJ/home/ubuntu/.local/lib/python3.10/site-packages/pytablewriter/_factory.pycreate_from_file_extensions(/    z-TableWriterFactory.create_from_file_extension format_namec Ks||}tD]#}||jvr)|jtj@s)|jdi|}td|j d|Sqt d |dd d | g) a Create a table writer class instance from a format name. Supported file format names are as follows: ============================================= =================================== Format name Writer Class ============================================= =================================== ``"adoc"`` :py:class:`~.AsciiDocTableWriter` ``"asciidoc"`` :py:class:`~.AsciiDocTableWriter` ``"css"`` :py:class:`~.CssTableWriter` ``"csv"`` :py:class:`~.CsvTableWriter` ``"elasticsearch"`` :py:class:`~.ElasticsearchWriter` ``"excel"`` :py:class:`~.ExcelXlsxTableWriter` ``"html"``/``"htm"`` :py:class:`~.HtmlTableWriter` ``"javascript"``/``"js"`` :py:class:`~.JavaScriptTableWriter` ``"json"`` :py:class:`~.JsonTableWriter` ``"json_lines"`` :py:class:`~.JsonLinesTableWriter` ``"latex_matrix"`` :py:class:`~.LatexMatrixWriter` ``"latex_table"`` :py:class:`~.LatexTableWriter` ``"ldjson"`` :py:class:`~.JsonLinesTableWriter` ``"ltsv"`` :py:class:`~.LtsvTableWriter` ``"markdown"``/``"md"`` :py:class:`~.MarkdownTableWriter` ``"mediawiki"`` :py:class:`~.MediaWikiTableWriter` ``"null"`` :py:class:`~.NullTableWriter` ``"pandas"`` :py:class:`~.PandasDataFrameWriter` ``"py"``/``"python"`` :py:class:`~.PythonCodeTableWriter` ``"rst"``/``"rst_grid"``/``"rst_grid_table"`` :py:class:`~.RstGridTableWriter` ``"rst_simple"``/``"rst_simple_table"`` :py:class:`~.RstSimpleTableWriter` ``"rst_csv"``/``"rst_csv_table"`` :py:class:`~.RstCsvTableWriter` ``"sqlite"`` :py:class:`~.SqliteTableWriter` ``"ssv"`` :py:class:`~.SpaceAlignedTableWriter` ``"tsv"`` :py:class:`~.TsvTableWriter` ``"toml"`` :py:class:`~.TomlTableWriter` ``"unicode"`` :py:class:`~.UnicodeTableWriter` ``"yaml"`` :py:class:`~.YamlTableWriter` ============================================= =================================== :param str format_name: Format name string (case insensitive). :param kwargs: Keyword arguments that pass to a writer class constructor. :return: Writer instance that coincides with the ``format_name``: :rtype: :py:class:`~pytablewriter.writer._table_writer.TableWriterInterface` :raises pytablewriter.WriterNotFoundError: |WriterNotFoundError_desc| for the format. rrrz (unknown format name).z acceptable format names are: {}.rNr)casefoldrnamesrrSECONDARY_NAMEr"rr! FORMAT_NAMEr r#r$get_format_names)r&r+r r(writerrrr)create_from_format_namebs 3  z*TableWriterFactory.create_from_format_namecCsttttddtDS)a4 :return: Available format names. :rtype: list :Example: .. code:: python >>> import pytablewriter as ptw >>> for name in ptw.TableWriterFactory.get_format_names(): ... print(name) ... adoc asciidoc bold_unicode borderless css csv elasticsearch excel htm html javascript js json json_lines jsonl latex_matrix latex_table ldjson ltsv markdown md mediawiki ndjson null numpy pandas pandas_pickle py python rst rst_csv rst_csv_table rst_grid rst_grid_table rst_simple rst_simple_table space_aligned sqlite ssv toml tsv unicode yaml css|]}|jVqdS)N)r-).0r(rrr) sz6TableWriterFactory.get_format_names..)sortedlistsetrr)r&rrr)r0s;z#TableWriterFactory.get_format_namescCs2t}tD] }|jD]}||q qtt|S)a# :return: Available file extensions. :rtype: list :Example: .. code:: python >>> import pytablewriter as ptw >>> for name in ptw.TableWriterFactory.get_extensions(): ... print(name) ... adoc asc asciidoc css csv htm html js json jsonl ldjson ltsv md ndjson py rst sqlite sqlite3 tex toml tsv xls xlsx yml )r7rraddr5r6)r&file_extension_setr(r rrr)r%s '   z!TableWriterFactory.get_extensionsN) __name__ __module__ __qualname____doc__ classmethodstrrr r*r2r6r0r%rrrr)r sKF<r )r=r itertoolsrtypingrrr_loggerr _table_formatrrerrorr r1r r rrrr)s