forked from gaugendre/ofx-processor
149 lines
4 KiB
Python
149 lines
4 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>",
|
|
"mailgun_api_key": "",
|
|
"mailgun_domain": "",
|
|
"mailgun_from": "",
|
|
"email_recipient": "",
|
|
}
|
|
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", help="Edit the config file.")
|
|
def edit_config():
|
|
config_file = get_config_file_name()
|
|
click.edit(filename=config_file)
|
|
|
|
|
|
@config.command("file", help="Print the config file path.")
|
|
def show_file_name():
|
|
config_file = get_config_file_name()
|
|
click.echo(config_file)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class Config:
|
|
account: str
|
|
budget_id: str
|
|
token: str
|
|
bank_identifier: Optional[str] = None
|
|
bank_password: Optional[str] = None
|
|
mailgun_api_key: Optional[str] = None
|
|
mailgun_domain: Optional[str] = None
|
|
mailgun_from: Optional[str] = None
|
|
email_recipient: Optional[str] = None
|
|
sms_user: Optional[str] = None
|
|
sms_key: Optional[str] = None
|
|
|
|
@property
|
|
def email_setup(self) -> bool:
|
|
"""Return true if all fields are setup for email."""
|
|
return all(
|
|
[
|
|
self.mailgun_from,
|
|
self.mailgun_domain,
|
|
self.mailgun_api_key,
|
|
self.email_recipient,
|
|
]
|
|
)
|
|
|
|
@property
|
|
def sms_setup(self) -> bool:
|
|
"""Return true if all fields are setup for sms."""
|
|
return all(
|
|
[
|
|
self.sms_user,
|
|
self.sms_key,
|
|
]
|
|
)
|
|
|
|
|
|
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")
|
|
mailgun_api_key = section.get("mailgun_api_key")
|
|
mailgun_domain = section.get("mailgun_domain")
|
|
mailgun_from = section.get("mailgun_from")
|
|
email_recipient = section.get("email_recipient")
|
|
sms_user = section.get("sms_user")
|
|
sms_key = section.get("sms_key")
|
|
except KeyError as e:
|
|
return handle_config_file_error(config_file, e)
|
|
|
|
return Config(
|
|
account,
|
|
budget_id,
|
|
token,
|
|
bank_identifier,
|
|
bank_password,
|
|
mailgun_api_key,
|
|
mailgun_domain,
|
|
mailgun_from,
|
|
email_recipient,
|
|
sms_user,
|
|
sms_key,
|
|
)
|
|
|
|
|
|
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)
|