From 044eef6a55ef68b69447a93d0c4ef54f32d071e5 Mon Sep 17 00:00:00 2001 From: Elijas <4084885+Elijas@users.noreply.github.com> Date: Wed, 10 Apr 2024 00:26:05 +0300 Subject: [PATCH] feat(import): enable more ergonomic imports --- pyproject.toml | 2 +- sec_parser/processing_steps/__init__.py | 42 ++++++++++++++++++++++--- 2 files changed, 39 insertions(+), 5 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 30de29c..6bd1b88 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "sec-parser" -version = "0.55.1" +version = "0.55.2" description = "Parse SEC EDGAR HTML documents into a tree of elements that correspond to the visual structure of the document." authors = ["Alphanome.AI "] readme = "README.md" diff --git a/sec_parser/processing_steps/__init__.py b/sec_parser/processing_steps/__init__.py index 4e49ce3..b14d3be 100644 --- a/sec_parser/processing_steps/__init__.py +++ b/sec_parser/processing_steps/__init__.py @@ -12,15 +12,39 @@ AbstractProcessingStep, ) from sec_parser.processing_steps.empty_element_classifier import EmptyElementClassifier +from sec_parser.processing_steps.highlighted_text_classifier import ( + HighlightedTextClassifier, +) from sec_parser.processing_steps.image_classifier import ImageClassifier from sec_parser.processing_steps.individual_semantic_element_extractor.individual_semantic_element_extractor import ( IndividualSemanticElementExtractor, ) +from sec_parser.processing_steps.individual_semantic_element_extractor.single_element_checks.image_check import ( + ImageCheck, +) +from sec_parser.processing_steps.individual_semantic_element_extractor.single_element_checks.table_check import ( + TableCheck, +) +from sec_parser.processing_steps.individual_semantic_element_extractor.single_element_checks.top_section_title_check import ( + TopSectionTitleCheck, +) +from sec_parser.processing_steps.individual_semantic_element_extractor.single_element_checks.xbrl_tag_check import ( + XbrlTagCheck, +) +from sec_parser.processing_steps.introductory_section_classifier import ( + IntroductorySectionElementClassifier, +) +from sec_parser.processing_steps.page_header_classifier import PageHeaderClassifier +from sec_parser.processing_steps.page_number_classifier import PageNumberClassifier from sec_parser.processing_steps.supplementary_text_classifier import ( SupplementaryTextClassifier, ) from sec_parser.processing_steps.table_classifier import TableClassifier +from sec_parser.processing_steps.table_of_contents_classifier import ( + TableOfContentsClassifier, +) from sec_parser.processing_steps.text_classifier import TextClassifier +from sec_parser.processing_steps.text_element_merger import TextElementMerger from sec_parser.processing_steps.title_classifier import TitleClassifier from sec_parser.processing_steps.top_section_manager_for_10q import ( TopSectionManagerFor10Q, @@ -29,12 +53,22 @@ __all__ = [ "AbstractProcessingStep", "AbstractElementwiseProcessingStep", - "TextClassifier", - "TitleClassifier", + "EmptyElementClassifier", + "HighlightedTextClassifier", + "ImageCheck", "ImageClassifier", - "TableClassifier", "IndividualSemanticElementExtractor", + "IntroductorySectionElementClassifier", + "PageHeaderClassifier", + "PageNumberClassifier", "SupplementaryTextClassifier", - "EmptyElementClassifier", + "TableCheck", + "TableClassifier", + "TableOfContentsClassifier", + "TextClassifier", + "TextElementMerger", + "TitleClassifier", "TopSectionManagerFor10Q", + "TopSectionTitleCheck", + "XbrlTagCheck", ]