def remove(team_id):
try:
DBHelper.connect()
token = Token.get(Token.issuer == team_id)
delete_count = token.delete_instance()
finally:
DBHelper.close()
def store(team_id, auth_token, app):
try:
DBHelper.connect()
token, created = Token.get_or_create(issuer=team_id,
app=app,
defaults={'token': auth_token})
token.token = auth_token
token.save()
finally:
DBHelper.close()
def thanks():
DBHelper.create_db_tables() # trying to create db tables
"""
This route is called by Slack after the user installs our app. It will
exchange the temporary authorization code Slack sends for an OAuth token
which we'll save on the bot object to use later.
To let the user know what's happened it will also render a thank you page.
"""
# Let's grab that temporary authorization code Slack's sent us from
# the request's parameters.
code_arg = request.args.get('code')
# The bot's auth method to handles exchanging the code for an OAuth token
pyBot.auth(code_arg)
return render_template("thanks.html")
def db_test():
try:
DBHelper.connect()
DBHelper.create_db_tables()
return str(Token.select().count())
finally:
DBHelper.close()
def tearDown(self):
challenges = Challenge.query.filter(Challenge.isTest == True)
for challenge in challenges:
DBHelper.delete(challenge)
series = Series.query.filter(Series.isTest == True)
for serie in series:
DBHelper.delete(serie)
series = Series.query.filter(Series.name.contains("test"))
for serie in series:
DBHelper.delete(serie)
tasks = Task.query.filter(Task.isTest == True).filter(
Series.name.contains("test"))
for task in tasks:
DBHelper.delete(task)
asks = Task.query.filter(Task.name.contains("test"))
for task in tasks:
DBHelper.delete(task)
def createEntity(self, entity):
result = {}
try:
DBHelper.insert(entity, True)
result = {"id": entity.id}
except:
DBHelper.rollback()
print(sys.exc_info())
finally:
DBHelper.close()
return result
def delete_task(jwt, task_id):
error = False
try:
task = Task.query.get(task_id)
if task is None:
error = True
else:
DBHelper.delete(task)
except:
DBHelper.rollback()
erro = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def delete_series(jwt, series_id):
error = False
try:
series = Series.query.get(series_id)
if series is None:
error = True
else:
DBHelper.delete(series)
except:
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def delete_challenge(jwt, challenge_id):
error = False
try:
challenge = Challenge.query.get(challenge_id)
if challenge is None:
error = True
else:
DBHelper.delete(challenge)
except:
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def create_series(jwt):
data = request.get_json()
error = False
try:
series = Series(name=data["name"], description=data["description"])
isTest = False
if "isTest" in data:
isTest = data["isTest"]
DBHelper.insert(series)
except:
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def create_challenge(jwt):
data = request.get_json()
error = False
try:
challenge = Challenge(task_id=data["task_id"],
series_id=data["series_id"])
isTest = False
if "isTest" in data:
isTest = data["isTest"]
DBHelper.insert(challenge)
except:
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def create_task(jwt):
data = request.get_json()
error = False
try:
task = Task(name=data["name"],
state=data["state"],
country=data["country"])
isTest = False
if "isTest" in data:
isTest = data["isTest"]
DBHelper.insert(task, isTest)
except:
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def update_challenge(jwt, challenge_id):
data = request.get_json()
error = False
try:
challenge = Challenge.query.get(challenge_id)
if challenge is None:
error = True
else:
if "task_id" in data and "series_id" in data:
challenge.task_id = data["task_id"]
challenge.series_id = data["series_id"]
DBHelper.update()
except:
print(sys.exc_info())
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def update_series(jwt, series_id):
error = False
data = request.get_json()
try:
series = Series.query.get(series_id)
if series is None:
error = True
else:
if "name" in data:
series.name = data["name"]
if "description" in data:
series.state = data["description"]
DBHelper.update()
except:
print(sys.exc_info())
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
def update_task(jwt, task_id):
data = request.get_json()
error = False
try:
task = Task.query.get(task_id)
if task is None:
error = True
else:
if "name" in data:
task.name = data["name"]
if "state" in data:
task.state = data["state"]
if "country" in data:
task.country = data["country"]
DBHelper.update()
except:
print(sys.exc_info())
DBHelper.rollback()
error = True
finally:
DBHelper.close()
if error:
abort(404)
return jsonify({"success": True})
datetime.date.today().strftime(DateFormats.COMMON))
@application.route('/db_test')
def db_test():
try:
DBHelper.connect()
DBHelper.create_db_tables()
return str(Token.select().count())
finally:
DBHelper.close()
@application.route('/dialog', methods=["GET", "POST"])
def dialog():
google_dialog = GoogleActionDialog(request,
kult_menuholder.get_current_menu())
return google_dialog.handle()
@application.after_request
def after_request(response):
DBHelper.close() # close if necessary
return response
if __name__ == '__main__':
logging.getLogger().setLevel(logging.DEBUG)
DBHelper.create_db_tables()
application.run(debug=True)
def after_request(response):
DBHelper.close() # close if necessary
return response
def get(team_id):
try:
DBHelper.connect()
return Token.get(Token.issuer == team_id).token
finally:
DBHelper.close()