a few small fixes to ekim's comments

thrimshim/thrimshim/main.py

@@ -5,10 +5,11 @@ import signal
 import uuid
 
 import argh
+import flask
 import gevent
 import gevent.backdoor
 from gevent.pywsgi import WSGIServer
-import flask
+import prometheus_client
 from psycopg2 import sql
 
 import common
@@ -33,6 +34,12 @@ def cors(app):
 	return handle
 
 
+@app.route('/metrics')
+def metrics():
+	"""Expose Prometheus metrics."""
+	return prometheus_client.generate_latest()
+
+
 @app.route('/thrimshim/<ident>', methods=['GET', 'POST'])
 def thrimshim(ident):
 	"""Comunicate between Thrimbletrimmer and the Wubloader database."""
@@ -100,11 +107,10 @@ def update_row(ident, new_row):
 
 	conn = app.db_manager.get_conn()
 	#check a row with id = ident is in the database
-	with database.transaction(conn):
+	results = database.query(conn, """
-		results = database.query(conn, """
+		SELECT id, state 
-			SELECT id, state 
+		FROM events
-			FROM events
+		WHERE id = %s""", ident)
-			WHERE id = %s;""", ident)
 	old_row = results.fetchone()
 	if old_row is None:
 		return 'Row {} not found'.format(ident), 404
@@ -144,19 +150,9 @@ def update_row(ident, new_row):
 				sql.Identifier(column), sql.Placeholder(column),
 			) for column in new_row.keys()
 		))
-	with database.transaction(conn):
+	result = database.query(conn, build_query, id=ident, **new_row)
-		result = database.query(conn, build_query, id=ident, **new_row)
 	if result.rowcount != 1:
-		if result.rowcount == 0.:
+		return 'Video likely already published', 403	
-			with database.transaction(conn):
-				check_result = database.query(conn, """
-					SELECT id, state
-					FROM events
-					WHERE id = %s;""", ident)
-			current_row = check_result.fetchone()
-			if current_row.state not in ['UNEDITED', 'EDITED', 'CLAIMED']:
-				return 'Video already published', 403	
-		raise Exception('Database consistancy error for id = {}'.format(ident))
 			
 	logging.info('Row {} updated to state {}'.format(ident, new_row['state']))
 	return ''