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()