datamule 1.1.1__py3-none-any.whl → 1.1.6__py3-none-any.whl

datamule/__init__.py

@@ -3,6 +3,8 @@ from .portfolio import Portfolio
 from .document import Document
 from .helper import _load_package_csv, load_package_dataset
 from .config import Config
+from .sheet import Sheet
+from .index import Index
 
 
 # Keep the notebook environment setup
@@ -32,5 +34,7 @@ __all__ = [
     'Portfolio',
     'Submission',
     'Document',
-    'Config'
+    'Config',
+    'Sheet',
+    'Index',
 ]

datamule/index.py

@@ -0,0 +1,62 @@
+from pathlib import Path
+from .sec.submissions.textsearch import query
+from .helper import _process_cik_and_metadata_filters, load_package_dataset
+
+class Index:
+    def __init__(self, path=None):
+        self.path = Path(path) if path else None
+        
+    def search_submissions(
+        self,
+        text_query,
+        start_date=None,
+        end_date=None,
+        submission_type=None,
+        cik=None,
+        ticker=None,
+        requests_per_second=5.0,
+        quiet=True,
+        **kwargs
+    ):
+        """
+        Search SEC filings for the given text query.
+        
+        Args:
+            text_query (str): Text to search for in SEC filings.
+            start_date (str or date, optional): Start date for filing search.
+            end_date (str or date, optional): End date for filing search.
+            submission_type (str, optional): Type of SEC submission to search.
+            cik (str, int, or list, optional): CIK(s) to filter by.
+            ticker (str or list, optional): Ticker(s) to filter by.
+            requests_per_second (float, optional): Rate limit for SEC API requests.
+            quiet (bool, optional): Whether to suppress output.
+            **kwargs: Additional filters to apply.
+            
+        Returns:
+            dict: Search results from the query function.
+        """
+        # Process CIK and ticker filters if provided
+        if cik is not None or ticker is not None:
+            cik_list = _process_cik_and_metadata_filters(cik, ticker, **kwargs)
+            # Add CIK filter to the query if we have results
+            if cik_list:
+                # Implementation note: Update as needed - this assumes your query function
+                # can accept a cik parameter, otherwise you may need additional logic here
+                kwargs['cik'] = cik_list
+            
+        # Execute the search query
+        results = query(
+            f'{text_query}',
+            filing_date=(start_date, end_date),
+            requests_per_second=requests_per_second,
+            quiet=quiet,
+            submission_type=submission_type,
+            **kwargs
+        )
+        
+        # Save results to path if specified
+        if self.path:
+            self._save_results(results, text_query)
+            
+        return results
+    

datamule/sec/submissions/monitor.py

@@ -5,7 +5,7 @@ from ..rss.monitor import start_monitor  # Import start_monitor directly
 import pytz
 
 
-async def _process_efts_hits(hits, collected_accession_numbers, data_callback=None):
+async def _process_efts_hits(hits, collected_accession_numbers, data_callback=None,rate_limiter=None):
     """Process EFTS hits, collect accession numbers, and call data callback."""
     processed_hits = []
     
