Add some type annotations

1 files changed, 8 insertions, 5 deletions

diff --git a/accounts/__init__.py b/accounts/__init__.py
index 48cd252..c9310fe 100644
--- a/accounts/__init__.py
+++ b/accounts/__init__.py
@@ -1,6 +1,6 @@
 # -*- coding: utf-8 -*-
 import os
-from flask import Flask, g, session
+from flask import Flask
 
 from .models import Service
 from .utils import get_backend
@@ -8,9 +8,12 @@ from .utils.confirmation import Confirmation
 from .utils.sessions import EncryptedSessionInterface
 from .utils.login import create_login_manager
 from .views import default, login, admin
+from .app import AccountsFlask
 
+from typing import Optional
 
-def absolute_paths(app, config):
+
+def absolute_paths(app: Flask, config: str) -> None:
     def handle_option(dirname, name):
         if app.config.get(name):
             app.config[name] = os.path.join(dirname, app.config[name])
@@ -19,15 +22,15 @@ def absolute_paths(app, config):
     handle_option(dirname, 'USERNAME_BLACKLIST_FILE')
 
 
-def load_config(app, configfile):
+def load_config(app: Flask, configfile: Optional[str]) -> None:
     if configfile is not None:
         filename = os.path.abspath(configfile)
         app.config.from_pyfile(filename)
         absolute_paths(app, filename)
 
 
-def create_app(config=None):
-    app = Flask(__name__)
+def create_app(config=None) -> Flask:
+    app = AccountsFlask(__name__)
     app.config.from_object('accounts.default_settings')
     load_config(app, os.environ.get('SPLINE_ACCOUNT_WEB_SETTINGS'))
     load_config(app, config)