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

mindsdb/integrations/handlers/sentence_transformers_handler/__init__.py

@@ -18,4 +18,4 @@ icon_path = "icon.svg"
 permanent = False
 execution_method = "subprocess_keep"
 
-__all__ = ["Handler", "version", "name", "type", "title", "description", "import_error", "icon_path"]
+__all__ = ["Handler", "version", "name", "type", "title", "description", "import_error", "icon_path"]

mindsdb/integrations/handlers/sheets_handler/connection_args.py

@@ -17,4 +17,4 @@ connection_args = OrderedDict(
 connection_args_example = OrderedDict(
     spreadsheet_id='12wgS-1KJ9ymUM-6VYzQ0nJYGitONxay7cMKLnEE2_d0',
     sheet_name='iris'
-)
+)

mindsdb/integrations/handlers/shopify_handler/shopify_handler.py

@@ -13,6 +13,7 @@ from mindsdb.integrations.libs.api_handler_exceptions import InvalidNativeQuery,
 
 logger = log.getLogger(__name__)
 
+
 class ShopifyHandler(APIHandler):
     """
     The Shopify handler implementation.
@@ -30,7 +31,7 @@ class ShopifyHandler(APIHandler):
         super().__init__(name)
 
         if kwargs.get("connection_data") is None:
-            raise MissingConnectionParams(f"Incomplete parameters passed to Shopify Handler")
+            raise MissingConnectionParams("Incomplete parameters passed to Shopify Handler")
 
         connection_data = kwargs.get("connection_data", {})
         self.connection_data = connection_data
@@ -78,7 +79,7 @@ class ShopifyHandler(APIHandler):
             return self.connection
 
         if self.kwargs.get("connection_data") is None:
-            raise MissingConnectionParams(f"Incomplete parameters passed to Shopify Handler")
+            raise MissingConnectionParams("Incomplete parameters passed to Shopify Handler")
 
         api_session = shopify.Session(self.connection_data['shop_url'], '2021-10', self.connection_data['access_token'])
 
@@ -106,8 +107,8 @@ class ShopifyHandler(APIHandler):
             shopify.Shop.current()
             response.success = True
         except Exception as e:
-            logger.error(f'Error connecting to Shopify!')
-            raise ConnectionFailed(f"Conenction to Shopify failed.")
+            logger.error('Error connecting to Shopify!')
+            raise ConnectionFailed("Conenction to Shopify failed.")
             response.error_message = str(e)
 
         if self.yotpo_app_key is not None and self.yotpo_access_token is not None:
@@ -138,6 +139,6 @@ class ShopifyHandler(APIHandler):
         """
         try:
             ast = parse_sql(query)
-        except Exception as e:
+        except Exception:
             raise InvalidNativeQuery(f"The query {query} is invalid.")
-        return self.query(ast)
+        return self.query(ast)

mindsdb/integrations/handlers/shopify_handler/shopify_tables.py

@@ -57,7 +57,7 @@ class ProductsTable(APITable):
         products_df = select_statement_executor.execute_query()
 
         return products_df
-    
+
     def insert(self, query: ast.Insert) -> None:
         """Inserts data into the Shopify "POST /products" API endpoint.
 
@@ -78,12 +78,12 @@ class ProductsTable(APITable):
         insert_statement_parser = INSERTQueryParser(
             query,
             supported_columns=['title', 'body_html', 'vendor', 'product_type', 'tags', 'status'],
-            mandatory_columns=['title' ],
+            mandatory_columns=['title'],
             all_mandatory=False
         )
         product_data = insert_statement_parser.parse_query()
         self.create_products(product_data)
-    
+
     def delete(self, query: ast.Delete) -> None:
         """
         Deletes data from the Shopify "DELETE /products" API endpoint.
@@ -407,7 +407,7 @@ class OrdersTable(APITable):
         ----------
         query : ast.Insert
             Given SQL INSERT query
-        
+
         Returns
         -------
         None
