app.py

#! /usr/bin/python
# encoding=utf-8

import json

from flask import Flask
from flask import json
from flask import request
from flask import send_from_directory
from flask import Response
import requests

import xmlparser
import base64
import hashlib

app = Flask(__name__, static_url_path='')

try:
    from urlparse import urlparse, urlunparse
except ImportError:
    from urllib.parse import urlparse, urlunparse


def get_url_from_req(request):
    url = from_request(request, 'url')
    parsed = urlparse(url)
    rgw_civetweb_port = app.config['RGW_CIVETWEB_PORT']
    rgw_address = app.config['RGW_ADDRESS'] + ":" + rgw_civetweb_port
    return urlunparse((parsed[0], rgw_address, parsed[2], parsed[3], parsed[4], parsed[5]))


def from_request(request, k):
    if not request.json:
        raise Exception("Invalid Request")
    return str(request.json[k])


def req(url, method):
    command = requests.get
    if method == 'DELETE':
        command = requests.delete
    elif method == 'PUT':
        command = requests.put
    elif method == 'POST':
        command = requests.post
    r = command(url)
    return r.status_code



@app.route("/createbucket", methods=['POST'])
def create():
    url = get_url_from_req(request)
    statuscode = req(url, 'PUT')

    if statuscode == 200:
        resp = Response(response='Success', status=statuscode)
    elif statuscode == 400:
        resp = Response(
            response='Invalid Bucket Name (minimum is 3 characters)', status=statuscode)
    elif statuscode == 409:
        resp = Response(response='Conflict: Bucket already Exists',
                        status=statuscode)
    elif statuscode == 403:
        resp = Response(response='Access Denied', status=statuscode)
    else:
        resp = Response(response='Unknown Error', status=500)
    return resp


@app.route("/deletebucket", methods=['DELETE'])
def delete():
    url = get_url_from_req(request)
    statuscode = req(url, 'DELETE')

    if statuscode == 204:
        resp = Response(status=200)
    elif statuscode == 404:
        resp = Response(response='Bucket does not exist', status=statuscode)
    elif statuscode == 403:
        resp = Response(response='Access Denied', status=statuscode)
    elif statuscode == 409:
        resp = Response(response='Conflict: Object still exists',
                        status=statuscode)
    else:
        resp = Response(response='Unknown Error', status=500)
    return resp


@app.route("/putcors", methods=['PUT'])
def putcors():
    corsurl = get_url_from_req(request)
    s3auth = from_request(request, 's3auth')
    date = from_request(request, 'date')

    cors = '''
<CORSConfiguration>
  <CORSRule>
      <AllowedMethod>PUT</AllowedMethod>
      <AllowedMethod>GET</AllowedMethod>
      <AllowedMethod>POST</AllowedMethod>
      <AllowedMethod>DELETE</AllowedMethod>
      <AllowedOrigin>*</AllowedOrigin>
      <AllowedHeader>*</AllowedHeader>
      <ExposeHeader>x-amz-acl</ExposeHeader>
      <ExposeHeader>ETag</ExposeHeader>
  </CORSRule>
</CORSConfiguration>'''
    content_md5 = base64.b64encode(hashlib.md5(cors.encode()).digest()).decode()

    headers = {
        'Content-type':'text/xml',
        'Content-MD5':content_md5,
        'Authorization':s3auth,
        'Date' : date
    }

    r  = requests.put(corsurl, headers=headers, data=cors)
    statuscode = r.status_code

    if statuscode == 200:
        resp = Response(status=statuscode)
    elif statuscode == 403:
        resp = Response(response='Access Denied', status=statuscode)
    else:
        resp = Response(response='Unknown Error', status=500)
    return resp


@app.route("/getservice", methods=['POST'])
def listbucketsurl():
    url = get_url_from_req(request)
    s3auth = from_request(request, 's3auth')
    date = from_request(request, 'date')

    headers = {'Authorization':s3auth, 'x-amz-date': date}

    r = requests.get(url, headers=headers)

    statuscode = r.status_code

    if statuscode != 200:
        if statuscode == 403:
            resp = Response(response='Access Denied', status=statuscode)
        else:
            resp = Response(response='Unknown Error', status=500)
        return resp

    content = r.text

    buckets = xmlparser.getListFromXml(content, 'Bucket')
    resp = Response(response=json.dumps(buckets), status=statuscode)
    resp.headers['Content-type'] = 'application/json; charset=UTF-8'
    return resp


@app.route("/")
def root():
    return app.send_static_file('buckets.html')


@app.route("/<path:path>")
def static_files(path):
    return send_from_directory('static', path)


if __name__ == '__main__':
    app.config.from_pyfile('sree.cfg')
    flask_port = app.config['SREE_PORT']
    flask_debug = app.config['FLASK_DEBUG']
    app.run(host='0.0.0.0', port=flask_port, threaded=True, debug=flask_debug)