adding python integration sample app

nkubala · nkubala · commit 7ee56fd69f19 · 2017-01-25T13:10:49.000-08:00
diff --git a/build.sh b/build.sh
@@ -0,0 +1,14 @@
+#!/bin/bash
+
+set -e
+
+export IMAGE_NAME=$1
+
+if [ -z "$1" ]; then
+  echo "Usage: ./build.sh [image_path]"
+  echo "Please provide fully qualified path to target image."
+  exit 1
+fi
+
+envsubst < cloudbuild.yaml.in > cloudbuild.yaml
+gcloud alpha container builds create . --config=cloudbuild.yaml
diff --git a/tests/integration/Dockerfile.in b/tests/integration/Dockerfile.in
@@ -0,0 +1,11 @@
+FROM ${STAGING_IMAGE}
+
+COPY . /app
+WORKDIR /app
+
+RUN pip install -r requirements.txt
+
+ENV GOOGLE_APPLICATION_CREDENTIALS=/app/auth.json
+
+ENTRYPOINT ["python"]
+CMD ["server.py"]
diff --git a/tests/integration/requirements.txt b/tests/integration/requirements.txt
@@ -0,0 +1,5 @@
+Flask
+google-cloud
+google-cloud-logging
+google-cloud-monitoring
+requests
diff --git a/tests/integration/server.py b/tests/integration/server.py
@@ -0,0 +1,117 @@
+#!/usr/bin/python
+
+from google.cloud import logging as gcloud_logging
+from google.cloud import monitoring
+from google.cloud.monitoring import MetricKind, ValueType
+from oauth2client.client import GoogleCredentials
+import logging
+from flask import Flask, request, abort, jsonify
+app = Flask(__name__)
+
+
+@app.route('/')
+def hello_world():
+	return 'Hello World!'
+
+
+@app.route('/logging', methods=['POST'])
+def logging():
+	request_data = request.get_json()
+	if request_data is None:
+		raise ErrorResponse("Unable to parse request JSON: did you set the Content-type header?")
+	log_name = request_data.get('log_name', '')
+	if log_name == '':
+		raise ErrorResponse("please provide log name")
+	token = request_data.get('token', '')
+	if token == '':
+		raise ErrorResponse("please provide token name")
+
+	_log("log name is {0}, token is {1}".format(log_name, token))
+	_log(token, log_name)
+
+	return ('', 204)
+
+
+def _log(token, log_name='stdout'):
+	# TODO (nkubala): write token to 'log_name' log, instead of stdout
+	# is this possible in non-standard (flex)???
+
+	# try:
+	# 	client = gcloud_logging.Client(credentials=GoogleCredentials.get_application_default())
+	# 	gcloud_logger = client.logger(log_name)
+	# 	gcloud_logger.log_text(token)
+	# except Exception as e:
+	# 	logging.error("error while writing logs")
+	# 	raise ErrorResponse("error while writing logs: {0}".format(e))
+
+	# logging.info(token)
+	print token
+
+
+@app.route('/monitoring', methods=['POST'])
+def monitoring():
+	request_data = request.get_json()
+	if request_data is None:
+		raise ErrorResponse("Unable to parse request JSON: did you set the Content-type header?")
+	name = request_data.get('name', '')
+	if name == '':
+		raise ErrorResponse("please provide name")
+	token = request_data.get('token', '')
+	if token == '':
+		raise ErrorResponse("please provide metric token")
+
+	client = monitoring.Client(credentials=GoogleCredentials.get_application_default())
+
+	descriptor = client.metric_descriptor(
+		'custom.googleapis.com/{0}'.format(name),
+		metric_kind=MetricKind.GAUGE,
+		value_type=ValueType.DOUBLE,
+		description=token
+		)
+	descriptor.create()
+
+	# resource = client.resource()
+
+	metric = client.metric(
+		type='custom.googleapis.com/{0}'.format(name))
+
+	client.write_point(metric=metric, value=token)
+
+	print 'OK'
+
+
+@app.route('/exception', methods=['POST'])
+def exception():
+	print ''
+
+
+@app.route('/trace', methods=['POST'])
+def trace():
+	print''
+
+
+class ErrorResponse(Exception):
+	status_code = 400
+
+	def __init__(self, message, status_code=None, payload=None):
+		Exception.__init__(self)
+		self.message = message
+		if status_code is not None:
+			self.status_code = status_code
+		self.payload = payload
+
+	def to_dict(self):
+		rv = dict(self.payload or ())
+		rv['message'] = self.message
+		return rv
+
+
+@app.errorhandler(ErrorResponse)
+def handle_invalid_usage(error):
+	response = jsonify(error.to_dict())
+	response.status_code = error.status_code
+	return response
+
+
+if __name__ == '__main__':
+	app.run(debug=True, host='0.0.0.0', port=8080)
