def mysql_function():
mysql.connect()
# print(mysql.update("insert into test(text) values('测试数据2')"))
# mysql.commit()
data = mysql.select("select id,text from test")
for item in data:
print("id:%d\t text: %s" % (item[0], item[1]))
mysql.connect_close()
def email_delete(id):
try:
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute("DELETE FROM email where eid='{}'".format(id))
conn.commit()
resp = jsonify({'message': 'Email Deleted'})
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def email_read_all():
try:
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute("SELECT * from email;")
rows = cur.fetchall()
resp = jsonify(rows)
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def user_read(email):
try:
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute("SELECT * from user where email='{}';".format(email))
rows = cur.fetchall()
resp = jsonify(rows)
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def user_update(email):
try:
data = request.get_json()
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute("UPDATE user SET password='******' where email='{}';".format(
data['password'], email))
conn.commit()
resp = jsonify({'message': 'User Updated'})
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def user_create():
try:
data = request.get_json()
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute("INSERT INTO user VALUES('{}','{}')".format(
data["email"], data["password"]))
conn.commit()
resp = jsonify({'message': 'User Created'})
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def email_all(email):
try:
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute(
"SELECT * from email where (sender='{}' or receiver='{}')".format(
email, email))
rows = cur.fetchall()
resp = jsonify(rows)
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def email_create():
try:
data = request.get_json()
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute(
"INSERT INTO email VALUES(0,'{}','{}','{}','{}',{})".format(
data["subject"], data["body"], data["sender"],
data["receiver"], predict(data["body"])))
# insert into email values(0,'sub','message','sender','receiver',spam_detector)
conn.commit()
resp = jsonify({'message': 'Email Created'})
resp.status_code = 200
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
def login():
try:
data = request.get_json()
conn = mysql.connect()
cur = conn.cursor(pymysql.cursors.DictCursor)
cur.execute(
"SELECT * FROM user WHERE email='{}' and password='******'".format(
data["email"], data["password"]))
rows = cur.fetchall()
if (len(rows) > 0):
resp = jsonify({'message': 'login success'})
resp.status_code = 200
else:
resp = jsonify({'message': 'login failed'})
resp.status_code = 400
return resp
except Exception as e:
print(e)
finally:
cur.close()
conn.close()
