import streamlit as st | |
from pymongo import MongoClient | |
from dotenv import load_dotenv, dotenv_values | |
import os | |
def init_db(): | |
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) | |
return client.event_data | |
class CollectionNames: | |
EVENT_URLS = "event_urls" | |
UNSORTED_URLS = "unsorted_urls" |