MindsDB 25.4.5.0__py3-none-any.whl → 25.5.4.0__py3-none-any.whl

mindsdb/integrations/handlers/discord_handler/discord_tables.py

@@ -76,7 +76,7 @@ class MessagesTable(APITable):
                     raise NotImplementedError(
                         f'Unsupported operator {op} for timestamp'
                     )
-            
+
             elif arg1 in ['author_id', 'author_username', 'author_global_name']:
                 if user_filter_flag:
                     raise NotImplementedError('Multiple user filters are not supported')
@@ -108,7 +108,7 @@ class MessagesTable(APITable):
 
         if not channel_filter_flag:
             raise NotImplementedError('Channel filter is required')
-        
+
         result = self.handler.call_discord_api(
             'get_messages', params=params, filters=filters
         )
@@ -183,7 +183,7 @@ class MessagesTable(APITable):
             try:
                 params = {'channel_id': message['channel_id'], 'text': message['text']}
                 self.handler.call_discord_api('send_message', params=params)
-                logger.info(f"Message sent to Discord channel successfully.")
+                logger.info("Message sent to Discord channel successfully.")
             except Exception as e:
                 logger.error(f"Error sending message to Discord channel: {e}")
                 raise e

mindsdb/integrations/handlers/discord_handler/tests/test_discord.py

@@ -1,16 +1,17 @@
 import unittest
 from unittest.mock import patch
-from mindsdb.integrations.handlers.discord_handler.discord_handler import DiscordHandler
 
 from mindsdb_sql_parser import ast
 from mindsdb_sql_parser.ast import BinaryOperation, Identifier, Constant
 from mindsdb_sql_parser.ast.select.star import Star
+from mindsdb.integrations.handlers.discord_handler.discord_handler import DiscordHandler
+
 
 class DiscordHandlerTest(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.handler = DiscordHandler(
-            name = 'discord_datasource', connection_data = {'token': 'test-discord-token'}
+            name='discord_datasource', connection_data={'token': 'test-discord-token'}
         )
 
     def test_0_check_connection(self):
@@ -28,7 +29,7 @@ class DiscordHandlerTest(unittest.TestCase):
             ),
         )
 
-        messages = self.handler._tables['messages'].select(query)
+        self.handler._tables['messages'].select(query)
         mock_get.assert_called_with(
             'https://discord.com/api/v10/channels/1234567890/messages',
             headers={
@@ -55,5 +56,6 @@ class DiscordHandlerTest(unittest.TestCase):
             json={'content': 'Test message'},
         )
 
+
 if __name__ == '__main__':
     unittest.main()

mindsdb/integrations/handlers/dockerhub_handler/dockerhub.py

@@ -41,15 +41,15 @@ class DockerHubClient:
     def get_repo_images(self, namespace, repo):
         url = f'{self.docker_hub_base_endpoint}namespaces/{namespace}/repositories/{repo}/images'
         return self.make_request(url)
-      
+
     def get_repo_tag(self, namespace, repo, tag):
         url = f'{self.docker_hub_base_endpoint}namespaces/{namespace}/repositories/{repo}/tags/{tag}'
         return self.make_request(url)
-      
+
     def get_repo_tags(self, namespace, repo):
         url = f'{self.docker_hub_base_endpoint}namespaces/{namespace}/repositories/{repo}/tags'
         return self.make_request(url)
-    
+
     def get_org_settings(self, namespace):
         url = f'{self.docker_hub_base_endpoint}orgs/{namespace}/settings'
         return self.make_request(url)

mindsdb/integrations/handlers/dockerhub_handler/dockerhub_handler.py

@@ -38,13 +38,13 @@ class DockerHubHandler(APIHandler):
 
         repo_images_stats_data = DockerHubRepoImagesSummaryTable(self)
         self._register_table("repo_images_summary", repo_images_stats_data)
-        
+
         repo_images_data = DockerHubRepoImagesTable(self)
         self._register_table("repo_images", repo_images_data)
 
         repo_tag_details_data = DockerHubRepoTagTable(self)
         self._register_table("repo_tag_details", repo_tag_details_data)
-        
+
         repo_tags_data = DockerHubRepoTagsTable(self)
         self._register_table("repo_tags", repo_tags_data)
 

mindsdb/integrations/handlers/dockerhub_handler/dockerhub_tables.py

@@ -100,7 +100,8 @@ class DockerHubRepoImagesSummaryTable(APITable):
             "active",
             "inactive"
         ]
