87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import os
|
|
import dotenv
|
|
import logging
|
|
import yaml
|
|
import sys
|
|
import atexit
|
|
import logging.config
|
|
|
|
import config.exit_codes as ec
|
|
from visuals.dash_app import app
|
|
|
|
|
|
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
|
|
# Fallback to a basic configuration
|
|
logging.basicConfig(level=logging.INFO)
|
|
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:
|
|
# Attempt to print a more informative error message
|
|
try:
|
|
parsed_yaml = yaml.safe_load_all(file)
|
|
for key, value in parsed_yaml:
|
|
print(f"Error parsing key '{key}': {value}")
|
|
except Exception as e:
|
|
logging.error(f"Error loading config.yaml: {e}")
|
|
sys.exit(ec.CORRUPTED_CONFIG_FILE)
|
|
except Exception as e:
|
|
logging.error(f"some other problem {e}")
|
|
|
|
|
|
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:
|
|
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
|