import snowflake.connector
import pandas as pd
import matplotlib.pyplot as plt
from matplotlib_venn import venn2
import numpy as np

USER = 'jpolikov'
PASSWORD = '<password>'
ACCOUNT = 'orchard'

# Connecting to Snowflake
cnx = snowflake.connector.connect(
  user=USER,
  password=PASSWORD,
  account=ACCOUNT,
)


cur = cnx.cursor()



artist_1 = (
"		select distinct user_id from facts.prod.fact_demographics		"
"		where artistid = 802380										"
"		and labelid = 23495												"
)

artist_2 = (
"		select distinct user_id from facts.prod.fact_demographics		"
"		where artistid = 757729										"
"		and labelid = 21989												"												
)


output1 = (pd.read_sql(artist_1, cnx))
# output1['USER_ID'] = output1['USER_ID'].astype(str).str.split(',')

output2 = (pd.read_sql(artist_2, cnx))
# output2['USER_ID'] = output2['USER_ID'].astype(str).str.split(',')


import pprint
pprint.pprint(output1)
pprint.pprint(output2)


set1 = set(output1['USER_ID'])
set2 = set(output2['USER_ID'])

venn2([set1, set2], ('Sofi Tukker', 'Darlingside'))

plt.show()