dbhydra 2.1.1__tar.gz → 2.1.2__tar.gz

{dbhydra-2.1.1 → dbhydra-2.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbhydra
-Version: 2.1.1
+Version: 2.1.2
 Summary: Data science friendly ORM combining Python
 Home-page: https://github.com/DovaX/dbhydra
 Author: DovaX

{dbhydra-2.1.1 → dbhydra-2.1.2}/dbhydra/src/abstract_db.py

@@ -1,10 +1,12 @@
 import abc
 import threading
 from contextlib import contextmanager
+from typing import Optional
 
 from dbhydra.src.migrator import Migrator
 from dbhydra.src.tables import AbstractTable
 
+
 def read_connection_details(config_file):
     def read_file(file):
         """Reads txt file -> list"""
@@ -102,6 +104,7 @@ class AbstractDb(abc.ABC):
         # self.connect_to_db()
         
         self.active_transactions=[]
+        self.last_table_inserted_into: Optional[str] = None
 
     @abc.abstractmethod
     def connect_locally(self):

{dbhydra-2.1.1 → dbhydra-2.1.2}/dbhydra/src/abstract_table.py

@@ -357,6 +357,8 @@ class AbstractTable(AbstractJoinable, abc.ABC):
 
         return df_copy
 
+    def extract_last_id(self) -> Any:
+        raise NotImplementedError("Method not implemented for this subclass")
 
     def insert_from_df(self, df, batch=1, try_mode=False, debug_mode=False, adjust_df=False, insert_id=False):
         if debug_mode:
@@ -403,7 +405,9 @@ class AbstractTable(AbstractJoinable, abc.ABC):
         #             rows[i][j] = "'" + record + "'"
         #print(rows)
         rows = df.values.tolist()
-        return self.insert(rows, batch=batch, try_mode=try_mode, debug_mode=False, insert_id=insert_id)
+        result = self.insert(rows, batch=batch, try_mode=try_mode, debug_mode=False, insert_id=insert_id)
+        self.db1.last_table_inserted_into = self.name
+        return result
 
     #TODO: need to solve inserting in different column_order
     #check df column names, permute if needed

{dbhydra-2.1.1 → dbhydra-2.1.2}/dbhydra/src/tables.py

@@ -675,6 +675,18 @@ class MysqlTable(AbstractTable):
 
         return python_types
 
+    def extract_last_id(self) -> Any:
+        """
+        Extract the last inserted ID from the DB.
+
+        LAST_INSERT_ID exists in the DB connection context, therefore is safe to use if DB session is request-scoped
+        In this case we only use global connection, but we use Lock to ensure thread-safety across different requests
+        This is a go-to mechanism for extracting the ID of the inserted record for multiple SQL DBs,
+        altho this specific query is applicable only to MySQL.
+        """
+        assert self.name == self.db1.last_table_inserted_into, "Last table inserted into is not the same as the table being queried"
+        return self.select("SELECT LAST_INSERT_ID()")[0][0]
+
     def get_nullable_columns(self):
         information_schema_table = MysqlTable(self.db1, 'INFORMATION_SCHEMA.COLUMNS')
         query = f"SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS where TABLE_SCHEMA = '{self.db1.DB_DATABASE}' and TABLE_NAME = '{self.name}' and IS_NULLABLE = 'YES'"
@@ -1014,6 +1026,7 @@ class XlsxTable(AbstractTable):
         df.reset_index(drop=True, inplace=True)
 
         self._save_table(df)
+        self.last_table_inserted_into = self.name
 
     def replace_from_df(self, df):
         assert len(df.columns) == len(self.columns)  # +1 because of id column
@@ -1113,4 +1126,9 @@ class XlsxTable(AbstractTable):
             df.drop(df[df[where_variable] == where_value].index, inplace=True)
             deleted_count = number_of_records - len(df)
         self.replace_from_df(df)
-        return deleted_count
+        return deleted_count    
+
+    def extract_last_id(self) -> Any:
+        assert self.name == self.db1.last_table_inserted_into, "Last table inserted into is not the same as the table being queried"
+        df = self.select_to_df()
+        return df.iloc[-1][self.id_column_name]

{dbhydra-2.1.1 → dbhydra-2.1.2}/dbhydra/src/xlsx_db.py

@@ -1,10 +1,11 @@
-from dbhydra.src.abstract_db import AbstractDb
-from dbhydra.src.tables import XlsxTable
-
 import contextlib
-import threading
-import pathlib
 import os
+import pathlib
+import threading
+from typing import Optional
+
+from dbhydra.src.abstract_db import AbstractDb
+from dbhydra.src.tables import XlsxTable
 
 
 class XlsxDb(AbstractDb):
@@ -29,7 +30,8 @@ class XlsxDb(AbstractDb):
         if self.db_directory_path is None:
             self.db_directory_path = pathlib.Path(self.name) 
             
-            
+        self.last_table_inserted_into: Optional[str] = None
+
         self.python_database_type_mapping = {
         'int': "int",
         'float': "double",

{dbhydra-2.1.1 → dbhydra-2.1.2}/dbhydra.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: dbhydra
-Version: 2.1.1
+Version: 2.1.2
 Summary: Data science friendly ORM combining Python
 Home-page: https://github.com/DovaX/dbhydra
 Author: DovaX

{dbhydra-2.1.1 → dbhydra-2.1.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
     
 setuptools.setup(
     name='dbhydra',
-    version='2.1.1',
+    version='2.1.2',
     author='DovaX',
     author_email='dovax.ai@gmail.com',
     description='Data science friendly ORM combining Python',