python-接口开发

import pymysql

import flask,json

def op_mysql(host,user,password,db,sql,port=3300,charset='utf8'):
   conn = pymysql.connect(host=host,user=user,
                     password=password,
                     port=port,
                     charset=charset,db=db)
   cur = conn.cursor(cursor=pymysql.cursors.DictCursor)
   cur.execute(sql)
   sql_start = sql[:6].upper() #取sql前6个字符串，判断它是什么类型的sql语句
   if sql_start=='SELECT' :
      res = cur.fetchall()
   else:
      conn.commit()
      res = 'ok'
   cur.close()
   conn.close()
   return res

server = flask.Flask(__name__) #把咱们这个app这个python文件当做一个server
@server.route('/get_user',methods=['get','post'])
def get_all_user():
   sql = 'select * from bt_stu;'
   res = op_mysql(
      host='200.100.200.00',
      user='jxz', password='123400',  # port这里一定要写int类型
      port=3300, db='jxz', charset='utf8', sql=sql)
   response = json.dumps(res,ensure_ascii=False) #把list转成json
   return response #return 的时候只能return字符串

@server.route('/add_user',methods=['post'])
def add_user():
   user_id = flask.request.values.get('id')  #这里的参数就是调用接口的时候传入的参数
   username = flask.request.values.get('u') #
   if user_id and username:
      sql = "insert into stu values ('%s','%s');"%(user_id,username)
      res = op_mysql(
         host='200.100.200.00',
         user='jxz', password='123400',  # port这里一定要写int类型
         port=3300, db='jxz', charset='utf8', sql=sql)
      response = {'code':308,'msg':'添加成功'}
   else:
      response = {'code':503,'msg':'必填参数未填！'}
   return json.dumps(response,ensure_ascii=False)


server.run(port=8080,debug=True)