Test refactor

Change-Id: I1827bfedba0c4af4e42d8d5a970bb09a8fa1f60f

Author: AndrewMBurke

flask/04-context/__pycache__/config.cpython-39.pyc

@@ -0,0 +1,38 @@
+#!/usr/bin/env python3
+# Copyright 2021 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not
+# use this file except in compliance with the License. You may obtain a copy of
+# the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+"""The Flask server configuration."""
+
+import os
+
+DATABASE_FILE_NAME = os.path.join(os.path.abspath(os.path.dirname(__file__)),
+                                    'data.sqlite')
+
+class Config(object):
+    # Note: A secret key is included in the sample so that it works.
+    # If you use this code in your application, replace this with a truly secret
+    # key. See https://flask.palletsprojects.com/quickstart/#sessions.
+    SECRET_KEY = os.environ.get(
+        'SECRET_KEY') or "REPLACE ME - this value is here as a placeholder."
+
+    # Configure the flask cookie settings per the iframe security recommendations:
+    # https://developers.google.com/classroom/eap/add-ons-alpha/iframes#iframe_security_guidelines
+    SESSION_COOKIE_SECURE = True
+    SESSION_COOKIE_HTTPONLY = True
+    SESSION_COOKIE_SAMESITE = "None"
+
+    # Point to a database file in the project root.
+
+    SQLALCHEMY_DATABASE_URI=f"sqlite:///{DATABASE_FILE_NAME}"
+    SQLALCHEMY_TRACK_MODIFICATIONS=False

flask/04-context/config.py

@@ -0,0 +1,6 @@
+Flask==2.0.2
+Flask_SQLAlchemy==2.5.1
+google_api_python_client==2.33.0
+google_auth_oauthlib==0.4.6
+protobuf==3.19.1
+requests==2.26.0

flask/04-context/requirements.txt

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+# Copyright 2021 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not
+# use this file except in compliance with the License. You may obtain a copy of
+# the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+"""Entry point for the Flask server.
+
+Loads the webapp module and starts the server."""
+
+from webapp import app
+import os
+
+if __name__ == "__main__":
+    # Allow the OAuth flow to adjust scopes.
+    os.environ["OAUTHLIB_RELAX_TOKEN_SCOPE"] = "1"
+
+    app.run(host="localhost",
+            ssl_context=("localhost.pem", "localhost-key.pem"),
+            debug=True)

flask/04-context/webapp.py

@@ -0,0 +1,15 @@
+from flask import Flask
+import config
+from flask_sqlalchemy import SQLAlchemy
+from os import path
+
+app = Flask(__name__)
+app.config.from_object(config.Config)
+
+db = SQLAlchemy(app)
+
+from webapp import routes, models
+
+# Initialize the database file if not created.
+if not path.exists(config.DATABASE_FILE_NAME):
+    db.create_all()

flask/04-context/webapp/__init__.py

@@ -0,0 +1,39 @@
+#!/usr/bin/env python3
+# Copyright 2021 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License"); you may not
+# use this file except in compliance with the License. You may obtain a copy of
+# the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
+# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
+# License for the specific language governing permissions and limitations under
+# the License.
+"""Data models for the Flask server.
+
+Uses the flask-sqlalchemy database object defined in the webapp module."""
+
+from webapp import db
+
+# Database model to represent a user.
+class User(db.Model):
+    # The user's identifying information:
+    id = db.Column(db.String(120), primary_key=True)
+    display_name = db.Column(db.String(80))
+    email = db.Column(db.String(120), unique=True)
+    portrait_url = db.Column(db.Text())
+
+    # The user's refresh token, which will be used to obtain an access token.
+    # Note that refresh tokens will become invalid if:
+    # - The refresh token has not been used for six months.
+    # - The user revokes your app's access permissions.
+    # - The user changes passwords.
+    # - The user belongs to a Google Cloud Platform organization
+    #   that has session control policies in effect.
+    refresh_token = db.Column(db.Text())
+
+    def __repr__(self):
+        return f"<User {self.display_name}, ID {self.id}>"