Я использую Streamlit в качестве инфраструктуры для обработки и визуализации данных.
Следуя документам расширенного кэширования с Streamlit , как создавать подключения к базе данных передать их другим функциям, я использую sqlite3 и пытаюсь создать одно соединение при запуске и передать «курсор», выполнив:
import streamlit as st
# create database connection
@st.cache(allow_output_mutation=True)
def get_database_connection():
conn = sqlite3.connect('collect/Data/fpl.db')
c = conn.cursor()
return c
Здесь я пытаюсь передать свой идентификатор подключения через hash_funcs, возвращая созданный курсор:
# pass connections around
@st.cache(allow_output_mutation=True, hash_funcs={sqlite3.Cursor:id})
def get_teams_basic(c):
df_teams = sql('SELECT * FROM TeamsBasic', c)
return df_teams
@st.cache(allow_output_mutation=True, hash_funcs={sqlite3.Cursor:id})
def get_players_basic(c):
df_player_basic = sql('SELECT * FROM PlayersBasic', c)
return df_player_basic
# and so on...
Это мой sql():
def sql(query, cursor):
'''
Takes an SQL query string, and outputs a
dataframe representation of the query result.
'''
# Execute the sql query
cursor.execute(query)
# Get the query into a dataframe and set columns
df_temp = pd.DataFrame(cursor.fetchall())
df_temp.columns = [x[0] for x in cursor.description]
# Set the sql id as the dataframe index
index_column = df_temp.columns[0]
df_temp.set_index(index_column, drop=True, inplace=True)
return df_temp
Затем в main() я создаю экземпляр соединение с базой данных и передать курсор моим функциям:
def main():
c = get_database_connection()
teams(c)
players(c)
if __name__ == '__main__':
main()
Но когда код достигает второго соединения, players() я получаю следующую ошибку:
ProgrammingError: SQLite objects created in a thread can only be used in that same thread. The object was created in thread id 123145444151296 and this is thread id 123145454661632.
Traceback:
File "/Users/me/anaconda2/envs/data_science/lib/python3.7/site-packages/streamlit/ScriptRunner.py", line 322, in _run_script
exec(code, module.__dict__)
File "/Volumes/Dados/Documents/Code/Apps/app.py", line 280, in <module>
main()
File "/Volumes/Dados/Documents/Code/Apps/app.py", line 273, in main
managers(c)
File "/Volumes/Dados/Documents/Code/Apps/app.py", line 149, in managers
df_players = get_players_basic(c)
File "/Users/me/anaconda2/envs/data_science/lib/python3.7/site-packages/streamlit/caching.py", line 594, in wrapped_func
return get_or_create_cached_value()
File "/Users/me/anaconda2/envs/data_science/lib/python3.7/site-packages/streamlit/caching.py", line 578, in get_or_create_cached_value
return_value = func(*args, **kwargs)
File "/Volumes/Dados/Documents/Code/Apps/app.py", line 64, in get_players_basic
df_player_basic = sql('SELECT * FROM PlayersBasic', c)
File "/Volumes/Dados/Documents/Code/Apps/collect/PythonFunctions/sqlfunctions.py", line 111, in sql
cursor.execute(query)
Что мне не хватает?