1 #从数据库获取数据 2 import pymysql,hashlib 3 import flask,json 4 5 def op_mysql(sql:str): 6 mysql_info = { 7 'host': 'xx.2xx.z.zz0', 8 'port': 3306, 9 'password': 'ssdfds6', 10 'user': 'xx', 11 'db': 'jxxx', 12 'charset': 'utf8', 13 'autocommit': True 14 } 15 result = '执行完成' 16 conn = pymysql.connect(**mysql_info) 17 cur = conn.cursor(pymysql.cursors.DictCursor) #建立游标 18 cur.execute(sql) 19 if sql.strip().lower().startswith('select'): 20 # result = cur.fetchone() 21 result = cur.fetchall() 22 cur.close() 23 conn.close() 24 return result 25 26 server = flask.Flask(__name__) 27 28 @server.route('/table_data') 29 def get_table_data(): 30 table_name = flask.request.args.get('table_name') 31 limit = flask.request.args.get('limit') 32 sql = 'select * from %s limit %s'%(table_name,limit) 33 result = op_mysql(sql) #结果是一个列表 34 #[{"id": 422, "username": "glw", "passwd": "123455", "is_admin": 123, "error_count": 0}, 35 # {"id": 424, "username": "glw1", "passwd": "123455", "is_admin": 123, "error_count": 0}, ] 36 return json.dumps(result,ensure_ascii=False) #list也可以转成json串 37 38 #优化 要是传参的时候 不传limit或者不传table_name或者传不存在的表名称或者limit传的不是数字 39 @server.route('/table_data') 40 def get_table_data(): 41 #没有实现校验表是否存在 42 #只能获取这几张表 43 tables = ['app_myuser','dsk_test app_student','pp_product'] 44 table_name = flask.request.args.get('table_name') 45 limit = flask.request.args.get('limit','10') 46 if table_name not in tables: 47 return json.dumps({'msg':'没有获取这个表的权限！'}) 48 if not table_name: 49 return json.dumps({'msg':'table_name是必填字段！'},ensure_ascii=False) 50 if limit.isdigit(): 51 sql = 'select * from %s limit %s'%(table_name,limit) 52 else: 53 return json.dumps({'msg':'limit请传入一个整数！'},ensure_ascii=False) 54 result = op_mysql(sql) 55 return json.dumps(result, ensure_ascii=False) # list也可以转成json串 56 57 def md5(s,): 58 s = (str(s)+'#$%$%').encode() 59 m = hashlib.md5(s)#加密 60 return m.hexdigest() 61 62 63 #实现添加成员 #并且密码加密存数据库 64 @server.route('/add_mem',methods=['post']) 65 def add_mem(): 66 username = flask.request.json.get('username') 67 password = flask.request.json.get('password') 68 if username and password: 69 sql = 'select * from app_myuser where username = "%s"' %username 70 if op_mysql(sql): 71 data = {'msg':'用户已存在！'} 72 else: 73 new_password = md5(password) 74 insert_sql = 'insert into app_myuser(username,passwd) value ("%s","%s")'%(username,new_password) 75 data = {'msg':'用户添加成功！'} 76 op_mysql(insert_sql) 77 else: 78 data = {'msg':'必填参数未填，请查看接口文档！'} 79 return json.dumps(data,ensure_ascii=False)