import snowflake.connector

import os
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives.asymmetric import dsa
from cryptography.hazmat.primitives import serialization

def snowflake_key_pair_connect():
    with open("/Users/rbomberg/.ssh/snowflake/rsa_key.p8", "rb") as key:
        p_key= serialization.load_pem_private_key(
            key.read(),
            #password=os.environ['PRIVATE_KEY_PASSPHRASE'].encode(),
            password="auto123auto".encode(),
            backend=default_backend()
        )

    pkb = p_key.private_bytes(
        encoding=serialization.Encoding.DER,
        format=serialization.PrivateFormat.PKCS8,
        encryption_algorithm=serialization.NoEncryption())

    ctx = snowflake.connector.connect(
        user='RBOMBERG',
        account='orchard',
        private_key=pkb,
        warehouse="DEV_OWS_WAREHOUSE",
        database="DEV_ENGINEERING",
        schema="EVENTS_STREAMS"
        )

    cur = ctx.cursor()
    return ctx, cur