Jake-Pullen
75 lines
2.3 KiB
Python
75 lines
2.3 KiB
Python
import os
|
|
import dotenv
|
|
import logging
|
|
import yaml
|
|
import sys
|
|
import atexit
|
|
import logging.config
|
|
import logging.handlers
|
|
|
|
import config.exit_codes as ec
|
|
from pipeline.pipeline_main import pipeline_main
|
|
|
|
def set_up_logging():
|
|
try:
|
|
with open('config/logging_config.yaml', 'r') as f:
|
|
log_config = yaml.safe_load(f)
|
|
logging.config.dictConfig(log_config)
|
|
except yaml.YAMLError as e:
|
|
print(f"Error parsing logging configuration file: {e}")
|
|
log_config = {} # Initialize log_config to an empty dictionary
|
|
logging.basicConfig(level=logging.INFO) # Fallback to a basic configuration
|
|
queue_handler = logging.getHandlerByName('queue_handler')
|
|
if queue_handler is not None:
|
|
queue_handler.listener.start()
|
|
atexit.register(queue_handler.listener.stop)
|
|
|
|
def load_config():
|
|
try:
|
|
with open('config/config.yaml', 'r') as file:
|
|
config = yaml.safe_load(file)
|
|
return config
|
|
except FileNotFoundError:
|
|
logging.error('config.yaml file not found')
|
|
sys.exit(ec.MISSING_CONFIG_FILE)
|
|
except yaml.YAMLError as e:
|
|
logging.error(f'Error loading config.yaml: {e}')
|
|
sys.exit(ec.CORRUPTED_CONFIG_FILE)
|
|
|
|
logger = logging.getLogger("data_pipeline_for_ynab")
|
|
os.makedirs('logs', exist_ok=True)
|
|
set_up_logging()
|
|
|
|
# Load environment variables
|
|
dotenv.load_dotenv()
|
|
|
|
API_TOKEN = os.getenv('API_TOKEN')
|
|
BUDGET_ID = os.getenv('BUDGET_ID')
|
|
|
|
if not API_TOKEN or not BUDGET_ID:
|
|
logging.error('API_TOKEN or BUDGET_ID is not set in .env file')
|
|
sys.exit(ec.MISSING_ENV_VARS)
|
|
|
|
if __name__ == '__main__':
|
|
config = load_config()
|
|
config['API_TOKEN'] = API_TOKEN
|
|
config['BUDGET_ID'] = BUDGET_ID
|
|
try:
|
|
pipeline_main(config)
|
|
|
|
# Check if the data was successfully created
|
|
data_exists = os.path.exists('data/processed') and os.listdir('data/processed')
|
|
if data_exists:
|
|
from dash_app import app
|
|
app.run() # debug=True
|
|
else:
|
|
logging.error('Data pipeline did not produce any data. Dash app will not run.')
|
|
sys.exit(ec.NO_DATA_PRODUCED)
|
|
except SystemExit as e:
|
|
exit_code = e.code
|
|
if exit_code == ec.SUCCESS:
|
|
logging.info('Program exited successfully')
|
|
else:
|
|
logging.error(f'Program exited with code {exit_code}')
|
|
raise
|