def new_password_t(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() sql = "SELECT password from staff where id=%s" id = L['id'] password = L['old_passwd'] new_password = L['new_passwd'] cursor.execute(sql, id) ((results, ), ) = cursor.fetchall() if check_password_hash(results, password): new_password = generate_password_hash(new_password) sql = "UPDATE staff set password=%s WHERE id = %s " cursor.execute(sql, (new_password, id)) db.commit() db.close() return '200' else: return '300'
def login(): """Render website's home page.""" # print("This is the session in login:"******"logged_in"]) if 'logged_in' in session: return redirect(url_for('home')) loginform = LoginForm() if request.method == "POST" and loginform.validate_on_submit(): username = loginform.username.data password = loginform.password.data db, cur = DB_Conn() q = 'select * from users_ where uname="{}" and passcode="{}";' q = q.format(username, password) cur.execute(q) r = cur.fetchall() cur.close print("this:", r) db.close() if r == []: flash("Incorrect username or password", "danger") redirect("login.html") else: session['logged_in'] = True session['user_id'] = r[0][0] flash('You were logged in') print("This is the session before home:", session["logged_in"]) return redirect(url_for('home')) flash_errors(loginform) return render_template('login.html', loginform=loginform)
def select_achievement(): db = pymysql.connect("127.0.0.1", "root", "638436", "adms") cursor = db.cursor() # username = "******" # r_mname = "ADNI_002_S_0295_MR_HarP_135_final_release_2015_Br_20150226095012465_S13408_I474728.nii" # mri = "select * from mri where m_name='" + r_mname + "'" # 方法一: # select Score, (select count(distinct Score) from Scores where Score>=s.Score) as Rank from Scores as s order by Score desc; # rank = "select ea_eid, ea_score, (select count(distinct ea_score) from eachievement where ea_score >= s.ea_score) " \ # "as Rank from eachievement as s order by ea_score desc" username = "******" resu = "select * from resu where r_uid='" + username + "'" cursor.execute(resu) rhipvlou_r = cursor.fetchall() print(rhipvlou_r) data = [] for i in rhipvlou_r: x_data = "海马体数量" + "-" + i[4].strftime('%Y.%m.%d') y_data = int(i[6]) print(x_data) print(y_data) data_dict = {"name":x_data,"num":y_data} data.append(data_dict) print(data) # r_uid = mri_r[0][1] # r_utime = mri_r[0][3] # print(mri_r) # print(r_uid) # print(r_utime) db.commit() db.close()
def Movie(title): """Render website's home page.""" if 'logged_in' in session: s = request.args.get('feedback') print("here2:", s) db, cur = DB_Conn() if s != None: #print "t: %s || s: %s" % (title, s) r = RekoCommander.RunCommand('5', session['user_id'], title, int(s) * 2) print "RekoAPI response:", r pass q = "SELECT * FROM {} WHERE tconst_='{}';" q = q.format("titles", title) cur.execute(q) res = cur.fetchall() #print("query result: %s" % res) # here is the error now img = GetTitleImage(title.strip()) db.close() return render_template('movie.html', themovie=res[0], imgurl=img, title=title) else: flash("Please log in first", "danger") return redirect(url_for("login"))
def heaven_for_users(): try: db = pymysql.connect(host="rm-uf6gw23j409s5ui7qmoo.mysql.rds.aliyuncs.com", port=3306, user="******", password="******", charset="utf8") res = request.get_json() user_id = res.get('user_id') open_id = res.get('open_id') new_user_id = res.get('new_user_id') new_open_id = res.get('new_open_id') if not all([user_id, open_id, new_user_id, new_open_id]): return jsonify(errno=-1, errmsg="参数不完整") cursor = db.cursor() timeStamp = int(time.time()) dateArray = datetime.datetime.utcfromtimestamp(timeStamp) otherStyleTime = dateArray.strftime("%Y-%m-%d %H:%M:%S") sql = f"INSERT INTO zjlivenew.zj_heaven_for_user(create_time,update_time,user_id,openid,new_user_id,new_openid,status)" \ f"VALUES('{otherStyleTime}','{otherStyleTime}',{int(user_id)},'{open_id}',{int(new_user_id)},'{new_open_id}',{int(0)})" print(sql) cursor.execute(sql) db.commit() db.close() return jsonify(errno=0, errmsg="OK") except Exception as e: Logging.logger.error('errmsg:{0}'.format(e)) return jsonify(errno=-1, errmsg='网络异常')
def promote_user(email, username, password): """Make a new admin""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT COUNT(*) FROM user') users = cur.fetchone() if users == (1, ): cur.execute('UPDATE user SET enable = True WHERE email = %s', (email, )) cur.execute('UPDATE user SET is_admin = True WHERE email = %s', (email, )) db.commit() db.close() click.echo('User (' + email + ') enabled and admin now.') else: click.echo('Access denied.') else: click.echo('Invalid password')
def add_players(tournamentID): form = AddPlayers(request.form) db = get_db() if request.method == 'POST': db.close() return redirect(url_for("round", tournamentID=tournamentID, round_num=1)) return render_template('add_players.html', form=form)
def newNotebook(): if current_user.enable: res = requests.post(f'{APISERVER}/api/notebook', json={'name': request.form['name']}) if res.status_code == 201: dates = json.loads(res.text) notebook_id = dates.get('id') db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'INSERT INTO user_object_id (object_id, object_type, user_name) VALUES (%s,%s,%s)', ( notebook_id, "notebook", current_user.username, )) db.commit() db.close() flash(f'Notebook created!') else: flash(res.status_code) return redirect(url_for('notebooks')) else: return redirect(url_for('403'))
def init_db(database=None): if not database: database = app.config["DATABASE"] db.init(database) db.connect() db.create_tables([Author, Content]) Author.create(id=1, name="author") db.close()
def newEndpoint(): if current_user.enable: name = request.form['name'] training_id = request.form['training_id'] db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'SELECT * FROM user_object_id WHERE object_id =%s AND object_type=%s AND user_name=%s', ( training_id, "training", current_user.username, )) training = cur.fetchone() db.close() if training == None: flash('Invalid training id') else: training_id = training[1] res = requests.post(f'{APISERVER}/api/endpoints/endpoint', json={ 'name': name, 'training_id': training_id }) if res.status_code == 201: dates = json.loads(res.text) endpoint_id = dates.get('id') db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'INSERT INTO user_object_id (object_id, object_type, user_name) VALUES (%s,%s,%s)', ( endpoint_id, "endpoint", current_user.username, )) db.commit() db.close() flash(f'Endpoint created!') else: flash(res.status_code) return redirect(url_for('endpoints')) else: return redirect(url_for('403'))
def questions(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() id = L["id"] cursor.execute("SELECT c_id from Enrolment where s_id=%s", id) course = cursor.fetchall() print(course) if len(course) != 3: return '404' questions = [] for i in course: cursor.execute( "SELECT question,answer FROM new_answer_new where course_id = %s ORDER BY time desc LIMIT 0,3", i) result = cursor.fetchall() questions.append([]) for j in range(len(result)): questions[-1].append(result[j][0]) questions[-1].append(result[j][1]) print(questions) qst = { "c1": course[0], "question11": questions[0][0], "answer11": questions[0][1], "question12": questions[0][2], "answer12": questions[0][3], "question13": questions[0][4], "answer13": questions[0][5], "c2": course[1], "question21": questions[1][0], "answer21": questions[1][1], "question22": questions[1][2], "answer22": questions[1][3], "question23": questions[1][4], "answer23": questions[1][5], "c3": course[2], "question31": questions[2][0], "answer31": questions[2][1], "question32": questions[2][2], "answer32": questions[2][3], "question33": questions[2][4], "answer33": questions[2][5] } print(qst) qst = json.dumps(qst) db.close() return qst
def load_tournament(): db = get_db() cur = db.execute('SELECT id, name, round_num FROM tournament') TOURNS = cur.fetchall() db.close() if request.method == 'POST': tourn_to_load = request.form['tourn'] return redirect(url_for("round", tournamentID=tourn_to_load, round_num=2)) return render_template('load_tournament.html', tourns=TOURNS)
def sign_in(): form = RegisterForm() if request.method == 'GET': return render_template("register.html", title='register', form=form) else: db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() sql = [ "INSERT INTO student(id,email,password,name) VALUES (%s,%s,%s,%s)", "INSERT INTO staff(id,email,password,name) VALUES (%s,%s,%s,%s)" ] if form.validate_on_submit(): id = form.id.data name = form.name.data email = form.email.data password = form.password.data SorT = form.SorT.data password = generate_password_hash(password) if SorT == 1: cursor.execute("SELECT * from student where id=%s", id) result = cursor.fetchall() if len(result) != 0: flash( 'Registration failed, zID has been registered, please log in directly.', 'error') return render_template("register.html", title='register', form=form) else: cursor.execute(sql[0], (id, email, password, name)) else: cursor.execute("SELECT * from staff where id=%s", id) result = cursor.fetchall() if len(result) != 0: flash( 'Registration failed, zID has been registered, please log in directly.', 'error') return render_template("register.html", title='register', form=form) else: cursor.execute(sql[1], (id, email, password, name)) db.commit() db.close() # flash('Congratulations, your register are success.', 'info') return redirect(url_for('login')) db.rollback() db.close() flash('Sorry, your register not success, please try again.', 'error') return render_template("register.html", title='register', form=form)
def home(): conn = sqlite3.connect(DATABASE) TOURNS = conn.execute('SELECT * FROM tournament').fetchall() if request.method == 'POST': if request.form['choice'] == 'Create new': return redirect(url_for("create_tournament")) db.close() elif request.form['choice'] == 'Load old': # go to load pages and pull in all tournament entries from the database return redirect(url_for("load_tournament")) return render_template('index.html', tourns=TOURNS, title="Main menu")
def satistics(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() id = L["id"] cursor.execute("SELECT c_id from Teaching where T_id=%s", id) course = cursor.fetchall() print(course) if len(course) != 3: return '404' sat = [] for i in course: cursor.execute( "SELECT COUNT(question) FROM new_answer_new where course_id = %s ", i) result = cursor.fetchall() sat.append(int(re.findall(r"\d+", str(result[0]))[0])) cursor.execute( "SELECT COUNT(question) FROM new_question where course_id = %s ", i) result = cursor.fetchall() sat.append(int(re.findall(r"\d+", str(result[0]))[0])) print(sat) s1 = int((sat[0] / (sat[0] + sat[1])) * 100) s2 = int((sat[2] / (sat[2] + sat[3])) * 100) s3 = int((sat[4] / (sat[4] + sat[5])) * 100) satistics = { "c1": course[0], "q1": sat[0], "qq1": sat[1], "s1": s1, "c2": course[1], "q2": sat[2], "qq2": sat[3], "s2": s2, "c3": course[2], "q3": sat[4], "qq3": sat[5], "s3": s3, } print(satistics) satistics = json.dumps(satistics) db.close() return satistics
def db_create(): db.connect() User.create_table(True) # True means fail siliently if table exists SocialCounter.create_table(True) SocialCount.create_table(True) ChannelCounter.create_table(True) Channel.create_table(True) UserFunction.create_table(True) FunctionResult.create_table(True) Crawler.create_table(True) CrawlerPage.create_table(True) Mozscape.create_table(True) MozscapeResult.create_table(True) MozscapeIndexMetadata.create_table(True) db.close()
def select_japply(): db = pymysql.connect("127.0.0.1", "root", "638436", "adms") cursor = db.cursor() username = "******" r_mname = "ADNI_002_S_0295_MR_HarP_135_final_release_2015_Br_20150226095012465_S13408_I474728.nii" mri = "select * from mri where m_name='" + r_mname + "'" cursor.execute(mri) mri_r = cursor.fetchall() r_uid = mri_r[0][1] r_utime = mri_r[0][3] print(mri_r) print(r_uid) print(r_utime) db.commit() db.close()
def deleteTraining(username, id_to_delete, password): """Delete endpoint""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): if user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'SELECT * FROM user_object_id WHERE user_name = %s AND object_id = %s', (user.username, id_to_delete)) endpoint_to_delete = cur.fetchone() db.close() if endpoint_to_delete: res = requests.delete( f'{APISERVER}/api/endpoint/{id_to_delete}') if res.status_code == 200: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'DELETE FROM user_object_id WHERE object_id = %s AND object_type = %s', ( id_to_delete, "notebook", )) db.commit() db.close() click.echo('\nEndpoint eliminated!\n') else: click.echo(res.status_code) else: click.echo('Invalid endpoint_id') else: click.echo('You can\'t delete endpoint') else: click.echo('Invalid password')
def read_sql(): list = [] db = pymysql.connect("127.0.0.1", "root", "123456", "db_demo1") # SQL 查询语句 sql = "SELECT terminalid FROM CAR " try: cursor = db.cursor() cursor.execute(sql) result = cursor.fetchall() db.close() except: logging.info(u"数据路查询异常") read_sql() for obj in result: list.append(obj[0]) return list
def get_data(self,db_name,table_name): global data sql_columns_name=str("desc %s" %table_name) db=pymysql.connect("localhost","root","123456",db_name,charset = 'utf8') cursor=db.cursor() cursor.execute(sql_columns_name) columns_name =list(cursor.fetchall()) sql_data_count=str("select count(*) from %s" %table_name) cursor.execute(sql_data_count) data_count=list(cursor.fetchall()) sql_data=str("select * from %s" %table_name) cursor.execute(sql_data) data = pd.DataFrame(np.array(cursor.fetchall())) data.columns=[np.array(columns_name)[:,0]] db.close() return columns_name,data_count,data
def newTraining(username, name, file, password): """Create a new training""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): if user.enable: upload_file = os.environ.get("UPLOAD_FOLDER") files = {'file': file} print(upload_file) print(file) print(files) res = requests.post(f'{APISERVER}/api/training/{name}', files=files) if res.status_code == 201: dates = json.loads(res.text) training_id = dates.get('id') db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'INSERT INTO user_object_id (object_id, object_type, user_name) VALUES (%s,%s,%s)', ( training_id, "training", user.username, )) db.commit() db.close() click.echo('Training created!\n') click.echo(res.content) else: click.echo(res.status_code) else: click.echo('You can\'t create training') else: click.echo('Invalid password')
def loop_run(self): count = 0 while 1: try: if count > 60: db.connect() maxid = self.get_maxid() self.delete(maxid - self.rest) count = 0 db.close() else: count += 1 except Exception as e: logger.error(e) time.sleep(1) finally: time.sleep(1) if self.is_quit: break
def create_tournament(): form = CreateTournament(request.form) db = get_db() if request.method == 'POST': nam = request.form['tourn_name'] loc = request.form['location'] cal = request.form['calendar'] sys = request.form['system'] tie = request.form['tie_break'] db.execute('''INSERT INTO tournament \ (id, name, location, calendar, system, tie_break, round_num) \ VALUES (NULL, ?, ?, ?, ?, ?, ?)''', (nam, loc, cal, sys, tie, 1)) db.commit() tournamentID = str( db.execute('''SELECT max(id) FROM tournament''').fetchone()[0]) db.close() return redirect(url_for('add_players', tournamentID=tournamentID)) return render_template('create_tournament.html', title='Create new tournament', form=form)
def new_email_t(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() sql = "SELECT email from staff where id=%s" id = L['id'] email = L['email'] cursor.execute(sql, id) sql = "UPDATE staff set email=%s WHERE id = %s " cursor.execute(sql, (email, id)) db.commit() db.close() return '200'
def notebooks(username, password): """Show your notebook list""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): if user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() notebooks = [] for notebook in objects: if notebook[3] == user.username and notebook[2] == "notebook": notebook_id = notebook[1] res = requests.get( f'{APISERVER}/api/notebook/{notebook_id}').content notebook = json.loads(res) notebooks.append(notebook) else: if user.is_admin and notebook[2] == "notebook": notebook_id = notebook[1] res = requests.get( f'{APISERVER}/api/notebook/{notebook_id}').content notebook = json.loads(res) notebooks.append(notebook) click.echo(notebooks) else: click.echo('You can\'t create notebook') else: click.echo('Invalid password')
def endpoints(username, password): """Show your endpoint list""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): if user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() endpoints = [] for endpoint in objects: if endpoint[3] == user.username and endpoint[2] == "endpoint": endpoint_id = endpoint[1] res = requests.get( f'{APISERVER}/api/endpoint/{endpoint_id}').content endpoint = json.loads(res) endpoints.append(endpoint) else: if user.is_admin and endpoint[2] == "endpoint": endpoint_id = endpoint[1] res = requests.get( f'{APISERVER}/api/endpoint/{endpoint_id}').content endpoint = json.loads(res) endpoints.append(endpoint) click.echo(endpoints) else: click.echo('You can\'t create endpoint') else: click.echo('Invalid password')
def classes_t(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() id = L["id"] cursor.execute("SELECT c_id from Teaching where T_id=%s", id) result = cursor.fetchall() print(result) if len(result) == 3: classes = {"c0": result[0], "c1": result[1], "c2": result[2]} return json.dumps(classes) db.commit() db.close() return "404"
def up_course_t(): L = request.get_data() L = json.loads(L) db = pymysql.connect( host='database-1.clr3d8nnckz4.us-east-2.rds.amazonaws.com', user='******', password='******', port=3306, db='chatbot') cursor = db.cursor() id = L["id"] cursor.execute("INSERT INTO Teaching (T_id,c_id) VALUES (%s,%s)", (id, str.upper(L["course1"]))) cursor.execute("INSERT INTO Teaching (T_id,c_id) VALUES (%s,%s)", (id, str.upper(L["course2"]))) cursor.execute("INSERT INTO Teaching (T_id,c_id) VALUES (%s,%s)", (id, str.upper(L["course3"]))) db.commit() db.close() return "200"
def deleteEndpoint(): if current_user.enable: id_to_delete = request.form['id_to_delete'] db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'SELECT * FROM user_object_id WHERE user_name = %s AND object_id = %s', (current_user.username, id_to_delete)) endpoint_to_delete = cur.fetchone() db.close() if endpoint_to_delete or current_user.is_admin: res = requests.delete(f'{APISERVER}/api/endpoint/{id_to_delete}') if res.status_code == 200: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'DELETE FROM user_object_id WHERE object_id = %s AND object_type = %s', ( id_to_delete, "endpoint", )) db.commit() db.close() flash(f'Endpoint eliminated!') else: flash(res.status_code) return redirect(url_for('endpoints')) else: return redirect(url_for('403'))
def init_db(): '''Initial database''' db.connect() print('Creating tables ...') db.create_tables(tables) print('Tables have been created.') print('Writing default configurations ...') Config.create(name='site_name', value=app.config['DEFAULT_SITE_NAME']) Config.create(name='site_url', value=app.config['DEFAULT_SITE_URL']) Config.create(name='count_topic', value=app.config['DEFAULT_TOPICS_PER_PAGE']) Config.create(name='count_post', value=app.config['DEFAULT_POSTS_PER_PAGE']) Config.create(name='count_list_item', value=app.config['DEFAULT_LIST_ITEM_PER_PAGE']) # Config.create(name='count_subpost', value=app.config['DEFAULT_SUBPOSTS_PER_PAGE']) print('Default configurations have been written into database.') db.close()
def trainings(): if current_user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() trainings = [] for training in objects: if training[3] == current_user.username and training[ 2] == "training": training_id = training[1] res = requests.get( f'{APISERVER}/api/training/{training_id}').content training = json.loads(res) trainings.append(training) else: if current_user.is_admin and training[2] == "training": training_id = training[1] res = requests.get( f'{APISERVER}/api/training/{training_id}').content training = json.loads(res) trainings.append(training) return render_template("trainings.html", title="Trainings", trainings=trainings) else: return redirect(url_for('403'))
def trainings(username, password): """Show your training list""" user = User.query.filter_by(username=username).first_or_404( description='Invalid username') if user.check_password(password): if user.enable == True: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() trainings = [] for training in objects: if training[3] == user.username and training[2] == "training": training_id = training[1] res = requests.get( f'{APISERVER}/api/training/{training_id}').content training = json.loads(res) trainings.append(training) elif user.is_admin and training[2] == "training": training_id = training[1] res = requests.get( f'{APISERVER}/api/training/{training_id}').content training = json.loads(res) trainings.append(training) click.echo(trainings) else: click.echo('You can\'t create training') else: click.echo('Invalid password')
def newTraining(): if current_user.enable: name = request.form['name'] file = request.files['file'] upload_file = os.environ.get("UPLOAD_FOLDER") file_path = os.path.join(upload_file, file.filename) file.save(file_path) files = {'file': open(file_path, 'rb')} res = requests.post(f'{APISERVER}/api/training/{name}', files=files) if res.status_code == 201: dates = json.loads(res.text) training_id = dates.get('id') db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute( 'INSERT INTO user_object_id (object_id, object_type, user_name) VALUES (%s,%s,%s)', ( training_id, "training", current_user.username, )) db.commit() db.close() flash(f'Training created!') else: flash(res.status_code) return redirect(url_for('trainings')) else: return redirect(url_for('403'))
def notebooks(): if current_user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() notebooks = [] for notebook in objects: if notebook[3] == current_user.username and notebook[ 2] == "notebook": notebook_id = notebook[1] res = requests.get( f'{APISERVER}/api/notebook/{notebook_id}').content notebook = json.loads(res) notebooks.append(notebook) else: if current_user.is_admin and notebook[2] == "notebook": notebook_id = notebook[1] res = requests.get( f'{APISERVER}/api/notebook/{notebook_id}').content notebook = json.loads(res) notebooks.append(notebook) return render_template("notebooks.html", title="Notebooks", notebooks=notebooks) else: return redirect(url_for('403'))
def endpoints(): if current_user.enable: db = mysql.connect(host=MYSQL_HOST, user=MYSQL_USER, password=MYSQL_PSSW, database=MYSQL_DB) cur = db.cursor() cur.execute('SELECT * FROM user_object_id') objects = cur.fetchall() db.close() endpoints = [] for endpoint in objects: if endpoint[3] == current_user.username and endpoint[ 2] == "endpoint": endpoint_id = endpoint[1] res = requests.get( f'{APISERVER}/api/endpoint/{endpoint_id}').content endpoint = json.loads(res) endpoints.append(endpoint) else: if current_user.is_admin and endpoint[2] == "endpoint": endpoint_id = endpoint[1] res = requests.get( f'{APISERVER}/api/endpoint/{endpoint_id}').content endpoint = json.loads(res) endpoints.append(endpoint) return render_template("endpoints.html", title="Endpoints", endpoints=endpoints) else: return redirect(url_for('403'))
def teardown_request(exception): """Commits again and closes the database again at the end of the request.""" db = getattr(g, '_database', None) if db is not None: db.close()
from app import app, db from app.models import * db.connect() for table in [ Track, User, Slot, SlotRequest, PlaylistTrack, Wish, RadioStation ]: db.create_table( table, safe = True ) db.close()
def __del__(self): db.close()
def teardown_request(exception): db = getattr(g, 'db', None) if db is not None: db.close()
def initialize(): db.connect() db.create_tables([Match, MatchPlayer, Player], safe=True) db.close()