flask webui looking good

crypto/sol/modules/webui.py

@@ -1,75 +1,112 @@
 from flask import Flask, jsonify, request, render_template, redirect, url_for
+# from flask_oauthlib.client import OAuth
 from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user
 import secrets
 from modules import storage
+import os
 
-app = Flask(__name__, template_folder='../templates', static_folder='../static')
-app.config['SECRET_KEY'] = 'your-secret-key'
-login_manager = LoginManager(app)
-login_manager.login_view = 'login'
+def init_app():
+    app = Flask(__name__, template_folder='../templates', static_folder='../static')
+    app.config['SECRET_KEY'] = 'your-secret-key'
+    login_manager = LoginManager(app)
+    login_manager.login_view = 'login'
 
-class User(UserMixin):
-    def __init__(self, id, username, email):
-        self.id = id
-        self.username = username
-        self.email = email
+    # oauth = OAuth(app)
+    # google = oauth.remote_app(
+    #     'google',
+    #     consumer_key='YOUR_GOOGLE_CLIENT_ID',
+    #     consumer_secret='YOUR_GOOGLE_CLIENT_SECRET',
+    #     request_token_params={
+    #         'scope': 'email'
+    #     },
+    #     base_url='https://www.googleapis.com/oauth2/v1/',
+    #     request_token_url=None,
+    #     access_token_method='POST',
+    #     access_token_url='https://accounts.google.com/o/oauth2/token',
+    #     authorize_url='https://accounts.google.com/o/oauth2/auth',
+    # )
 
-@login_manager.user_loader
-def load_user(user_id):
-    user_data = storage.get_user_by_id(user_id)
-    if user_data:
-        return User(id=user_data['id'], username=user_data['username'], email=user_data['email'])
-    return None
 
-@app.route('/')
-def index():
-    return render_template('index.html')
+    login_manager = LoginManager()
+    login_manager.init_app(app)
+    
+    @app.route('/login/google/authorized')
+    def authorized():
+        # resp = google.authorized_response()
+        # if resp is None or resp.get('access_token') is None:
+        #     return 'Access denied: reason={} error={}'.format(
+        #         request.args['error_reason'],
+        #         request.args['error_description']
+        #     )
+        # session['google_token'] = (resp['access_token'], '')
+        # user_info = google.get('userinfo')
+        # user = storage.get_or_create_user(user_info.data['email'], user_info.data['id'])
+        # login_user(user)
+        return redirect(url_for('index'))
+    
+    
+    class User(UserMixin):
+        def __init__(self, id, username, email):
+            self.id = id
+            self.username = username
+            self.email = email
 
-@app.route('/login', methods=['GET', 'POST'])
-def login():
-    if request.method == 'POST':
-        username = request.form.get('username')
-        password = request.form.get('password')
-        
-        user = storage.authenticate_user(username, password)
-        if user:
-            login_user(User(id=user['id'], username=user['username'], email=user['email']))
-            return redirect(url_for('dashboard'))
-        else:
-            return render_template('login.html', error='Invalid credentials')
-    return render_template('login.html')
+    @login_manager.user_loader
+    def load_user(user_id):
+        user_data = storage.get_user_by_id(user_id)
+        if user_data:
+            return User(id=user_data['id'], username=user_data['username'], email=user_data['email'])
+        return None
 
-@app.route('/logout')
-@login_required
-def logout():
-    logout_user()
-    return redirect(url_for('index'))
+    @app.route('/')
+    def index():
+        return render_template('index.html')
 
-@app.route('/dashboard')
-@login_required
-def dashboard():
-    return render_template('dashboard.html')
+    @app.route('/login', methods=['GET', 'POST'])
+    def login():
+        if request.method == 'POST':
+            username = request.form.get('username')
+            password = request.form.get('password')
+            user = storage.authenticate_user(username, password)
+            if user:
+                login_user(User(id=user['id'], username=user['username'], email=user['email']))
+                return redirect(url_for('dashboard'))
+            else:
+                return render_template('login.html', error='Invalid credentials')
+        elif request.args.get('google'):
+            return google.authorize(callback=url_for('authorized', _external=True))
+        return render_template('login.html')
 
-@app.route('/generate_api_key', methods=['POST'])
-@login_required
-def generate_api_key():
-    api_key = secrets.token_urlsafe(32)
-    storage.store_api_key(current_user.id, api_key)
-    return jsonify({'api_key': api_key})
+    @app.route('/logout')
+    @login_required
+    def logout():
+        logout_user()
+        return redirect(url_for('index'))
 
-@app.route('/wallet/<int:wallet_id>/transactions', methods=['GET'])
-@login_required
-def get_transactions(wallet_id):
-    transactions = storage.get_transactions(wallet_id)
-    return jsonify(transactions)
+    @app.route('/dashboard')
+    @login_required
+    def dashboard():
+        return render_template('dashboard.html')
 
-@app.route('/wallet/<int:wallet_id>/holdings', methods=['GET'])
-@login_required
-def get_holdings(wallet_id):
-    holdings = storage.get_holdings(wallet_id)
-    return jsonify(holdings)
+    @app.route('/generate_api_key', methods=['POST'])
+    @login_required
+    def generate_api_key():
+        api_key = secrets.token_urlsafe(32)
+        storage.store_api_key(current_user.id, api_key)
+        return jsonify({'api_key': api_key})
+
+    @app.route('/wallet/<int:wallet_id>/transactions', methods=['GET'])
+    @login_required
+    def get_transactions(wallet_id):
+        transactions = storage.get_transactions(wallet_id)
+        return jsonify(transactions)
+
+    @app.route('/wallet/<int:wallet_id>/holdings', methods=['GET'])
+    @login_required
+    def get_holdings(wallet_id):
+        holdings = storage.get_holdings(wallet_id)
+        return jsonify(holdings)
 
 # Implement other routes for reports, price alerts, following accounts, etc.
 
-def init_app():
     return app