@@ -420,26 +420,26 @@ class OrdersTable(APITable):
         insert_statement_parser = INSERTQueryParser(
             query,
             supported_columns=['address1_ba', 'address2_ba', 'city_ba', 'company_ba', 'country_ba',
-                               'country_code_ba', 'first_name_ba', 'last_name_ba', 'latitude_ba', 
-                               'longitude_ba', 'name_ba', 'phone_ba', 'province_ba', 'province_code_ba', 
-                               'zip_ba', 
+                               'country_code_ba', 'first_name_ba', 'last_name_ba', 'latitude_ba',
+                               'longitude_ba', 'name_ba', 'phone_ba', 'province_ba', 'province_code_ba',
+                               'zip_ba',
                                'address1_sa', 'address2_sa', 'city_sa', 'company_sa',
-                               'country_sa', 'country_code_sa', 'first_name_sa', 'last_name_sa', 
-                               'latitude_sa', 'longitude_sa', 'name_sa', 'phone_sa', 'province_sa', 
-                               'province_code_sa', 'zip_sa', 
+                               'country_sa', 'country_code_sa', 'first_name_sa', 'last_name_sa',
+                               'latitude_sa', 'longitude_sa', 'name_sa', 'phone_sa', 'province_sa',
+                               'province_code_sa', 'zip_sa',
                                'amount_dc', 'code_dc', 'type_dc',
-                               'gift_card_li', 'grams_li',  'price_li', 'quantity_li', 'title_li', 
-                               'vendor_li', 'fulfillment_status_li', 'sku_li', 'variant_title_li', 
+                               'gift_card_li', 'grams_li', 'price_li', 'quantity_li', 'title_li',
+                               'vendor_li', 'fulfillment_status_li', 'sku_li', 'variant_title_li',
                                'name_li', 'value_li',
                                'price_tl', 'rate_tl', 'title_tl', 'channel_liable_tl',
                                'name_na', 'value_na',
-                               'code_sl', 'price_sl', 'discounted_price_sl', 'source_sl', 
-                               'title_sl', 
-                               'carrier_identifier_sl', 'requested_fulfillment_service_id_sl', 
+                               'code_sl', 'price_sl', 'discounted_price_sl', 'source_sl',
+                               'title_sl',
+                               'carrier_identifier_sl', 'requested_fulfillment_service_id_sl',
                                'is_removed_sl',
-                               'buyer_accepts_marketing', 'currency', 'email', 'financial_status', 
-                               'fulfillment_status', 'note', 'phone', 'po_number', 'processed_at', 
-                               'referring_site', 'source_name', 'source_identifier', 'source_url', 
+                               'buyer_accepts_marketing', 'currency', 'email', 'financial_status',
+                               'fulfillment_status', 'note', 'phone', 'po_number', 'processed_at',
+                               'referring_site', 'source_name', 'source_identifier', 'source_url',
                                'tags', 'taxes_included', 'test', 'total_tax', 'total_weight'],
             mandatory_columns=['price_li', 'title_li'],
             all_mandatory=False
@@ -447,7 +447,6 @@ class OrdersTable(APITable):
         order_data = insert_statement_parser.parse_query()
         self.create_orders(order_data)
 
-
     def update(self, query: ast.Update) -> None:
         """Updates data in the Shopify "PUT /orders" API endpoint.
 
@@ -525,15 +524,15 @@ class OrdersTable(APITable):
         shopify.ShopifyResource.activate_session(api_session)
         # separate columns by API object
         line_items_columns = {'gift_card_li', 'grams_li', 'price_li', 'quantity_li', 'title_li',
-                             'vendor_li', 'fulfillment_status_li', 'sku_li', 'variant_title_li'}
+                              'vendor_li', 'fulfillment_status_li', 'sku_li', 'variant_title_li'}
         billing_address_columns = {'address1_ba', 'address2_ba', 'city_ba', 'company_ba',
-                                   'country_ba', 'country_code_ba', 'first_name_ba', 'last_name_ba', 
+                                   'country_ba', 'country_code_ba', 'first_name_ba', 'last_name_ba',
                                    'latitude_ba', 'longitude_ba', 'name_ba', 'phone_ba',
                                    'province_ba', 'province_code_ba', 'zip_ba'}
         shipping_address_columns = {'address1_sa', 'address2_sa', 'city_sa', 'company_sa',
-                                   'country_sa', 'country_code_sa', 'first_name_sa', 'last_name_sa',
-                                   'latitude_sa', 'longitude_sa', 'name_sa', 'phone_sa',
-                                   'province_sa', 'province_code_sa', 'zip_sa'}
+                                    'country_sa', 'country_code_sa', 'first_name_sa', 'last_name_sa',
+                                    'latitude_sa', 'longitude_sa', 'name_sa', 'phone_sa',
+                                    'province_sa', 'province_code_sa', 'zip_sa'}
         discount_codes_columns = {'amount_dc', 'code_dc', 'type_dc'}
         tax_lines_columns = {'price_tl', 'rate_tl', 'title_tl', 'channel_liable_tl'}
         note_attributes_columns = {'name_na', 'value_na'}
@@ -541,15 +540,15 @@ class OrdersTable(APITable):
                                   'source_sl', 'title_sl', 'carrier_identifier_sl',
                                   'requested_fulfillment_service_id_sl', 'is_removed_sl'}
         line_items_properties_columns = {'name_li', 'value_li'}
-        all_columns = (line_items_columns | billing_address_columns | shipping_address_columns |
-                       discount_codes_columns | tax_lines_columns | note_attributes_columns |
-                       shipping_lines_columns | line_items_properties_columns)
+        all_columns = (line_items_columns | billing_address_columns | shipping_address_columns
+                       | discount_codes_columns | tax_lines_columns | note_attributes_columns
+                       | shipping_lines_columns | line_items_properties_columns)
         modified_order_data = []
 
         for order in order_data:
             # separate values by object
             order_data_trimmed = {key: val for key, val in order.items()
-                                if key not in all_columns}
+                                  if key not in all_columns}
             line_items_data = OrdersTable._extract_data_helper(order, line_items_columns)
             billing_address_data = OrdersTable._extract_data_helper(order, billing_address_columns)
             shipping_address_data = OrdersTable._extract_data_helper(order, shipping_address_columns)
@@ -604,6 +603,7 @@ class OrdersTable(APITable):
         orders = shopify.Order.find(**kwargs)
         return [order.to_dict() for order in orders]
 
+
 class InventoryLevelTable(APITable):
     """The Shopify Inventory Table implementation"""
 
@@ -687,6 +687,7 @@ class InventoryLevelTable(APITable):
         inventories = shopify.InventoryLevel.find(**kwargs)
         return [inventory.to_dict() for inventory in inventories]
 
+
 class LocationTable(APITable):
     """The Shopify Location Table implementation"""
 
@@ -738,6 +739,7 @@ class LocationTable(APITable):
         locations = shopify.Location.find(**kwargs)
         return [location.to_dict() for location in locations]
 
+
 class CustomerReviews(APITable):
     """The Shopify Customer Reviews Table implementation"""
 
@@ -794,6 +796,7 @@ class CustomerReviews(APITable):
         json_response = requests.get(url, headers=headers).json()
         return [review for review in json_response['reviews']] if 'reviews' in json_response else []
 
+
 class CarrierServiceTable(APITable):
     """The Shopify carrier service Table implementation. Example carrier services like usps, dhl etc."""
 
@@ -831,11 +834,11 @@ class CarrierServiceTable(APITable):
             where_conditions,
             order_by_conditions
         )
-        
+
         carrier_service_df = select_statement_executor.execute_query()
 
         return carrier_service_df
-    
+
     def insert(self, query: ast.Insert) -> None:
         """Inserts data into the Shopify "POST /carrier_services" API endpoint.
 
@@ -895,7 +898,6 @@ class CarrierServiceTable(APITable):
         carrier_service_ids = carrier_services_df['id'].tolist()
         self.delete_carrier_services(carrier_service_ids)
 
-
     def update(self, query: ast.Update) -> None:
         """Updates data from the Shopify "PUT /carrier_services" API endpoint.
 
@@ -925,8 +927,6 @@ class CarrierServiceTable(APITable):
         carrier_service_ids = carrier_services_df['id'].tolist()
         self.update_carrier_service(carrier_service_ids, values_to_update)
 
-
-
     def get_columns(self) -> List[Text]:
         return ["id", "name", "active", "service_discovery", "carrier_service_type", "admin_graphql_api_id"]
 
@@ -935,7 +935,7 @@ class CarrierServiceTable(APITable):
         shopify.ShopifyResource.activate_session(api_session)
         services = shopify.CarrierService.find()
         return [service.to_dict() for service in services]
-    
+
     def create_carrier_service(self, carrier_service_data: List[Dict[Text, Any]]) -> None:
         api_session = self.handler.connect()
         shopify.ShopifyResource.activate_session(api_session)
@@ -968,7 +968,6 @@ class CarrierServiceTable(APITable):
         logger.info(f'Carrier Service {carrier_service_id} updated')
 
 
-
 class ShippingZoneTable(APITable):
     """The Shopify shipping zone Table implementation"""
 
@@ -1000,7 +999,6 @@ class ShippingZoneTable(APITable):
 
         shipping_zone_df = pd.json_normalize(self.get_shipping_zone(), record_path="countries", meta=["id", "name"], record_prefix="countries_")
 
-
         select_statement_executor = SELECTQueryExecutor(
             shipping_zone_df,
             selected_columns,
@@ -1008,7 +1006,7 @@ class ShippingZoneTable(APITable):
             order_by_conditions,
             result_limit
         )
-        
+
         shipping_zone_df = select_statement_executor.execute_query()
 
         return shipping_zone_df
@@ -1029,12 +1027,13 @@ class ShippingZoneTable(APITable):
         zones = shopify.ShippingZone.find()
         return [self.clean_response(zone.to_dict()) for zone in zones]
 
+
 class SalesChannelTable(APITable):
     """The Shopify Sales Channel Table implementation"""
 
     def select(self, query: ast.Select) -> pd.DataFrame:
         """Pulls data from the Shopify "GET /publication API endpoint, as Channel API endpoint is deprecated
-        
+
 
         Parameters
         ----------
@@ -1071,13 +1070,11 @@ class SalesChannelTable(APITable):
         sales_channel_df = select_statement_executor.execute_query()
         return sales_channel_df
 
-
     def get_columns(self) -> List[Text]:
         return pd.json_normalize(self.get_sales_channel(limit=1)).columns.tolist()
-    
 
     def get_sales_channel(self, **kwargs) -> List[Dict]:
         api_session = self.handler.connect()
         shopify.ShopifyResource.activate_session(api_session)
         sales_channels = shopify.Publication.find(**kwargs)
-        return  [sales_channel.to_dict() for  sales_channel in sales_channels]
+        return [sales_channel.to_dict() for sales_channel in sales_channels]

mindsdb/integrations/handlers/singlestore_handler/__about__.py

@@ -6,4 +6,4 @@ __author__ = 'MindsDB Inc'
 __github__ = 'https://github.com/mindsdb/mindsdb'
 __pypi__ = 'https://pypi.org/project/mindsdb/'
 __license__ = 'MIT'
-__copyright__ = 'Copyright 2022- mindsdb'
+__copyright__ = 'Copyright 2022- mindsdb'

mindsdb/integrations/handlers/singlestore_handler/__init__.py

@@ -11,7 +11,6 @@ from .__about__ import __version__ as version, __description__ as description
 
 title = 'SingleStore'
 name = 'singlestore'
-version = 0.1
 type = HANDLER_TYPE.DATA
 icon_path = 'icon.svg'
 

mindsdb/integrations/handlers/singlestore_handler/singlestore_handler.py

@@ -1,6 +1,7 @@
 from mindsdb.integrations.handlers.mysql_handler import Handler as MySQLHandler
 from .__about__ import __version__ as version
 
+
 class SingleStoreHandler(MySQLHandler):
     """
     This handler handles connection and execution of the SingleStore statements.

mindsdb/integrations/handlers/singlestore_handler/tests/test_singlestore_handler.py

@@ -27,10 +27,10 @@ class MySQLHandlerTest(unittest.TestCase):
     def test_2_get_tables(self):
         tbls = self.handler.get_tables()
         assert tbls['type'] is not RESPONSE_TYPE.ERROR
-    
+
     def test_5_drop_table(self):
         res = self.handler.native_query("DROP TABLE IF EXISTS test_mdb")
-        assert res['type'] is not RESPONSE_TYPE.ERROR 
+        assert res['type'] is not RESPONSE_TYPE.ERROR
 
     def test_4_create_table(self):
         res = self.handler.native_query("CREATE TABLE IF NOT EXISTS test_mdb (test_col INT)")
@@ -39,4 +39,4 @@ class MySQLHandlerTest(unittest.TestCase):
     def test_7_select_query(self):
         query = "SELECT * FROM test_mdb WHERE 'id'='a'"
         result = self.handler.native_query(query)
-        assert result['type'] is RESPONSE_TYPE.TABLE
+        assert result['type'] is RESPONSE_TYPE.TABLE

mindsdb/integrations/handlers/slack_handler/__init__.py

@@ -9,7 +9,7 @@ try:
     import_error = None
 except Exception as e:
     Handler = None
-    import_error = e 
+    import_error = e
 
 title = 'Slack'
 name = 'slack'
@@ -21,8 +21,8 @@ __all__ = [
     'version',
     'name',
     'type',
-    'title', 
+    'title',
     'description',
     'import_error',
     'icon_path'
-]
+]

mindsdb/integrations/handlers/snowflake_handler/requirements.txt

@@ -1,2 +1,2 @@
-snowflake-connector-python[pandas]==3.13.1
+snowflake-connector-python[pandas]==3.15.0
 snowflake-sqlalchemy==1.7.0

mindsdb/integrations/handlers/snowflake_handler/snowflake_handler.py

@@ -1,14 +1,16 @@
 import psutil
 import pandas
 from pandas import DataFrame
+from pandas.api import types as pd_types
 from snowflake.sqlalchemy import snowdialect
 from snowflake import connector
 from snowflake.connector.errors import NotSupportedError
+from snowflake.connector.cursor import SnowflakeCursor, ResultMetadata
 
-from mindsdb.utilities import log
 from mindsdb_sql_parser.ast.base import ASTNode
 from mindsdb_sql_parser.ast import Select, Identifier
 
+from mindsdb.utilities import log
 from mindsdb.integrations.libs.base import DatabaseHandler
 from mindsdb.utilities.render.sqlalchemy_render import SqlalchemyRender
 from mindsdb.integrations.libs.response import (
@@ -61,10 +63,105 @@ def _map_type(internal_type_name: str) -> MYSQL_DATA_TYPE:
         if internal_type_name in db_types_list:
             return mysql_data_type
 
-    logger.warning(f"Snowflake handler type mapping: unknown type: {internal_type_name}, use VARCHAR as fallback.")
+    logger.debug(f"Snowflake handler type mapping: unknown type: {internal_type_name}, use VARCHAR as fallback.")
     return MYSQL_DATA_TYPE.VARCHAR
 
 
+def _make_table_response(result: DataFrame, cursor: SnowflakeCursor) -> Response:
+    """Build response from result and cursor.
+    NOTE: Snowflake return only 'general' type in description, so look on result's
+          DF types and use types from description only if DF type is 'object'
+
+    Args:
+        result (DataFrame): result of the query.
+        cursor (SnowflakeCursor): cursor object.
+
+    Returns:
+        Response: response object.
+    """
+    description: list[ResultMetadata] = cursor.description
+    mysql_types: list[MYSQL_DATA_TYPE] = []
+    for column in description:
+        column_dtype = result[column.name].dtype
+        description_column_type = connector.constants.FIELD_ID_TO_NAME.get(column.type_code)
+        if pd_types.is_integer_dtype(column_dtype):
+            column_dtype_name = column_dtype.name
+            if column_dtype_name in ('int8', 'Int8'):
+                mysql_types.append(MYSQL_DATA_TYPE.TINYINT)
+            elif column_dtype in ('int16', 'Int16'):
+                mysql_types.append(MYSQL_DATA_TYPE.SMALLINT)
+            elif column_dtype in ('int32', 'Int32'):
+                mysql_types.append(MYSQL_DATA_TYPE.MEDIUMINT)
+            elif column_dtype in ('int64', 'Int64'):
+                mysql_types.append(MYSQL_DATA_TYPE.BIGINT)
+            else:
+                mysql_types.append(MYSQL_DATA_TYPE.INT)
+            continue
+        if pd_types.is_float_dtype(column_dtype):
+            column_dtype_name = column_dtype.name
+            if column_dtype_name in ('float16', 'Float16'):  # Float16 does not exists so far
+                mysql_types.append(MYSQL_DATA_TYPE.FLOAT)
+            elif column_dtype_name in ('float32', 'Float32'):
+                mysql_types.append(MYSQL_DATA_TYPE.FLOAT)
+            elif column_dtype_name in ('float64', 'Float64'):
+                mysql_types.append(MYSQL_DATA_TYPE.DOUBLE)
+            else:
+                mysql_types.append(MYSQL_DATA_TYPE.FLOAT)
+            continue
+        if pd_types.is_bool_dtype(column_dtype):
+            mysql_types.append(MYSQL_DATA_TYPE.BOOLEAN)
+            continue
+        if pd_types.is_datetime64_any_dtype(column_dtype):
+            mysql_types.append(MYSQL_DATA_TYPE.DATETIME)
+            series = result[column.name]
+            # snowflake use pytz.timezone
+            if series.dt.tz is not None and getattr(series.dt.tz, 'zone', 'UTC') != 'UTC':
+                series = series.dt.tz_convert('UTC')
+                result[column.name] = series.dt.tz_localize(None)
+            continue
+
+        if pd_types.is_object_dtype(column_dtype):
+            if description_column_type == 'TEXT':
+                # we can also check column.internal_size, if == 16777216 then it is TEXT, else VARCHAR(internal_size)
+                mysql_types.append(MYSQL_DATA_TYPE.TEXT)
+                continue
+            elif description_column_type == 'BINARY':
+                # if column.internal_size == 8388608 then BINARY, else VARBINARY(internal_size)
+                mysql_types.append(MYSQL_DATA_TYPE.BINARY)
+                continue
+            elif description_column_type == 'DATE':
+                mysql_types.append(MYSQL_DATA_TYPE.DATE)
+                continue
+            elif description_column_type == 'TIME':
+                mysql_types.append(MYSQL_DATA_TYPE.TIME)
+                continue
+
+        if description_column_type == 'FIXED':
+            if column.scale == 0:
+                mysql_types.append(MYSQL_DATA_TYPE.INT)
+            else:
+                # It is NUMBER, DECIMAL or NUMERIC with scale > 0
+                mysql_types.append(MYSQL_DATA_TYPE.FLOAT)
+            continue
+        elif description_column_type == 'REAL':
+            mysql_types.append(MYSQL_DATA_TYPE.FLOAT)
+            continue
+
+        mysql_types.append(MYSQL_DATA_TYPE.TEXT)
+
+    df = DataFrame(
+        result,
+        columns=[column.name for column in description],
+    )
+
+    return Response(
+        RESPONSE_TYPE.TABLE,
+        data_frame=df,
+        affected_rows=None,
+        mysql_types=mysql_types
+    )
+
+
 class SnowflakeHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the Snowflake statements.
@@ -214,10 +311,7 @@ class SnowflakeHandler(DatabaseHandler):
                                 raise MemoryError('Not enought memory')
                         # endregion
                     if len(batches) > 0:
-                        response = Response(
-                            RESPONSE_TYPE.TABLE,
-                            pandas.concat(batches, ignore_index=True)
-                        )
+                        response = _make_table_response(result=pandas.concat(batches, ignore_index=True), cursor=cur)
                     else:
                         response = Response(
                             RESPONSE_TYPE.TABLE,

mindsdb/integrations/handlers/solr_handler/connection_args.py

@@ -36,11 +36,11 @@ connection_args = OrderedDict(
 )
 
 connection_args_example = OrderedDict(
-    username = "demo_user",
-    password =  "demo_password",
-    host = "127.0.0.1",
-    port = 8981,
-    server_path = "solr",
-    collection = "gettingstarted",
-    use_ssl = False,
+    username="demo_user",
+    password="demo_password",
+    host="127.0.0.1",
+    port=8981,
+    server_path="solr",
+    collection="gettingstarted",
+    use_ssl=False,
 )

mindsdb/integrations/handlers/solr_handler/solr_handler.py

@@ -17,6 +17,7 @@ from mindsdb.integrations.libs.response import (
 
 logger = log.getLogger(__name__)
 
+
 class SolrHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the Solr SQL statements.
@@ -131,7 +132,7 @@ class SolrHandler(DatabaseHandler):
                     RESPONSE_TYPE.TABLE,
                     pd.DataFrame(
                         result,
-                        columns = columns
+                        columns=columns
                     )
                 )
             else:

mindsdb/integrations/handlers/solr_handler/tests/test_solr_handler.py

@@ -34,5 +34,6 @@ class SolrHandlerTest(unittest.TestCase):
         result = self.handler.query(query)
         assert result['type'] is RESPONSE_TYPE.TABLE
 
+
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

mindsdb/integrations/handlers/sqlany_handler/sqlany_handler.py

@@ -18,6 +18,7 @@ from mindsdb.integrations.libs.response import (
 
 logger = log.getLogger(__name__)
 
+
 class SQLAnyHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the SAP SQL Anywhere statements.
@@ -61,7 +62,7 @@ class SQLAnyHandler(DatabaseHandler):
             self.encryption = "SIMPLE"
         else:
             self.encryption = "NONE"
-        
+
         connection = sqlanydb.connect(
             host=self.host,
             userid=self.userid,
@@ -160,7 +161,7 @@ class SQLAnyHandler(DatabaseHandler):
         List all tables in SAP SQL Anywhere in the current schema
         """
 
-        return self.native_query(f"""
+        return self.native_query("""
             SELECT USER_NAME(ob.UID) AS SCHEMA_NAME
             , st.TABLE_NAME
             , st.TABLE_TYPE

mindsdb/integrations/handlers/sqlite_handler/sqlite_handler.py

@@ -20,6 +20,7 @@ from mindsdb.integrations.libs.response import (
 
 logger = log.getLogger(__name__)
 
+
 class SQLiteHandler(DatabaseHandler):
     """
     This handler handles connection and execution of the SQLite statements.

mindsdb/integrations/handlers/sqreamdb_handler/connection_args.py

@@ -20,7 +20,7 @@ connection_args = OrderedDict(
     port={
         'type': ARG_TYPE.INT,
         'description': 'Specify port to connect SQreamDB server'
-    }, 
+    },
     database={
         'type': ARG_TYPE.STR,
         'description': 'Specify database name  to connect SQreamDB server'