Libraries refactor and add some sinks #4

Closed
m wants to merge 136 commits from library-refactor into master
3 changed files with 29 additions and 71 deletions
Showing only changes of commit 240c223344 - Show all commits

View File

@ -20,14 +20,14 @@ from notify import Notify
from markets import Markets from markets import Markets
from money import Money from money import Money
# from sinks.nordigen import Nordigen from sinks.nordigen import Nordigen
# from sinks.yapily import Yapily
from sinks.truelayer import TrueLayer from sinks.truelayer import TrueLayer
from sinks.fidor import Fidor from sinks.fidor import Fidor
init_map = None init_map = None
# TODO: extend this with more
def cleanup(sig, frame): def cleanup(sig, frame):
if init_map: if init_map:
try: try:
@ -74,11 +74,13 @@ class WebApp(object):
# self.tx.transaction(parsed) # self.tx.transaction(parsed)
return dumps(True) return dumps(True)
# set up another connection to a bank
@app.route("/signin", methods=["GET"]) @app.route("/signin", methods=["GET"])
def signin(self, request): def signin(self, request):
auth_url = self.truelayer.create_auth_url() auth_url = self.truelayer.create_auth_url()
return f'Please sign in <a href="{auth_url}" target="_blank">here.</a>' return f'Please sign in <a href="{auth_url}" target="_blank">here.</a>'
# endpoint called after we finish setting up a connection above
@app.route("/callback-truelayer", methods=["POST"]) @app.route("/callback-truelayer", methods=["POST"])
def signin_callback(self, request): def signin_callback(self, request):
code = request.args[b"code"] code = request.args[b"code"]
@ -88,7 +90,7 @@ class WebApp(object):
@app.route("/accounts", methods=["GET"]) @app.route("/accounts", methods=["GET"])
def balance(self, request): def balance(self, request):
accounts = self.truelayer.get_accounts() accounts = self.truelayer.get_accounts()
return dumps(accounts) return dumps(accounts, indent=2)
if __name__ == "__main__": if __name__ == "__main__":
@ -98,8 +100,7 @@ if __name__ == "__main__":
"agora": Agora(), "agora": Agora(),
"markets": Markets(), "markets": Markets(),
"revolut": Revolut(), "revolut": Revolut(),
# "nordigen": Nordigen(), "nordigen": Nordigen(),
# "yapily": Yapily(),
"truelayer": TrueLayer(), "truelayer": TrueLayer(),
"fidor": Fidor(), "fidor": Fidor(),
"tx": Transactions(), "tx": Transactions(),
@ -110,19 +111,19 @@ if __name__ == "__main__":
util.xmerge_attrs(init_map) util.xmerge_attrs(init_map)
# Setup the authcode -> refresh token and refresh_token -> auth_token stuff # Setup the authcode -> refresh token and refresh_token -> auth_token stuff
util.setup_call_loops( # util.setup_call_loops(
token_setting=settings.Revolut.SetupToken, # token_setting=settings.Revolut.SetupToken,
function_init=init_map["revolut"].setup_auth, # function_init=init_map["revolut"].setup_auth,
function_continuous=init_map["revolut"].get_new_token, # function_continuous=init_map["revolut"].get_new_token,
delay=int(settings.Revolut.RefreshSec), # delay=int(settings.Revolut.RefreshSec),
function_post_start=init_map["revolut"].setup_webhook, # function_post_start=init_map["revolut"].setup_webhook,
) # )
util.setup_call_loops( # util.setup_call_loops(
token_setting=settings.TrueLayer.SetupToken, # token_setting=settings.TrueLayer.SetupToken,
function_init=init_map["truelayer"].setup_auth, # function_init=init_map["truelayer"].setup_auth,
function_continuous=init_map["truelayer"].get_new_token, # function_continuous=init_map["truelayer"].get_new_token,
delay=int(settings.TrueLayer.RefreshSec), # delay=int(settings.TrueLayer.RefreshSec),
) # )
# Set up the loops to put data in ES # Set up the loops to put data in ES
init_map["tx"].setup_loops() init_map["tx"].setup_loops()

View File

@ -485,3 +485,13 @@ class IRCCommands(object):
def run(cmd, spl, length, authed, msg, agora, revolut, tx, notify): def run(cmd, spl, length, authed, msg, agora, revolut, tx, notify):
total = tx.money.get_profit(True) total = tx.money.get_profit(True)
msg(f"Profit: {total}USD") msg(f"Profit: {total}USD")
class signin(object):
name = "signin"
authed = True
helptext = "Generate a TrueLayer signin URL."
@staticmethod
def run(cmd, spl, length, authed, msg, agora, revolut, tx, notify):
auth_url = agora.truelayer.create_auth_url()
msg(f"Auth URL: {auth_url}")

View File

@ -1,53 +0,0 @@
# Twisted/Klein imports
from twisted.logger import Logger
# Other library imports
import requests
from requests.auth import HTTPBasicAuth
from json import dumps
from simplejson.errors import JSONDecodeError
# Project imports
from settings import settings
class Yapily(object):
"""
Class to manage calls to Open Banking APIs through Yapily.
"""
def __init__(self):
self.log = Logger("yapily")
self.auth = HTTPBasicAuth(settings.Yapily.ID, settings.Yapily.Key)
print(self.get_institutions())
authorisation = self.get_authorisation("monzo_ob", settings.Yapily.CallbackURL)
print("authirisation", authorisation)
def get_institutions(self, filter_name=None):
"""
Get a list of supported institutions.
"""
path = f"{settings.Yapily.Base}/institutions"
r = requests.get(path, auth=self.auth)
try:
parsed = r.json()
except JSONDecodeError:
self.log.error("Error parsing institutions response: {content}", content=r.content)
return False
return parsed
def get_authorisation(self, institution_id, callback_url):
"""
Get an authorisation URL for linking a bank account of an institution.
"""
headers = {"Content-Type": "application/json"}
data = {"applicationUserId": "account-data-and-transactions-tutorial", "institutionId": institution_id, "callback": callback_url}
path = f"{settings.Yapily.Base}/account-auth-requests"
r = requests.post(path, headers=headers, data=dumps(data), auth=self.auth)
try:
parsed = r.json()
except JSONDecodeError:
self.log.error("Error parsing institutions response: {content}", content=r.content)
return False
return parsed