datamule 0.415__cp39-cp39-macosx_10_9_universal2.whl

datamule/portfolio.py

@@ -0,0 +1,16 @@
+from pathlib import Path
+from .submission import Submission
+
+class Portfolio:
+    def __init__(self, path):
+        self.path = Path(path)
+        self.submissions = []
+        
+        # Load all subdirectories as submissions
+        for folder in self.path.iterdir():
+            if folder.is_dir():
+                self.submissions.append(Submission(folder))
+
+
+    def __iter__(self):
+        return iter(self.submissions)

datamule/submission.py

@@ -0,0 +1,61 @@
+from pathlib import Path
+import json
+from .document import Document
+
+class Submission:
+    def __init__(self, path):
+        self.path = Path(path)
+        self._load_metadata()
+    
+    def _load_metadata(self):
+        metadata_path = self.path / 'metadata.json'
+        with metadata_path.open('r') as f:
+            self.metadata = json.load(f)
+            
+    def keep(self, document_types):
+        """Keep files of specified document types, delete others
+        Args:
+            document_types: string or list of strings representing document types to keep
+        """
+        # Convert single string to list for consistent handling
+        if isinstance(document_types, str):
+            document_types = [document_types]
+            
+        for doc in self.metadata['documents']:
+            filename = doc.get('FILENAME')
+            if filename is None:
+                continue
+                
+            filepath = self.path / filename
+            # Delete if document type isn't in our keep list
+            if doc['TYPE'] not in document_types and filepath.exists():
+                filepath.unlink()
+                
+    def drop(self, document_types):
+        """Delete files of specified document types, keep others
+        Args:
+            document_types: string or list of strings representing document types to drop
+        """
+        # Convert single string to list for consistent handling
+        if isinstance(document_types, str):
+            document_types = [document_types]
+            
+        for doc in self.metadata['documents']:
+            filename = doc.get('FILENAME')
+            if filename is None:
+                continue
+                
+            filepath = self.path / filename
+            # Delete if document type is in our drop list
+            if doc['TYPE'] in document_types and filepath.exists():
+                filepath.unlink()
+
+    def document_type(self, document_type):
+        for doc in self.metadata['documents']:
+            if doc['TYPE'] == document_type:
+                filename = doc.get('FILENAME')
+                if filename is None:
+                    continue
+                    
+                document_path = self.path / filename
+                yield Document(doc['TYPE'], document_path)

datamule-0.415.dist-info/METADATA

@@ -0,0 +1,36 @@
+Metadata-Version: 2.1
+Name: datamule
+Version: 0.415
+Summary: Making it easier to use SEC filings.
+Home-page: https://github.com/john-friedman/datamule-python
+Author: John Friedman
+License: UNKNOWN
+Platform: UNKNOWN
+Requires-Dist: aiohttp
+Requires-Dist: aiolimiter
+Requires-Dist: tqdm
+Requires-Dist: requests
+Requires-Dist: nest-asyncio
+Requires-Dist: aiofiles
+Requires-Dist: polars
+Requires-Dist: setuptools
+Requires-Dist: selectolax
+Requires-Dist: pytz
+Requires-Dist: zstandard
+Provides-Extra: all
+Requires-Dist: pandas; extra == "all"
+Requires-Dist: google-generativeai; extra == "all"
+Requires-Dist: flask; extra == "all"
+Requires-Dist: psutil; extra == "all"
+Requires-Dist: openai; extra == "all"
+Provides-Extra: dataset_builder
+Requires-Dist: pandas; extra == "dataset-builder"
+Requires-Dist: google-generativeai; extra == "dataset-builder"
+Requires-Dist: psutil; extra == "dataset-builder"
+Provides-Extra: mulebot
+Requires-Dist: openai; extra == "mulebot"
+Provides-Extra: mulebot_server
+Requires-Dist: flask; extra == "mulebot-server"
+
+UNKNOWN
+

datamule-0.415.dist-info/RECORD

