upload-a-file-to-GCS-bucket-with-HTML-forms-using-Policy-Document-Python/main.py at main · PawanSenapati/upload-a-file-to-GCS-bucket-with-HTML-forms-using-Policy-Document-Python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import datetime
import flask
from google.oauth2 import service_account

credentials = service_account.Credentials.from_service_account_file('/path/to/key.json')

from google.cloud import storage
app = flask.Flask(__name__)
default_bucket_name = 'BUCKET_NAME'

def generate_signed_post_policy_v4(bucket_name, blob_name):
    """Generates a v4 POST Policy and prints an HTML form."""
    # bucket_name = 'your-bucket-name'
    # blob_name = 'your-object-name'

    storage_client = storage.Client()

    policy = storage_client.generate_signed_post_policy_v4(
        bucket_name,
        blob_name,
        expiration=datetime.timedelta(minutes=10),
        fields={
          'x-goog-meta-test': 'data'
        },
        credentials=credentials
    )

    print(policy)
    # Create an HTML form with the provided policy
    header = "<form action='{}' method='POST' enctype='multipart/form-data'>\n"
    form = header.format(policy["url"])

    # Include all fields returned in the HTML form as they're required
    for key, value in policy["fields"].items():
        form += f"  <input name='{key}' value='{value}' type='hidden'/>\n"

    form += "  <input type='file' name='file'/><br />\n"
    form += "  <input type='submit' value='Upload File' /><br />\n"
    form += "</form>"

    print(form)

    return form

@app.route('/')
def hello():

    response = "<html><body>\n"
    response += "<h1>Test file upload:</h1>\n"
    response += "<p>Please allow each upload to complete before proceeding to the next step.</p>"
    response += generate_signed_post_policy_v4(default_bucket_name, "file1")
    return response

if __name__ == "__main__":
    # Used when running locally only. When deploying to Google App
    # Engine, a webserver process such as Gunicorn will serve the app. This
    # can be configured by adding an `entrypoint` to app.yaml.
    app.run(host="0.0.0.0", port=8080, debug=True)