-      
+
+
 class DockerHubOrgSettingsTable(APITable):
     """The DockerHub Repo Org Settings Table implementation"""
 
@@ -186,6 +187,7 @@ class DockerHubOrgSettingsTable(APITable):
             "restricted_images_allow_verified_publishers"
         ]
 
+
 class DockerHubRepoImagesTable(APITable):
     """The DockerHub Repo Images Table implementation"""
 
@@ -273,13 +275,13 @@ class DockerHubRepoImagesTable(APITable):
         """
 
         return ["namespace",
-            "repository",
-            "digest",
-            "tags",
-            "last_pushed",
-            "last_pulled",
-            "status"
-            ]
+                "repository",
+                "digest",
+                "tags",
+                "last_pushed",
+                "last_pulled",
+                "status"
+                ]
 
 
 class DockerHubRepoTagTable(APITable):
@@ -346,22 +348,22 @@ class DockerHubRepoTagTable(APITable):
 
         content = response["content"]
 
-        repo_tag_summary_df = pd.json_normalize({"creator": content["creator"], 
-            "id": content["id"], 
-            "images": content["images"], 
-            "last_updated": content["last_updated"], 
-            "last_updater": content["last_updater"], 
-            "last_updater_username": content["last_updater_username"], 
-            "name": content["name"], 
-            "repository": content["repository"], 
-            "full_size": content["full_size"], 
-            "v2": content["v2"], 
-            "tag_status": content["tag_status"], 
-            "tag_last_pulled": content["tag_last_pulled"], 
-            "tag_last_pushed": content["tag_last_pushed"], 
-            "media_type": content["media_type"], 
-            "content_type": content["media_type"]
-        })
+        repo_tag_summary_df = pd.json_normalize({"creator": content["creator"],
+                                                 "id": content["id"],
+                                                 "images": content["images"],
+                                                 "last_updated": content["last_updated"],
+                                                 "last_updater": content["last_updater"],
+                                                 "last_updater_username": content["last_updater_username"],
+                                                 "name": content["name"],
+                                                 "repository": content["repository"],
+                                                 "full_size": content["full_size"],
+                                                 "v2": content["v2"],
+                                                 "tag_status": content["tag_status"],
+                                                 "tag_last_pulled": content["tag_last_pulled"],
+                                                 "tag_last_pushed": content["tag_last_pushed"],
+                                                 "media_type": content["media_type"],
+                                                 "content_type": content["media_type"]
+                                                 })
 
         select_statement_executor = SELECTQueryExecutor(
             repo_tag_summary_df,
@@ -388,22 +390,23 @@ class DockerHubRepoTagTable(APITable):
             List of columns
         """
 
-        return ["creator", 
-                "id", 
-                "images", 
-                "last_updated", 
-                "last_updater", 
-                "last_updater_username", 
-                "name", 
-                "repository", 
-                "full_size", 
-                "v2", 
-                "tag_status", 
-                "tag_last_pulled", 
-                "tag_last_pushed", 
-                "media_type", 
+        return ["creator",
+                "id",
+                "images",
+                "last_updated",
+                "last_updater",
+                "last_updater_username",
+                "name",
+                "repository",
+                "full_size",
+                "v2",
+                "tag_status",
+                "tag_last_pulled",
+                "tag_last_pushed",
+                "media_type",
                 "content_type"
-            ]
+                ]
+
 
 class DockerHubRepoTagsTable(APITable):
     """The DockerHub Repo Tags Table implementation"""
