berryworld 1.0.0.180563__py3-none-any.whl → 1.0.0.181358__py3-none-any.whl

berryworld/sql_conn.py

@@ -13,7 +13,8 @@ from .credentials import SQLCredentials
 class SQLConn:
     """ Connect to Microsoft SQL """
 
-    def __init__(self, server_creds=None, master=False, trusted_certificate=True, **kwargs):
+    def __init__(self, server_creds=None, master=False, trusted_certificate=True,
+                 multi_db=False, **kwargs):
         """ Initialize the class
         -----------------------------
         server_creds = {
@@ -30,11 +31,13 @@ class SQLConn:
         :param server_creds: Dictionary containing the info to connect to the Server
         :param master: Indicate whether the connection will be done to master or to a specific database
         :param trusted_certificate: Indicate whether the connection will be done using the TrustServerCertificate
+        :param multi_db: Indicate whether the connection will be done to a specific database or to multiple databases
         :param kwargs: Additional parameters to be passed to the connection
         """
         if (kwargs == {}) & (server_creds is None):
             raise ValueError('Please provide a valid server_creds or kwargs')
 
+        self.multi_db = multi_db
         self.master = master
         if trusted_certificate:
             self.trusted_certificate = '&TrustServerCertificate=yes'
@@ -80,12 +83,16 @@ class SQLConn:
         :param commit_as_transaction: Indicate whether the connection will be done using the autocommit option or not
         :return: The opened connection
         """
+        database = self.db_name
+        if self.multi_db:
+            database = str(self.db_name).lower().replace('primary;', '')
+
         if self.master:
             self.con_string = 'mssql+pyodbc://' + self.user_name + ':%s@' + self.server + '/master' + \
                               '?driver=' + self.driver + '&trusted_connection=yes' + self.trusted_certificate
             self.engine = sa.create_engine(self.con_string % parse.quote_plus(self.password))
         else:
-            self.con_string = 'mssql+pyodbc://' + self.user_name + ':%s@' + self.server + '/' + self.db_name + \
+            self.con_string = 'mssql+pyodbc://' + self.user_name + ':%s@' + self.server + '/' + database + \
                               '?driver=' + self.driver + self.trusted_certificate
             self.engine = sa.create_engine(self.con_string % parse.quote_plus(self.password))
         if not commit_as_transaction:

{berryworld-1.0.0.180563.dist-info → berryworld-1.0.0.181358.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: berryworld
-Version: 1.0.0.180563
+Version: 1.0.0.181358
 Summary: Handy classes to improve ETL processes
 Home-page: https://www.berryworld.com
 Author: BerryWorld ltd

{berryworld-1.0.0.180563.dist-info → berryworld-1.0.0.181358.dist-info}/RECORD

@@ -18,7 +18,7 @@ berryworld/postgres_connection.py,sha256=whKDnchd5Feqpmxpoh2vlyn36EKHR-dVEULYq0N
 berryworld/power_automate.py,sha256=9rDuRy0v-Ttq-SThid4lOB_tD4ibkyEmobiROpa--g4,25414
 berryworld/sharepoint_con.py,sha256=TuH-Vxk1VxjTi7x80KFssf_J8YPLRXpV27RBaFZi37U,22254
 berryworld/snowflake_conn.py,sha256=52ugfkyVCbR_Xr8YiZtNwPrwe93_1uE9uRy_VNPT6Gs,2428
-berryworld/sql_conn.py,sha256=meDAf_92hNW5Zp5sv_PPeIUA8NDchaDtVN7zo77veIA,47491
+berryworld/sql_conn.py,sha256=3_uPrn3rJjVr7-9wcXkh0CSitFD2ncQGGZjttFl6ELM,47803
 berryworld/teams_logging.py,sha256=8NwXyWr4fLj7W6GzAm2nRQCGFDxibQpAHDHHD24FrP8,6997
 berryworld/transportation_solver.py,sha256=gdlHRgl2wlp5EV0uhKIOYtkkbAlHaFHGktnMgvuENWE,5022
 berryworld/verify_keys.py,sha256=J2J505PcmBsQ9bj0XSRtXjpY-8qwpPU1A5LQdFRicFU,4257
@@ -29,8 +29,8 @@ tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 tests/test_allocation_config.py,sha256=e12l6fE9U57eSPS35g6ekJ_hol7-RHg89JV60_m1BlE,4633
 tests/test_handy_mix_config.py,sha256=Un56mz9KJmdn4K4OwzHAHLSRzDU1Xv2nFrONNuzOG04,2594
 tests/test_xml_parser.py,sha256=3QTlhFEd6KbK6nRFKZnc35tad6wqukTbe4QrFi8mr_8,859
-berryworld-1.0.0.180563.dist-info/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
-berryworld-1.0.0.180563.dist-info/METADATA,sha256=umOs8wBl5xHoyCgkIidPhIiJ6tgAcDNEXuIrV_psS_A,1174
-berryworld-1.0.0.180563.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-berryworld-1.0.0.180563.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
-berryworld-1.0.0.180563.dist-info/RECORD,,
+berryworld-1.0.0.181358.dist-info/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
+berryworld-1.0.0.181358.dist-info/METADATA,sha256=znF4x6wLc65ByBfuts_wxOdfDJYTrxRJgcsPbdFnwnc,1174
+berryworld-1.0.0.181358.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+berryworld-1.0.0.181358.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
+berryworld-1.0.0.181358.dist-info/RECORD,,

{berryworld-1.0.0.180563.dist-info → berryworld-1.0.0.181358.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.1.0)
+Generator: setuptools (75.2.0)
 Root-Is-Purelib: true
 Tag: py3-none-any