@@ -0,0 +1,37 @@
+datamule/__init__.py,sha256=8zx1hqAJeTxqmJ43zjJpJl8SRIRBx1zj8RNrsvYrQhE,1972
+datamule/document.py,sha256=Yn8UqUjKwYPE29MrMjreHK_HY9eTqOSjPyM5B1VBrHQ,5144
+datamule/helper.py,sha256=tr3AQWus9dHNZFKpLSglWjcb8zmm5qDXjOWACMhvMxQ,4594
+datamule/monitor.py,sha256=mRaM8v5NgcMF9DJ1s_YBzucjrbr-3yFwW422MVml-_Q,9114
+datamule/packageupdater.py,sha256=rw4hTnQGjZttee8QmTadg9vlhEjVp_dwdSFv8uQNvss,9584
+datamule/portfolio.py,sha256=piKibT_xZ9vZDpep1xS69eQceRrdq8FKqeRDf_-aQ38,425
+datamule/submission.py,sha256=aTpXH79vgVXzvpja6Kax-RdtyXiR7S9BwwkpMFmq7rA,2274
+datamule/data/company_former_names.csv,sha256=HE9cAv-_QKFX6jT-_-D0rHmaDyQuAzL4MJwank5O1U8,706380
+datamule/data/company_metadata.csv,sha256=yPovrCVjYwLWTU_hBUFJymp8iNO0NBYuq_QwOkRLoN8,3068599
+datamule/data/company_tickers.csv,sha256=GW6lOP54RiGJCx-d9N5jEBy7tGVgU3zI-5xHJXrZfSI,400363
+datamule/data/sec-glossary.csv,sha256=-cN7GjiadLw5C1sv4zSeCnfeZZDYeSgJl-0ydarMAo0,251209
+datamule/data/xbrl_descriptions.csv,sha256=SQ9wUURNqG424rnTiZtopsxV2q-PvU4NMj52LqgDsvg,2621524
+datamule/dataset_builder/dataset_builder.py,sha256=NCvNbDwlEkA_eAbqbsG--YlqPBDREFTVSM1GJquR0RE,9747
+datamule/mulebot/__init__.py,sha256=YvZXV6xQ0iP-oGD8rloufjdwJL6D46P3NNr0CY9PQCA,29
+datamule/mulebot/helper.py,sha256=olztOwltfELZ-IERM2bRNLBavD04kfB6ueWTisJAleA,1080
+datamule/mulebot/mulebot.py,sha256=XbtgvXBSFu9OaaLW_k1KDgHVTNQGV8_0ZwNMFad-pPU,5837
+datamule/mulebot/search.py,sha256=mwvbB6Fex5dEQkfxkCL53ne5pXdVno-5KlZ5vZyGnJQ,2073
+datamule/mulebot/tools.py,sha256=ctnGc2HItR-Roi-QXkc7GEaAOEYQiFRtfmdmIxNxYXk,2940
+datamule/mulebot/mulebot_server/__init__.py,sha256=x1QhXys7BWxi2g9_ZHUYA6S6rL3VL2718x4rYtGaaIg,33
+datamule/mulebot/mulebot_server/server.py,sha256=M7kU4aZUoi8X8DUKZNckLewBiI637Krbeap31qD2jt8,3547
+datamule/mulebot/mulebot_server/static/css/minimalist.css,sha256=Tz1tz8oF_esbfCvLTJBmTfb-5MIiqjfhU_4A4nto1mo,2974
+datamule/mulebot/mulebot_server/static/scripts/artifacts.js,sha256=WUAoI3LtEBEt3x-Ri0gwd6YT0JtGNwDZ_b8tuhWWSsg,2258
+datamule/mulebot/mulebot_server/static/scripts/chat.js,sha256=q8vV_KtzuNCXCfXqavM1HROIkYHItOAmaR8P1OjSqa0,3108
+datamule/mulebot/mulebot_server/static/scripts/filingArtifacts.js,sha256=wxeIM2RzF6Zh_9ivnYuNyTzIgIcEz0-zX8gTCvyACJo,2034
+datamule/mulebot/mulebot_server/static/scripts/listArtifacts.js,sha256=DZFLe-45mmzWvJPO1be5Ivfqx0BInrXfduQ1IhbHWzk,429
+datamule/mulebot/mulebot_server/static/scripts/main.js,sha256=NEIVih1WJeQ-qo5k8hnmgFHd7N839Mr6hJur856oXVQ,1882
+datamule/mulebot/mulebot_server/static/scripts/prefilledPrompt.js,sha256=mGhAXQnjnSxYqVqg1mE5g_ev0-aDhh849xunQtRchnY,1093
+datamule/mulebot/mulebot_server/static/scripts/suggestions.js,sha256=TCyz8OYuXeIG9qNRgwU2fhz18YNXpy4Bl9mk66lXefo,1795
+datamule/mulebot/mulebot_server/static/scripts/tableArtifacts.js,sha256=UtkUpLvELNI4Ibpb7VstgVA9Tk-8jbkxXhmXsgufFa4,4437
+datamule/mulebot/mulebot_server/static/scripts/utils.js,sha256=oGPMtyT9dvuqHqrfZj33t4vLZiF8UJrMXB1hpPXRNu4,1255
+datamule/mulebot/mulebot_server/templates/chat-minimalist.html,sha256=MsTbgpnLD0JCQiKKP3XeeNJRNsRqKsRa1j_XXW7nBKw,6975
+datamule/parser/sgml_parsing/sgml_parser_cy.c,sha256=UwXSuLGSBLKfO5bM7xTzjFLnGDV3-NNnCjgUCPAY1gk,796689
+datamule/parser/sgml_parsing/sgml_parser_cy.cpython-39-darwin.so,sha256=iHMgFC3SaB8jttskCj-7xvVZxk7D_Aq4A-hrsiAY0dc,362392
+datamule-0.415.dist-info/METADATA,sha256=UhfdwTuFt8AnoUVaA6lvfw6Jt_9vYsYIfh903yiaOCI,1051
+datamule-0.415.dist-info/WHEEL,sha256=Kpi8Y7ETJ76qV_mhWFgtod7iK9qT4KSd2dlvMqnPXEY,113
+datamule-0.415.dist-info/top_level.txt,sha256=iOfgmtSMFVyr7JGl_bYSTDry79JbmsG4p8zKq89ktKk,9
+datamule-0.415.dist-info/RECORD,,

datamule-0.415.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: bdist_wheel (0.45.1)
+Root-Is-Purelib: false
+Tag: cp39-cp39-macosx_10_9_universal2
+

datamule-0.415.dist-info/top_level.txt

@@ -0,0 +1 @@
+datamule