forked from gaugendre/ofx-processor
93 lines
2.5 KiB
Python
93 lines
2.5 KiB
Python
import configparser
|
|
import os
|
|
import sys
|
|
from dataclasses import dataclass
|
|
from typing import Optional
|
|
|
|
import click
|
|
|
|
DEFAULT_CONFIG_DIR = click.get_app_dir("ofx_processor")
|
|
DEFAULT_CONFIG_FILENAME = "config.ini"
|
|
|
|
|
|
def get_default_config():
|
|
default_config = configparser.ConfigParser()
|
|
default_config["DEFAULT"] = {
|
|
"token": "<YOUR API TOKEN>",
|
|
"budget": "<YOUR BUDGET ID>",
|
|
}
|
|
default_config["bpvf"] = {"account": "<YOUR ACCOUNT ID>"}
|
|
default_config["revolut"] = {"account": "<YOUR ACCOUNT ID>"}
|
|
default_config["ce"] = {"account": "<YOUR ACCOUNT ID>"}
|
|
default_config["lcl"] = {
|
|
"account": "<YOUR ACCOUNT ID>",
|
|
"bank_identifier": "login",
|
|
"bank_password": "password",
|
|
}
|
|
|
|
return default_config
|
|
|
|
|
|
def get_config_file_name():
|
|
config_file = os.path.join(DEFAULT_CONFIG_DIR, DEFAULT_CONFIG_FILENAME)
|
|
return config_file
|
|
|
|
|
|
@click.group()
|
|
def config():
|
|
"""Manage configuration."""
|
|
|
|
|
|
@config.command("edit")
|
|
def edit_config():
|
|
config_file = get_config_file_name()
|
|
click.edit(filename=config_file)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Config:
|
|
account: str
|
|
budget_id: str
|
|
token: str
|
|
bank_identifier: Optional[str] = None
|
|
bank_password: Optional[str] = None
|
|
|
|
|
|
def get_config(account: str) -> Config:
|
|
config = configparser.ConfigParser()
|
|
config_file = get_config_file_name()
|
|
|
|
if not os.path.isfile(config_file):
|
|
os.makedirs(DEFAULT_CONFIG_DIR, exist_ok=True)
|
|
config = get_default_config()
|
|
with open(config_file, "w") as file_:
|
|
config.write(file_)
|
|
click.secho("Editing config file...")
|
|
click.pause()
|
|
click.edit(filename=config_file)
|
|
|
|
try:
|
|
config.read(config_file)
|
|
except configparser.Error as e:
|
|
return handle_config_file_error(config_file, e)
|
|
|
|
try:
|
|
section = config[account]
|
|
budget_id = section["budget"]
|
|
token = section["token"]
|
|
account = section["account"]
|
|
bank_identifier = section.get("bank_identifier")
|
|
bank_password = section.get("bank_password")
|
|
except KeyError as e:
|
|
return handle_config_file_error(config_file, e)
|
|
|
|
return Config(account, budget_id, token, bank_identifier, bank_password)
|
|
|
|
|
|
def handle_config_file_error(config_file, e):
|
|
click.secho(f"Error while parsing config file: {str(e)}", fg="red", bold=True)
|
|
click.secho("Opening the file...")
|
|
click.pause()
|
|
click.edit(filename=config_file)
|
|
click.secho("Exiting...", fg="red", bold=True)
|
|
sys.exit(1)
|