@@ -491,19 +494,19 @@ class DockerHubRepoTagsTable(APITable):
             List of columns
         """
 
-        return ["creator", 
-                "id", 
-                "images", 
-                "last_updated", 
-                "last_updater", 
-                "last_updater_username", 
-                "name", 
-                "repository", 
-                "full_size", 
-                "v2", 
-                "tag_status", 
-                "tag_last_pulled", 
-                "tag_last_pushed", 
-                "media_type", 
+        return ["creator",
+                "id",
+                "images",
+                "last_updated",
+                "last_updater",
+                "last_updater_username",
+                "name",
+                "repository",
+                "full_size",
+                "v2",
+                "tag_status",
+                "tag_last_pulled",
+                "tag_last_pushed",
+                "media_type",
                 "content_type"
-            ]
+                ]

mindsdb/integrations/handlers/dremio_handler/__init__.py

@@ -1,7 +1,7 @@
 from mindsdb.integrations.libs.const import HANDLER_TYPE
 
 from .__about__ import __version__ as version, __description__ as description
-from .connection_args import connection_args, connection_args
+from .connection_args import connection_args
 try:
     from .dremio_handler import DremioHandler as Handler
     import_error = None
@@ -17,4 +17,4 @@ icon_path = 'icon.svg'
 __all__ = [
     'Handler', 'version', 'name', 'type', 'title', 'description',
     'connection_args', 'connection_args_example', 'import_error', 'icon_path'
-]
+]

mindsdb/integrations/handlers/druid_handler/__init__.py

@@ -17,4 +17,4 @@ icon_path = 'icon.svg'
 __all__ = [
     'Handler', 'version', 'name', 'type', 'title', 'description',
     'connection_args', 'connection_args_example', 'import_error', 'icon_path'
-]
+]

mindsdb/integrations/handlers/druid_handler/druid_handler.py

@@ -105,8 +105,8 @@ class DruidHandler(DatabaseHandler):
         need_to_close = self.is_connected is False
 
         try:
-            conn=self.connect()
-            conn.cursor().execute('select 1')  # raise exception if provided wrong credentials 
+            conn = self.connect()
+            conn.cursor().execute('select 1')  # raise exception if provided wrong credentials
 
             response.success = True
         except Exception as e:

mindsdb/integrations/handlers/edgelessdb_handler/tests/test_edgelessdb_handler.py

@@ -2,6 +2,7 @@ import unittest
 from mindsdb.integrations.handlers.edgelessdb_handler.edgelessdb_handler import EdgelessDBHandler
 from mindsdb.api.executor.data_types.response_type import RESPONSE_TYPE
 
+
 class EdgelessDBHandlerTest(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
@@ -18,7 +19,7 @@ class EdgelessDBHandlerTest(unittest.TestCase):
 
     def test_0_connect(self):
         assert self.handler.connect()
-    
+
     def test_1_drop_table(self):
         res = self.handler.query("DROP TABLE IF EXISTS TEST_TABLE")
         assert res.type is RESPONSE_TYPE.OK
@@ -26,20 +27,20 @@ class EdgelessDBHandlerTest(unittest.TestCase):
     def test_2_create_table(self):
         res = self.handler.query(
             '''CREATE TABLE TEST_TABLE (
-                ID INT PRIMARY KEY, 
+                ID INT PRIMARY KEY,
                 NAME VARCHAR(14)
                 )'''
-            )
+        )
         assert res.type is RESPONSE_TYPE.OK
 
     def test_3_insert(self):
         res = self.handler.query(
-            """INSERT INTO TEST_TABLE 
-            VALUES 
+            """INSERT INTO TEST_TABLE
+            VALUES
                 (100,'ONE HUNDRED'),
                 (200,'TWO HUNDRED'),
                 (300,'THREE HUNDRED')"""
-            )
+        )
         assert res.type is RESPONSE_TYPE.OK
 
     def test_4_select(self):
@@ -59,8 +60,7 @@ class EdgelessDBHandlerTest(unittest.TestCase):
 
     def test_8_disconnect(self):
         assert self.handler.disconnect()
-    
 
-        
+
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

mindsdb/integrations/handlers/email_handler/email_client.py

