120 строки
3.6 KiB
Python
120 строки
3.6 KiB
Python
import base64
|
|
import io
|
|
import os
|
|
|
|
from flask import Flask, make_response, request
|
|
from flask.json import jsonify
|
|
|
|
import boto3
|
|
import httplib2
|
|
|
|
from apiclient.discovery import build
|
|
from oauth2client.service_account import ServiceAccountCredentials
|
|
|
|
|
|
app = Flask(__name__)
|
|
app.config['SECRET_KEY'] = os.getenv('SECRET_KEY')
|
|
|
|
|
|
def get_credentials():
|
|
"""Fetch playstore credentials from SSM"""
|
|
|
|
# Fetch credentials from AWS SSM
|
|
ssm = boto3.client('ssm')
|
|
key_response = ssm.get_parameter(
|
|
Name='SumoPlaystoreReviewsKey', WithDecryption=True)
|
|
account_response = ssm.get_parameter(
|
|
Name='SumoPlaystoreReviewsAccount', WithDecryption=True)
|
|
|
|
# Parse SSM responses
|
|
account = account_response['Parameter']['Value']
|
|
key = io.BytesIO(base64.b64decode(key_response['Parameter']['Value']))
|
|
|
|
credentials = {
|
|
'key': key,
|
|
'account': account
|
|
}
|
|
return credentials
|
|
|
|
|
|
def get_reviews_service():
|
|
"""Create new instance of google play API service."""
|
|
|
|
scope = 'https://www.googleapis.com/auth/androidpublisher'
|
|
credentials = get_credentials()
|
|
service_credentials = ServiceAccountCredentials.from_p12_keyfile_buffer(
|
|
credentials['account'], credentials['key'], scopes=scope
|
|
)
|
|
http = httplib2.Http()
|
|
http = service_credentials.authorize(http)
|
|
service = build('androidpublisher', 'v3', http=http, cache_discovery=False)
|
|
return service.reviews()
|
|
|
|
|
|
@app.route('/reviews', methods=['GET'])
|
|
def get_reviews():
|
|
"""Get playstore reviews. Proxy playstore API requests."""
|
|
|
|
packageName = request.args.get('packageName', None)
|
|
nextPageToken = request.args.get('token', None)
|
|
|
|
if not packageName:
|
|
content = jsonify(msg='Missing `packageName` from request query')
|
|
return make_response(content, 400)
|
|
|
|
reviews_service = get_reviews_service()
|
|
app.logger.info('Playstore API: GET list')
|
|
query = reviews_service.list(packageName=packageName, token=nextPageToken)
|
|
|
|
return make_response(jsonify(query.execute()), 200)
|
|
|
|
|
|
@app.route('/reviews', methods=['POST'])
|
|
def post_reviews():
|
|
"""Post playstore reviews."""
|
|
|
|
# Flag to enable/disable playstore upload
|
|
UPLOADS_ENABLED = os.getenv('SUMO_PLAYSTORE_UPLOADS_ENABLED', False)
|
|
|
|
packageName = request.args.get('packageName', None)
|
|
reviewId = request.args.get('reviewId', None)
|
|
replyText = request.args.get('text', None)
|
|
|
|
if not packageName:
|
|
content = jsonify(msg='Missing `packageName` from request')
|
|
return make_response(content, 400)
|
|
|
|
if not reviewId:
|
|
content = jsonify(msg='Missing `review_id` from request')
|
|
return make_response(content, 400)
|
|
|
|
if not replyText:
|
|
content = jsonify(msg='Missing `text` from request')
|
|
return make_response(content, 400)
|
|
|
|
payload = {
|
|
'packageName': packageName,
|
|
'reviewId': reviewId,
|
|
'replyText': replyText
|
|
}
|
|
|
|
app.logger.info('Playstore API: POST')
|
|
app.logger.info('Playstore API - upload - call params: {}'.format(payload))
|
|
app.logger.info('Playstore API - uploads enabled: {}'.format(UPLOADS_ENABLED))
|
|
|
|
if UPLOADS_ENABLED:
|
|
reviews_service = get_reviews_service()
|
|
body = {
|
|
'replyText': replyText
|
|
}
|
|
query = reviews_service.reply(
|
|
packageName=packageName,
|
|
reviewId=reviewId,
|
|
body=body
|
|
)
|
|
response = query.execute()
|
|
app.logger.info('Playstore API - response: {}'.format(response))
|
|
return make_response(jsonify(response), 200)
|
|
|
|
return make_response(jsonify(msg='POST method not implemented'), 501)
|