datamule 2.0.4__py3-none-any.whl → 2.0.5__py3-none-any.whl

datamule/datamule/datamule_mysql_rds.py

@@ -77,8 +77,14 @@ class DatamuleMySQL:
             if value is None:
                 continue
                 
+            # *** HIGHLIGHTED CHANGE: Special logic for members in simple_xbrl table ***
+            if table == 'simple_xbrl' and key == 'members':
+                if isinstance(value, list):
+                    filters[key] = {"type": "find_in_set", "values": value}
+                else:
+                    filters[key] = {"type": "find_in_set", "values": [value]}
             # Special logic for cik
-            if key == 'cik':
+            elif key == 'cik':
                 if isinstance(value, list):
                     value = [int(val) for val in value]
                 else:
@@ -100,6 +106,12 @@ class DatamuleMySQL:
         for key, filter_obj in filters.items():
             if filter_obj["type"] == "range":
                 query_desc.append(f"{key}={filter_obj['values'][0]} to {filter_obj['values'][1]}")
+            # *** HIGHLIGHTED CHANGE: Display logic for find_in_set type ***
+            elif filter_obj["type"] == "find_in_set":
+                if len(filter_obj["values"]) == 1:
+                    query_desc.append(f"{key} contains {filter_obj['values'][0]}")
+                else:
+                    query_desc.append(f"{key} contains any of {filter_obj['values']}")
             elif len(filter_obj["values"]) == 1:
                 query_desc.append(f"{key}={filter_obj['values'][0]}")
             else:
@@ -177,6 +189,7 @@ def query_mysql_rds(table, api_key=None, **kwargs):
     Parameters:
     - table: Table name (e.g., 'simple_xbrl')
     - cik: Company CIK number(s), can be int, string, or list
+    - members: For simple_xbrl table, search within comma-separated member strings
     - Any other filter parameters as keyword arguments
     - page_size: Number of records per page (max 25000, default 25000)
     - quiet: Boolean, whether to suppress progress output and summary (default False)
@@ -186,6 +199,7 @@ def query_mysql_rds(table, api_key=None, **kwargs):
     - Single value: Exact match
     - List: OR condition (any of the values)
     - Tuple: Range condition (between first and second values)
+    - members (simple_xbrl only): Searches within comma-separated strings using FIND_IN_SET
     
     Returns:
     - List of dictionaries containing the requested data (ready for pandas DataFrame)
@@ -224,8 +238,14 @@ def _query_mysql_rds_single(table, api_key=None, **kwargs):
         if value is None:
             continue
             
+        # *** HIGHLIGHTED CHANGE: Special logic for members in simple_xbrl table ***
+        if table == 'simple_xbrl' and key == 'members':
+            if isinstance(value, list):
+                filters[key] = {"type": "find_in_set", "values": value}
+            else:
+                filters[key] = {"type": "find_in_set", "values": [value]}
         # special logic for cik
-        if key == 'cik':
+        elif key == 'cik':
             if isinstance(value, list):
                 value = [int(val) for val in value]
             else:

{datamule-2.0.4.dist-info → datamule-2.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: datamule
-Version: 2.0.4
+Version: 2.0.5
 Summary: Work with SEC submissions at scale.
 Home-page: https://github.com/john-friedman/datamule-python
 Author: John Friedman

{datamule-2.0.4.dist-info → datamule-2.0.5.dist-info}/RECORD

@@ -10,7 +10,7 @@ datamule/submission.py,sha256=qcb5TogrB2q6x4zcGPKFf4dkrAy0bAPzY71Ops_xW44,14437
 datamule/data/listed_filer_metadata.csv,sha256=dT9fQ8AC5P1-Udf_UF0ZkdXJ88jNxJb_tuhi5YYL1rc,2426827
 datamule/datamule/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamule/datamule/datamule_lookup.py,sha256=e8djAg-ctSyHiKk7BjbtgugZ3p8roUjzsym5z3AihUg,9468
-datamule/datamule/datamule_mysql_rds.py,sha256=Oj_xPTBKkzWsuRlb_tphjJrBW1eua1cOuxjGwJx581k,10591
+datamule/datamule/datamule_mysql_rds.py,sha256=P5vL3RJnOwLz25hPKuoYmxSX7XeDe83YETnwT6miRMo,11858
 datamule/datamule/downloader.py,sha256=aTyVUuIwynPtHB0Z9BvCasy9Ao5wfHptNAsjN-7yDTk,18525
 datamule/datamule/sec_connector.py,sha256=VwOaODpHoAWy8JIky6kLR1-orW_PB61RHw7pIGRpkow,3288
 datamule/document/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -65,7 +65,7 @@ datamule/seclibrary/bq.py,sha256=C8sb_rpXTvchprrFLcbRar4Qi0XWW25tnv1YsHSS5o4,180
 datamule/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 datamule/utils/construct_submissions_data.py,sha256=NB_hvfxlRXPyt4Fgc-5qA8vJRItkLhBedCSTaxwW7Jg,5887
 datamule/utils/format_accession.py,sha256=60RtqoNqoT9zSKVb1DeOv1gncJxzPTFMNW4SNOVmC_g,476
-datamule-2.0.4.dist-info/METADATA,sha256=CNSfwZgqLh3WR4TObVkP8Y1p2wWx4To_NMthG4EvhEQ,560
-datamule-2.0.4.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-datamule-2.0.4.dist-info/top_level.txt,sha256=iOfgmtSMFVyr7JGl_bYSTDry79JbmsG4p8zKq89ktKk,9
-datamule-2.0.4.dist-info/RECORD,,
+datamule-2.0.5.dist-info/METADATA,sha256=UzOW91CX56mST6_QMXKrYeewjnylQc8_B0gyK7fcpc8,560
+datamule-2.0.5.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+datamule-2.0.5.dist-info/top_level.txt,sha256=iOfgmtSMFVyr7JGl_bYSTDry79JbmsG4p8zKq89ktKk,9
+datamule-2.0.5.dist-info/RECORD,,