Spaces:

TrueDFS
/

NFL_DFS_Range_Of_Outcomes_Origin

Running

NFL_DFS_Range_Of_Outcomes_Origin / src /database.py

James McCool

instituting database.py

d9a4d3d about 1 year ago

429 Bytes

import streamlit as st
import pymongo
import os

@st.cache_resource
def init_conn():
         # Try to get from environment variable first, fall back to secrets
        uri = os.getenv('MONGO_URI')
        if not uri:
            uri = st.secrets['mongo_uri']
        client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
        db = client["NFL_Database"]

        return db
    
db = init_conn()

	import streamlit as st
	import pymongo
	import os

	@st.cache_resource
	def init_conn():
	# Try to get from environment variable first, fall back to secrets
	uri = os.getenv('MONGO_URI')
	if not uri:
	uri = st.secrets['mongo_uri']
	client = pymongo.MongoClient(uri, retryWrites=True, serverSelectionTimeoutMS=500000)
	db = client["NFL_Database"]

	return db

	db = init_conn()