sovai 0.2.18__tar.gz → 0.2.20__tar.gz

{sovai-0.2.18 → sovai-0.2.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: sovai
-Version: 0.2.18
+Version: 0.2.20
 Summary: python for asset management
 License: Proprietary License
 Author: Derek Snow

{sovai-0.2.18 → sovai-0.2.20}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "sovai"
-version = "0.2.18"
+version = "0.2.20"
 description = "python for asset management"
 authors = ["Derek Snow <d.snow@outlook.com>"]
 license = "Proprietary License"

{sovai-0.2.18 → sovai-0.2.20}/sovai/get_data.py

@@ -620,7 +620,7 @@ def data(
                 verbose_mode.vprint(f"header: not pyarrow")
                 data = stream_data(res)
             
-            if HAS_FULL_INSTALL:
+            if HAS_FULL_INSTALL and not use_polars and not isinstance(data, CustomDataFrame):
                 data = CustomDataFrame(data)
             else:
                 data = pd.DataFrame(data)
@@ -631,13 +631,27 @@ def data(
         # Example usage in your existing code:
         if is_all(tickers):
             verbose_mode.vprint("All ticker Initialized")
-            url = res.text.strip('"')
-            verbose_mode.vprint(url, " (download link)")
-            data = read_parquet(url, use_polars=use_polars)
+            urls = [u.strip() for u in res.text.strip('"').split(',') if u.strip()]
+            data = None
+            for i, url in enumerate(urls):
+                verbose_mode.vprint(f"Attempting URL {i+1}: {url} (download link)")
+                try:
+                    data = read_parquet(url, use_polars=use_polars)
+                    verbose_mode.vprint(f"Successfully downloaded data from URL {i+1}")
+                    break
+                except Exception as e:
+                    verbose_mode.vprint(f"Failed to download from URL {i+1}: {str(e)}")
+            
+            if data is None:
+                raise Exception("Failed to download data from all provided URLs")
             
             # Apply filters
             data = filter_data(data, columns=columns, start_date=start_date, end_date=end_date, use_polars=use_polars)
 
+            # Wrap the result with CustomDataFrame if HAS_FULL_INSTALL is True
+            if HAS_FULL_INSTALL and not use_polars:
+                data = CustomDataFrame(data)
+
         verbose_mode.vprint("It reached the DF")
         try:
             from sovai.extensions.pandas_extensions import CustomDataFrame