@@ -36,7 +36,7 @@ async def _process_efts_hits(hits, collected_accession_numbers, data_callback=No
     
     # Call data callback if provided
     if data_callback and processed_hits:
-        await data_callback(processed_hits)
+        await data_callback(processed_hits, rate_limiter)
         
     return processed_hits
 
@@ -61,7 +61,7 @@ async def _master_monitor_impl(data_callback=None, poll_callback=None, submissio
     
     # Prepare a wrapper callback to collect accession numbers
     async def process_callback(hits):
-        await _process_efts_hits(hits, collected_accession_numbers, data_callback)
+         await _process_efts_hits(hits, collected_accession_numbers, data_callback, efts_query.limiter)
     
     # Create an EFTSQuery instance
     efts_query = EFTSQuery(requests_per_second=requests_per_second)

datamule/{book/book.py → sheet.py}

@@ -1,8 +1,8 @@
 from pathlib import Path
-from ..helper import _process_cik_and_metadata_filters, load_package_dataset
-from ..sec.xbrl.downloadcompanyfacts import download_company_facts
+from .helper import _process_cik_and_metadata_filters, load_package_dataset
+from .sec.xbrl.downloadcompanyfacts import download_company_facts
 
-class Book:
+class Sheet:
     def __init__(self, path):
         self.path = Path(path)
 

{datamule-1.1.1.dist-info → datamule-1.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: datamule
-Version: 1.1.1
+Version: 1.1.6
 Summary: Making it easier to use SEC filings.
 Home-page: https://github.com/john-friedman/datamule-python
 Author: John Friedman

{datamule-1.1.1.dist-info → datamule-1.1.6.dist-info}/RECORD

@@ -1,11 +1,11 @@
-datamule/__init__.py,sha256=0npnB3i2F7YB7etG315oDiCd-eMo-A6MP5LX2gQclHY,914
+datamule/__init__.py,sha256=l6YlwT5EeRxPlCtO5Jd4I8l266rSRUJyfFe97cRtSCM,991
 datamule/config.py,sha256=Y--CVv7JcgrjJkMOSLrvm2S8B9ost6RMSkGviP-MKtg,883
 datamule/document.py,sha256=BC8jdVy9pMOA9ghIqV5N2XJidmVNThqbBohsuSAnVoY,10813
 datamule/helper.py,sha256=xgOVnea-lUlQ5I-U0vYUp0VeKPNZehNhqjJvegA3lYE,3342
+datamule/index.py,sha256=0txvbzPcvY1GsdxA-wGdLzAByxSeE_1VyyBp9mZEQRM,2292
 datamule/portfolio.py,sha256=JmZlTrom_g7FXKXxWp_CiQTyC7p6_cDP08G0kFUja48,6982
+datamule/sheet.py,sha256=WwumRdniClGU7W3AXVLOpCdMnepLC7KMrRpQlA6_NUY,1022
 datamule/submission.py,sha256=JsxYlEz1Ywu6eC32OS15p4p-p8qB6SWd_rXuf2p5UfY,1247
-datamule/book/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-datamule/book/book.py,sha256=QWiowVNqb84o-JcVo0fpKumxnIbBge2ZeKwHxqkVMqw,1023
 datamule/mapping_dicts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamule/mapping_dicts/txt_mapping_dicts.py,sha256=DQPrGYbAPQxomRUtt4iiMGrwuF7BHc_LeFBQuYBzU9o,6311
 datamule/mapping_dicts/xml_mapping_dicts.py,sha256=Z22yDVwKYonUfM5foQP00dVDE8EHhhMKp0CLqVKV5OI,438
@@ -18,7 +18,7 @@ datamule/sec/rss/monitor.py,sha256=6r4EYaSlGu6VYErlj9zXJsIMLVie1cfacSZU-ESfuBI,1
 datamule/sec/submissions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamule/sec/submissions/downloader.py,sha256=HxbSkNotLLW6ROmU30rnXPlCo9gY3SoB1Z4ZWvj9FIY,2669
 datamule/sec/submissions/eftsquery.py,sha256=v6YMBZzksqweqHnNIllMFN-frWypAgvZPKx2FH1UrL4,22515
-datamule/sec/submissions/monitor.py,sha256=XkwH5nvzr_dNttmFRQ52m7344IKbOtWDfOZIEdie4H8,5234
+datamule/sec/submissions/monitor.py,sha256=F24I9yn1k8ggbCJQ-Vk7go_qJHlpkBzVKFYKDs_CWLs,5287
 datamule/sec/submissions/streamer.py,sha256=hc61le7gGIIWp1KEaOv_PhriUxf7YYFkQrSKELlZ3pg,9748
 datamule/sec/submissions/textsearch.py,sha256=oEIUrcO3HW-4dcyPCiOTvM7UUimNEM4HNIb-Juvc1BQ,4642
 datamule/sec/xbrl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -29,7 +29,7 @@ datamule/sec/xbrl/xbrlmonitor.py,sha256=TKFVfSyyUUfUgFQw4WxEVs4g8Nh-2C0tygNIRmTq
 datamule/seclibrary/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamule/seclibrary/downloader.py,sha256=Zb1TxsIz887tO3MJVP66siYVtNus89ti-g9oZ6VywrM,11500
 datamule/seclibrary/query.py,sha256=qGuursTERRbOGfoDcYcpo4oWkW3PCBW6x1Qf1Puiak4,7352
-datamule-1.1.1.dist-info/METADATA,sha256=VZQtm8MY--M1N08bbO90e6jS3893-g2V8hFCT_9PGrc,512
-datamule-1.1.1.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-datamule-1.1.1.dist-info/top_level.txt,sha256=iOfgmtSMFVyr7JGl_bYSTDry79JbmsG4p8zKq89ktKk,9
-datamule-1.1.1.dist-info/RECORD,,
+datamule-1.1.6.dist-info/METADATA,sha256=2kV79oqPrN2_HTcd4bxbJ0Q2_nP7Ta5Z79yskrST5S8,512
+datamule-1.1.6.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+datamule-1.1.6.dist-info/top_level.txt,sha256=iOfgmtSMFVyr7JGl_bYSTDry79JbmsG4p8zKq89ktKk,9
+datamule-1.1.6.dist-info/RECORD,,