storage module
This commit is contained in:
@ -14,13 +14,14 @@ from config import LIQUIDITY_TOKENS, YOUR_WALLET
|
||||
|
||||
from modules import storage, utils, SolanaAPI
|
||||
from modules.utils import async_safe_call, decode_instruction_data
|
||||
from modules.storage import Storage
|
||||
import os
|
||||
import logging
|
||||
from datetime import datetime
|
||||
|
||||
on_transaction = None
|
||||
|
||||
def init_app(tr_handler=None):
|
||||
async def init_app(tr_handler=None):
|
||||
global on_transaction
|
||||
on_transaction = tr_handler
|
||||
app = Flask(__name__, template_folder='../templates', static_folder='../static')
|
||||
@ -29,6 +30,15 @@ def init_app(tr_handler=None):
|
||||
executor = ThreadPoolExecutor(max_workers=10) # Adjust the number of workers as needed
|
||||
login_manager = LoginManager(app)
|
||||
login_manager.login_view = 'login'
|
||||
|
||||
storage = Storage()
|
||||
|
||||
# Ensure database connection
|
||||
async def ensure_storage_connection():
|
||||
if not storage.is_connected():
|
||||
await storage.connect()
|
||||
|
||||
asyncio.run(ensure_storage_connection())
|
||||
|
||||
# oauth = OAuth(app)
|
||||
# google = oauth.remote_app(
|
||||
@ -481,6 +491,9 @@ def init_app(tr_handler=None):
|
||||
|
||||
return app
|
||||
|
||||
def teardown_app():
|
||||
# Close the database connection
|
||||
storage.disconnect()
|
||||
|
||||
# Function to find the latest log file
|
||||
def get_latest_log_file(wh:bool):
|
||||
@ -499,4 +512,4 @@ def get_latest_log_file(wh:bool):
|
||||
utils.log.error(f"Error fetching latest log file: {e}")
|
||||
return None
|
||||
|
||||
export = init_app
|
||||
export = init_app, teardown_app
|
||||
|
Reference in New Issue
Block a user