從數據庫獲取數據-接口開發

#從數據庫獲取數據
import pymysql,hashlib
import flask,json

def op_mysql(sql:str):
    mysql_info = {
        'host': 'xx.2xx.z.zz0',
        'port': 3306,
        'password': 'ssdfds6',
        'user': 'xx',
        'db': 'jxxx',
        'charset': 'utf8',
        'autocommit': True
    }
    result = '執行完成'
    conn = pymysql.connect(**mysql_info)
    cur = conn.cursor(pymysql.cursors.DictCursor) #建立游標
    cur.execute(sql)
    if sql.strip().lower().startswith('select'):
        # result  = cur.fetchone()
        result  = cur.fetchall()
    cur.close()
    conn.close()
    return result

server = flask.Flask(__name__)

@server.route('/table_data')
def get_table_data():
    table_name = flask.request.args.get('table_name')
    limit = flask.request.args.get('limit')
    sql = 'select * from %s limit %s'%(table_name,limit)
    result = op_mysql(sql) #結果是一個列表
 #[{"id": 422, "username": "glw", "passwd": "123455", "is_admin": 123, "error_count": 0},
# {"id": 424, "username": "glw1", "passwd": "123455", "is_admin": 123, "error_count": 0}, ]
    return json.dumps(result,ensure_ascii=False) #list也可以轉成json串

#優化 要是傳參的時候 不傳limit或者不傳table_name或者傳不存在的表名稱或者limit傳的不是數字
@server.route('/table_data')
def get_table_data():
    #沒有實現校驗表是否存在
    #只能獲取這幾張表
    tables = ['app_myuser','dsk_test app_student','pp_product']
    table_name = flask.request.args.get('table_name')
    limit = flask.request.args.get('limit','10')
    if table_name not in tables:
        return json.dumps({'msg':'沒有獲取這個表的權限！'})
    if not table_name:
        return json.dumps({'msg':'table_name是必填字段！'},ensure_ascii=False)
    if limit.isdigit():
        sql = 'select * from %s limit %s'%(table_name,limit)
    else:
        return json.dumps({'msg':'limit請傳入一個整數！'},ensure_ascii=False)
    result = op_mysql(sql)
    return json.dumps(result, ensure_ascii=False)  # list也可以轉成json串

def md5(s,):
    s = (str(s)+'#$%$%').encode()
    m = hashlib.md5(s)#加密
    return m.hexdigest()


#實現添加成員 #並且密碼加密存數據庫
@server.route('/add_mem',methods=['post'])
def add_mem():
    username = flask.request.json.get('username')
    password = flask.request.json.get('password')
    if username and password:
        sql = 'select * from app_myuser where username = "%s"' %username
        if op_mysql(sql):
            data = {'msg':'用戶已存在！'}
        else:
            new_password = md5(password)
            insert_sql = 'insert into app_myuser(username,passwd) value ("%s","%s")'%(username,new_password)
            data = {'msg':'用戶添加成功！'}
            op_mysql(insert_sql)
    else:
        data = {'msg':'必填參數未填，請查看接口文檔！'}
    return json.dumps(data,ensure_ascii=False)