berryworld 1.0.0.179222__py3-none-any.whl → 1.0.0.180127__py3-none-any.whl

berryworld/credentials.py

@@ -253,3 +253,23 @@ class MicrosoftTeamsCredentials:
 
         except ValueError as e:
             raise ValueError("Variable %s not found" % str(e))
+
+
+class SnowflakeCredentials:
+    def __init__(self, db_name):
+        if db_name is None:
+            raise ValueError("Please provide a value for db_name")
+        self.db_name = db_name
+
+    def simple_creds(self):
+        try:
+            account = os.environ.get("SNOWFLAKE_" + self.db_name.upper() + '_ACCOUNT')
+            user_name = os.environ.get("SNOWFLAKE_" + self.db_name.upper() + '_USERNAME')
+            password = os.environ.get("SNOWFLAKE_" + self.db_name.upper() + '_PASSWORD')
+
+            return {
+                'account': account,
+                'user_name': user_name,
+                'password': password}
+        except ValueError as e:
+            raise ValueError("Variable %s not found" % str(e))

berryworld/snowflake_conn.py

@@ -0,0 +1,77 @@
+import pandas as pd
+import snowflake.connector
+from credentials import SnowflakeCredentials
+
+pd.set_option('future.no_silent_downcasting', True)
+
+
+class SnowflakeConn:
+    def __init__(self, server_creds=None, **kwargs):
+        """ Initialize the class
+        -----------------------------
+        server_creds = {
+                        "account": "",
+                        "db_name": "",
+                        "user_name": "",
+                        "password": ""
+                        }
+
+        -----------------------------
+        :param server_creds: Dictionary containing the info to connect to the Server
+        :param kwargs: Additional parameters to be passed to the connection
+        """
+
+        self.conn_str = None
+        self.conn = None
+
+        if kwargs != {}:
+            try:
+                db = kwargs['db_name']
+                server_creds = SnowflakeCredentials(db).simple_creds()
+            except KeyError:
+                raise KeyError('Please provide a valid db_name and server_type')
+
+        self.db = kwargs['db_name']
+        self.user = server_creds['user_name']
+        self.pw = server_creds['password']
+        self.account = server_creds['account']
+
+        if self.user and self.pw and self.account:
+            self.connect()
+
+    def connect(self):
+        """ Open the connection to Snowflake """
+        self.conn = snowflake.connector.connect(
+            user=self.user,
+            password=self.pw,
+            account=self.account,
+            database=self.db)
+
+    def close(self):
+        """ Close the connection to Snowflake """
+        self.conn.close()
+
+    def query(self, sql_query):
+        """ Read data from Snowflake according to the sql_query
+        -----------------------------
+        query_str = "SELECT * FROM %s" & table
+        con_.query(query_str)
+        -----------------------------
+        :param sql_query: Query to be sent to Snowflake
+        :return: DataFrame gathering the requested data
+        """
+        cursor = None
+        self.connect()
+        try:
+            cursor = self.conn.cursor()
+            cursor.execute(sql_query)
+            rows = cursor.fetchall()
+            col_names = [desc[0] for desc in cursor.description]
+            result = pd.DataFrame(rows, columns=col_names)
+            return result
+        except Exception as e:
+            raise Exception(e)
+        finally:
+            if cursor:
+                cursor.close()
+            self.close()

{berryworld-1.0.0.179222.dist-info → berryworld-1.0.0.180127.dist-info}/METADATA

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

{berryworld-1.0.0.179222.dist-info → berryworld-1.0.0.180127.dist-info}/RECORD

@@ -4,7 +4,7 @@ berryworld/allocation_solver.py,sha256=asFtaCAze6-eHUGWXA0kAp67UBS-Upj1KKdrVLj_t
 berryworld/app_logs.py,sha256=MKzKPYd3JuPfOQNAapIgaeZeFHw1z_w2mbn9I6QCADE,4180
 berryworld/app_logs_query.py,sha256=U94b-z3X9cuY_KFozupUcfaYciXWBn7p_RHkoRsfROU,4376
 berryworld/cache_data.py,sha256=AWAhV4PPkilZ4Xb1pUXuuu6UdHICAx8QqDtb9rVnjDM,2254
-berryworld/credentials.py,sha256=SeBLFqgadtUZclV1WbUC4uqI6xkcBlpV8miuifc8oqY,10290
+berryworld/credentials.py,sha256=L70ZzSNaS2mbHtSS8lb3iesd1XSwZNnUzGznlEKxlJY,11018
 berryworld/devops.py,sha256=BAsVonVwCXoApUOovkt-BCzwc6KnXjxRDGff_ejSGw8,9719
 berryworld/email_con.py,sha256=uSBzs_Ijz9pUPWt9e7U3TCB7i6q7hU1bB5vhsTB_tmw,14448
 berryworld/email_logging.py,sha256=LeSrTExhQhar49gJR2wGC1dS0lqsNpFl9pS3eYWqnuo,4936
@@ -17,6 +17,7 @@ berryworld/pickle_management.py,sha256=O49ojVtTqYCT510rVRTbZWWaur_-5q3HSVG03Azn8
 berryworld/postgres_connection.py,sha256=whKDnchd5Feqpmxpoh2vlyn36EKHR-dVEULYq0N_4wA,8287
 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=qd7kjeGDBUtMQqSfR0rCVz1Qn6d9U3zskBvfLRNSofc,2480
 berryworld/sql_conn.py,sha256=meDAf_92hNW5Zp5sv_PPeIUA8NDchaDtVN7zo77veIA,47491
 berryworld/teams_logging.py,sha256=8NwXyWr4fLj7W6GzAm2nRQCGFDxibQpAHDHHD24FrP8,6997
 berryworld/transportation_solver.py,sha256=gdlHRgl2wlp5EV0uhKIOYtkkbAlHaFHGktnMgvuENWE,5022
@@ -28,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.179222.dist-info/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
-berryworld-1.0.0.179222.dist-info/METADATA,sha256=QgD_R8cGkJU0mNXoC8WvrwuDxfgC3RG1hFsKsA1HLMo,1091
-berryworld-1.0.0.179222.dist-info/WHEEL,sha256=uCRv0ZEik_232NlR4YDw4Pv3Ajt5bKvMH13NUU7hFuI,91
-berryworld-1.0.0.179222.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
-berryworld-1.0.0.179222.dist-info/RECORD,,
+berryworld-1.0.0.180127.dist-info/LICENSE,sha256=vtkVCJM6E2af2gnsi2XxKPr4WY-uIbvzVLXieFND0UU,1074
+berryworld-1.0.0.180127.dist-info/METADATA,sha256=k_9Yaln2DC97Fes4tsW8U2GPrfZ9t2pzXDHOOCPhCcA,1091
+berryworld-1.0.0.180127.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+berryworld-1.0.0.180127.dist-info/top_level.txt,sha256=GIZ5qy-P5oxfEH755vA1IMFeTVdX3-40JxMe6nOe5I8,17
+berryworld-1.0.0.180127.dist-info/RECORD,,

{berryworld-1.0.0.179222.dist-info → berryworld-1.0.0.180127.dist-info}/WHEEL

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