from flask import Flask, json, request, jsonify, redirect, url_for

import json
import pymysql
import pymysql.cursors
app = Flask(__name__)
def db_connection():
conn = None
try:
conn = pymysql.connect(host='database-1.cmxzj2fchhnw.ap-southeast-
1.rds.amazonaws.com',
database='database_1',
user='admin',
password='nhattan2512',
charset='utf8mb4',
cursorclass=pymysql.cursors.DictCursor)

except pymysql.Error as e:
print(e)
return conn
@app.route('/account/',methods=["GET"])
def single_account():
conn = db_connection()
cursor = conn.cursor()
username = request.args.get('username')
password = request.args.get('password')
if request.method == "GET":
cursor.execute("""SELECT `balance` FROM `account` WHERE username ='%s' AND
password ='%s'""" % (username,password))
balance = cursor.fetchone()
if balance is not None:
return balance ,202
else:
return "NO",404
return "NO",404

@app.route('/payment/',methods=["GET"])
def transaction():
conn = db_connection()
cursor = conn.cursor()
username = request.args.get('username')
busid = request.args.get('busid')
balance_later = dict()
if request.method == "GET":
cursor.execute("""SELECT`type_ticket` FROM `account` WHERE username
='%s'""" % (username))
type_ticket = cursor.fetchone()
cursor.execute("""SELECT `balance` FROM `account` WHERE username ='%s'""" %
(username))
balance = cursor.fetchone()
if type_ticket == {"type_ticket": "price1"}:
cursor.execute("""SELECT `price1` FROM bus_system WHERE busid ='%s'"""
%(busid))
price = cursor.fetchone()
balance_later['balance_later'] = balance['balance'] - price['price1']
if balance_later['balance_later'] > 0:
sql = "INSERT INTO
`transaction`(`username`,`busid`,`type_ticket`,`balance`,`price`,`balance_later`)
VALUES (%s, %s, %s, %s, %s, %s)"
cursor.execute(sql, (username, busid, type_ticket['type_ticket'],
balance['balance'],price['price1'],balance_later['balance_later']))
cursor.execute("UPDATE `account` set balance = %s where username =
%s", (balance_later['balance_later'],username))
conn.commit()
return "Thanh toan thanh cong"
else:
return "Khong the thanh toan"
else:
cursor.execute("""SELECT `price2` FROM bus_system WHERE busid ='%s'"""
%(busid))
price = cursor.fetchone()
balance_later['balance_later'] = balance['balance'] - price['price2']
if balance_later['balance_later'] > 0:
sql = "INSERT INTO
`transaction`(`username`,`busid`,`type_ticket`,`balance`,`price`,`balance_later`)
VALUES (%s, %s, %s, %s, %s, %s)"
cursor.execute(sql, (username, busid, type_ticket['type_ticket'],
balance['balance'],price['price2'],balance_later['balance_later']))
cursor.execute("UPDATE `account` set balance = %s where username =
%s", (balance_later['balance_later'],username))
conn.commit()
return "Thanh toan thanh cong"
else:
return "Khong the thanh toan"
if __name__ == '__main__':
app.run(host = '0.0.0.0', port = 7000, debug=True)