blacken

Author: Avantol13

flask-google-login/app.py

@@ -32,9 +32,12 @@
 # https://flask-login.readthedocs.io/en/latest
 login_manager = LoginManager()
 login_manager.init_app(app)
+
+
 @login_manager.unauthorized_handler
 def unauthorized():
-    return 'You must be logged in to access this content.', 403
+    return "You must be logged in to access this content.", 403
+
 
 # Naive database setup
 try:
@@ -51,6 +54,7 @@ def unauthorized():
 def load_user(user_id):
     return User.get(user_id)
 
+
 @app.route("/")
 def index():
     if current_user.is_authenticated:
@@ -65,6 +69,7 @@ def index():
     else:
         return '<a class="button" href="/login">Login with Google</a>'
 
+
 @app.route("/login")
 def login():
     # Find out what URL to hit for Google login
@@ -80,6 +85,7 @@ def login():
     )
     return redirect(request_uri)
 
+
 @app.route("/login/callback")
 def callback():
     # Get authorization code Google sent back to you
@@ -95,7 +101,7 @@ def callback():
         token_endpoint,
         authorization_response=request.url,
         redirect_url=request.base_url,
-        code=code
+        code=code,
     )
     token_response = requests.post(
         token_url,
@@ -139,14 +145,17 @@ def callback():
     # Send user back to homepage
     return redirect(url_for("index"))
 
+
 @app.route("/logout")
 @login_required
 def logout():
     logout_user()
     return redirect(url_for("index"))
 
+
 def get_google_provider_cfg():
     return requests.get(GOOGLE_DISCOVERY_URL).json()
 
+
 if __name__ == "__main__":
     app.run(ssl_context="adhoc")

flask-google-login/db.py

@@ -5,34 +5,37 @@
 from flask import current_app, g
 from flask.cli import with_appcontext
 
+
 def get_db():
     if "db" not in g:
-        g.db = sqlite3.connect(
-            "sqlite_db", detect_types=sqlite3.PARSE_DECLTYPES
-        )
+        g.db = sqlite3.connect("sqlite_db", detect_types=sqlite3.PARSE_DECLTYPES)
         g.db.row_factory = sqlite3.Row
 
     return g.db
 
+
 def close_db(e=None):
     db = g.pop("db", None)
 
     if db is not None:
         db.close()
 
+
 def init_db():
     db = get_db()
 
     with current_app.open_resource("schema.sql") as f:
         db.executescript(f.read().decode("utf8"))
 
+
 @click.command("init-db")
 @with_appcontext
 def init_db_command():
     """Clear the existing data and create new tables."""
     init_db()
     click.echo("Initialized the database.")
 
+
 def init_app(app):
     app.teardown_appcontext(close_db)
-    app.cli.add_command(init_db_command)
+    app.cli.add_command(init_db_command)

flask-google-login/user.py

@@ -2,6 +2,7 @@
 
 from db import get_db
 
+
 class User(UserMixin):
     def __init__(self, id_, name, email, profile_pic):
         self.id = id_
@@ -12,15 +13,11 @@ def __init__(self, id_, name, email, profile_pic):
     @staticmethod
     def get(user_id):
         db = get_db()
-        user = db.execute(
-            "SELECT * FROM user WHERE id = ?", (user_id,)
-        ).fetchone()
+        user = db.execute("SELECT * FROM user WHERE id = ?", (user_id,)).fetchone()
         if not user:
             return None
 
-        user = User(
-            id_=user[0], name=user[1], email=user[2], profile_pic=user[3]
-        )
+        user = User(id_=user[0], name=user[1], email=user[2], profile_pic=user[3])
         return user
 
     @staticmethod
@@ -30,4 +27,4 @@ def create(id_, name, email, profile_pic):
             "INSERT INTO user (id, name, email, profile_pic) VALUES (?, ?, ?, ?)",
             (id_, name, email, profile_pic),
         )
-        db.commit()
+        db.commit()