@@ -70,7 +70,7 @@ class EmailClient:
     def send_email(self, to_addr: str, subject: str, body: str):
         '''
         Sends an email to the given address.
-        
+
         Parameters:
             to_addr (str): The email address to send the email to.
             subject (str): The subject of the email.

mindsdb/integrations/handlers/email_handler/email_ingestor.py

@@ -15,6 +15,7 @@ class EmailIngestor:
     Parses emails into a DataFrame.
     Does some preprocessing on the raw HTML to extract meaningful text.
     """
+
     def __init__(self, email_client: EmailClient, search_options: EmailSearchOptions):
         self.email_client = email_client
         self.search_options = search_options
@@ -83,4 +84,3 @@ class EmailIngestor:
             df['datetime'] = pd.to_datetime(df['datetime'], utc=True, format="%a, %d %b %Y %H:%M:%S %z", errors='coerce')
 
         return df
-

mindsdb/integrations/handlers/email_handler/email_tables.py

@@ -1,5 +1,4 @@
 import datetime as dt
-import ast
 import pytz
 
 import pandas as pd

mindsdb/integrations/handlers/email_handler/settings.py

@@ -57,4 +57,3 @@ class EmailConnectionDetails(BaseModel):
             }
         }
         extra = "forbid"
-

mindsdb/integrations/handlers/eventstoredb_handler/eventstoredb_handler.py

@@ -8,9 +8,10 @@ from mindsdb.integrations.libs.response import (
     RESPONSE_TYPE
 )
 
-from .utils.helpers import *
+from .utils.helpers import get_auth_string, build_basic_url, build_health_url, build_stream_url, build_streams_url, build_next_url, entry_to_df, build_stream_url_last_event
 import requests
 import pandas as pd
+import re
 
 logger = log.getLogger(__name__)
 

mindsdb/integrations/handlers/firebird_handler/firebird_handler.py

@@ -166,7 +166,7 @@ class FirebirdHandler(DatabaseHandler):
         """
 
         query = """
-            SELECT RDB$RELATION_NAME 
+            SELECT RDB$RELATION_NAME
             FROM RDB$RELATIONS
             WHERE (RDB$SYSTEM_FLAG <> 1 OR RDB$SYSTEM_FLAG IS NULL) AND RDB$VIEW_BLR IS NULL
             ORDER BY RDB$RELATION_NAME;

mindsdb/integrations/handlers/flaml_handler/flaml_handler.py

@@ -12,26 +12,26 @@ class FLAMLHandler(BaseMLEngine):
     def create(self, target: str, df: Optional[pd.DataFrame] = None, args: Optional[Dict] = None) -> None:
         if args is None:
             args = {}
-            
+
         if df is not None:
             target_dtype = infer_types(df, 0).to_dict()["dtypes"][target]
             model = AutoML(verbose=0)
 
             if target_dtype in ['binary', 'categorical', 'tags']:
                 model.fit(X_train=df.drop(columns=[target]),
-                      y_train=df[target],
-                      task='classification',
-                      **args.get('using'))
+                          y_train=df[target],
+                          task='classification',
+                          **args.get('using'))
 
             elif target_dtype in ['integer', 'float', 'quantity']:
                 model.fit(X_train=df.drop(columns=[target]),
-                      y_train=df[target],
-                      task='regression',
-                      **args.get('using'))
-                
+                          y_train=df[target],
+                          task='regression',
+                          **args.get('using'))
+
             self.model_storage.json_set('args', args)
             self.model_storage.file_set('model', dill.dumps(model))
-            
+
         else:
             raise Exception(
                 "Data is empty!!"

mindsdb/integrations/handlers/frappe_handler/frappe_client.py

@@ -5,7 +5,7 @@ from typing import Dict, List
 
 class FrappeClient(object):
     """Client to interact with the Frappe API.
-    
+
     Attributes:
         domain (str): Path to Frappe domain to use (e.g. https://mindsdbfrappe.com).
         access_token (str): Frappe authorization token to use for all API requests.
@@ -25,7 +25,7 @@ class FrappeClient(object):
 
     def get_document(self, doctype: str, name: str) -> Dict:
         """Gets a document matching the given doctype from Frappe.
-        
+
         See https://frappeframework.com/docs/v14/user/en/api/rest#listing-documents
         Args:
             doctype (str): The document type to retrieve.
@@ -40,7 +40,7 @@ class FrappeClient(object):
 
     def get_documents(self, doctype: str, limit: int = None, fields: List[str] = None, filters: List[List] = None) -> List[Dict]:
         """Gets all documents matching the given doctype from Frappe.
-        
+
         See https://frappeframework.com/docs/v14/user/en/api/rest#listing-documents
         Args:
             doctype (str): The document type to retrieve.
@@ -83,7 +83,7 @@ class FrappeClient(object):
             data: Dict):
         """Creates a new document of the given doctype.
         See https://frappeframework.com/docs/v14/user/en/api/rest#listing-documents
-        
+
         Args:
             doctype (str): Type of the document to create.
             data (Dict): Document object.
@@ -99,7 +99,7 @@ class FrappeClient(object):
 
     def ping(self) -> bool:
         """Sends a basic request to the Frappe API to see if it succeeds.
-        
+
         Returns whether or not the connection to the Frappe API is valid.
         See https://frappeframework.com/docs/v14/user/en/api/rest#1-token-based-authentication
         """

mindsdb/integrations/handlers/frappe_handler/frappe_handler.py

@@ -15,6 +15,7 @@ from mindsdb_sql_parser import parse_sql
 
 logger = log.getLogger(__name__)
 
+
 class FrappeHandler(APIHandler):
     """A class for handling connections and interactions with the Frappe API.
 
@@ -36,9 +37,9 @@ class FrappeHandler(APIHandler):
         self.is_connected = False
 
         args = kwargs.get('connection_data', {})
-        if not 'access_token' in args:
+        if 'access_token' not in args:
             raise ValueError('"access_token" parameter required for authentication')
-        if not 'domain' in args:
+        if 'domain' not in args:
             raise ValueError('"domain" parameter required to connect to your Frappe instance')
         self.access_token = args['access_token']
         self.domain = args['domain']
@@ -52,8 +53,8 @@ class FrappeHandler(APIHandler):
             'register_sales_invoice': 'have to be used by assistant to register a sales invoice. Input is JSON object serialized as a string. Due date have to be passed in format: "yyyy-mm-dd".',
             'check_company_exists': 'useful to check the company is exist. Input is company',
             'check_expense_type': 'useful to check the expense_type is exist. Input is expense_type',
-            'check_customer':  'useful to check the customer is exist. Input is customer',
-            'check_item_code':  'have to be used to check the item code. Input is item_code',
+            'check_customer': 'useful to check the customer is exist. Input is customer',
+            'check_item_code': 'have to be used to check the item code. Input is item_code',
         }
         return {
             'tools': tools,
@@ -96,7 +97,7 @@ class FrappeHandler(APIHandler):
             self.client.post_document('Sales Invoice', invoice)
         except Exception as e:
             return f"Error: {e}"
-        return f"Success"
+        return "Success"
 
     def check_item_code(self, item_code):
         self.connect()

mindsdb/integrations/handlers/frappe_handler/frappe_tables.py

@@ -10,7 +10,7 @@ class FrappeDocumentsTable(APITable):
 
     def select(self, query: ast.Select) -> pd.DataFrame:
         """Selects data from the Frappe API and returns it as a pandas DataFrame.
-        
+
         Returns dataframe representing the Frappe API results.
 
         Args:
@@ -30,7 +30,7 @@ class FrappeDocumentsTable(APITable):
             else:
                 filters.append([arg1, op, arg2])
 
-        if not 'doctype' in params:
+        if 'doctype' not in params:
             raise ValueError('"doctype" parameter required')
 
         if query.limit:

mindsdb/integrations/handlers/github_handler/connection_args.py

@@ -26,7 +26,7 @@ connection_args = OrderedDict(
 )
 
 connection_args_example = OrderedDict(
-    repository="mindsdb/mindsdb", 
-    api_key="ghp_z91InCQZWZAMlddOzFCX7xHJrf9Fai35HT7", 
+    repository="mindsdb/mindsdb",
+    api_key="ghp_z91InCQZWZAMlddOzFCX7xHJrf9Fai35HT7",
     github_url="https://github.com/mindsdb/mindsdb"
 )

mindsdb/integrations/handlers/github_handler/github_handler.py

@@ -4,13 +4,7 @@ from mindsdb_sql_parser import parse_sql
 
 from mindsdb.integrations.handlers.github_handler.github_tables import (
     GithubIssuesTable,
-    GithubPullRequestsTable,
-    GithubCommitsTable,
-    GithubReleasesTable,
-    GithubBranchesTable,
-    GithubContributorsTable,
-    GithubMilestonesTable,
-    GithubProjectsTable, GithubFilesTable
+    GithubFilesTable
 )
 from mindsdb.integrations.handlers.github_handler.generate_api import get_github_types, get_github_methods, GHTable
 from mindsdb.integrations.libs.api_handler import APIHandler
@@ -59,7 +53,6 @@ class GithubHandler(APIHandler):
             table = GHTable(self, github_types=github_types, method=method)
             self._register_table(method.table_name, table)
 
-
     def connect(self) -> StatusResponse:
         """Set up the connection required by the handler.