diff --git a/bin/update-ofxsource b/bin/update-ofxsource new file mode 100755 index 0000000..4871543 --- /dev/null +++ b/bin/update-ofxsource @@ -0,0 +1,25 @@ +#!/usr/bin/env python3 +import logging +import uuid + +import vanth.main +import vanth.tables + +LOGGER = logging.getLogger(__name__) +def main(): + vanth.main.setup_logging() + config = vanth.main.get_config() + engine = vanth.main.create_db_connection(config) + + new_uuid = uuid.uuid4() + query = vanth.tables.OFXSource.insert().values( # pylint: disable=no-value-for-parameter + uuid = str(new_uuid), + name = 'America First Credit Union', + fid = '54324', + bankid = '324377516', + ) + results = engine.execute(query) + LOGGER.info("Created OFX source %s", results.inserted_primary_key[0]) + +if __name__ == '__main__': + main() diff --git a/vanth/main.py b/vanth/main.py index c7760cf..7eece88 100644 --- a/vanth/main.py +++ b/vanth/main.py @@ -11,11 +11,13 @@ import vanth.tables LOGGER = logging.getLogger(__name__) -def create_application(config): - sepiida.log.setup_logging() +def create_db_connection(config): engine = chryso.connection.Engine(config.db, vanth.tables) chryso.connection.store(engine) + return engine +def create_application(config): + create_db_connection(config) LOGGER.info("Starting up vanth version %s", vanth.version.VERSION) application = vanth.server.create_app(config) @@ -23,11 +25,19 @@ def create_application(config): return application -def main(): +def setup_logging(): logging.getLogger().setLevel(logging.DEBUG) logging.basicConfig() - config = sepiida.config.load('/etc/vanth.yaml', vanth.config.SPECIFICATION) + sepiida.log.setup_logging() + +def get_config(): + return sepiida.config.load('/etc/vanth.yaml', vanth.config.SPECIFICATION) + +def main(): + setup_logging() + config = get_config() + application = create_application(config) try: host = os.getenv('HOST', 'localhost')