File size: 701 Bytes
4e9265f f69b0e1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 | import os
import streamlit as st
from dotenv import load_dotenv
from pymongo import MongoClient
@st.cache_resource
def init_connection():
load_dotenv()
uri = f"mongodb+srv://{os.getenv('MONGO_USERNAME')}:{os.getenv('MONGO_PASSWORD')}@{os.getenv('MONGO_HOST')}/?retryWrites=true&w=majority&appName=Cluster0"
client = MongoClient(uri)
db = client.event_data
# Create Indizes if not present
index_keys = [
"schedule.startdate",
"schedule.enddate",
"taxonomies.event_category",
"taxonomies.event_format",
"locations.geolocation.components._normalized_city"
]
for i in index_keys:
db.events.create_index(i)
return db
|