接口开发

#!/usr/bin/env python
# -*- coding: utf-8 -*-
MYSQL_INFO = {
    ‘host‘: ‘127.0.0.1‘,
    ‘user‘: ‘jxz‘,
    ‘password‘: ‘123456‘,
    ‘db‘: ‘jxz‘,
    ‘port‘: 3306,
    ‘charset‘: ‘utf8‘
}   # MySQL 的配置参数

REDIS_INFO = {
    ‘host‘: ‘127.0.0.1‘,
    ‘password‘: ‘HK139bc&*‘,
    ‘port‘: 6379,
    ‘db‘: 14
}     # redis 配置参数

SERVER_PORT = 8999      # 服务端端口

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import pymysql, redis, hashlib, time
from config.setting import MYSQL_INFO, REDIS_INFO
def my_db(sql):    # 传入sql语句，操作数据库
    conn = pymysql.connect( **MYSQL_INFO)
    cur = conn.cursor()
    cur.execute(sql)
    if sql.strip()[:6].upper() == ‘SELECT‘:
        res = cur.fetchall()    # 获取数据库的数据
    else:
        conn.commit()   # 除select语句外的其他语句
        res = ‘OK‘
    cur.close()
    conn.close()
    return res

def op_redis(k, v=None, seconds=None):  # 操作redis，如果只传入key,则获取value，
    # 如果传入key, value，将key,value写入redis,如果三个参数都有，则写入redis，有时限
    r = redis.Redis( **REDIS_INFO)
    if v and seconds:
        r.setex(k, v, seconds)
        res = ‘OK‘
    elif v :
        r.set(k, v)
        res = ‘OK‘
    else:
        if r.get(k):
            res = r.get(k).decode()
        else:
            res = None
    return res

def my_md5(s, salt=None):   # 如果传入 salt,则加密原字符串加salt，不传入salt,则只加密原字符串
    md = hashlib.md5()
    if salt:
        md.update((s + salt).encode())
        res = md.hexdigest()
    else:
        md.update(s.encode())
        res = md.hexdigest()
    return res

def my_time():      # 返回时间，格式如 20180508151213 2018-5-8 15:12:13
    a = time.localtime()
    res = str(a.tm_year) + str(a.tm_mon).zfill(2) + str(a.tm_mday).zfill(2) + str(a.tm_hour).zfill(2) + str(
        a.tm_min).zfill(2) + str(a.tm_sec).zfill(2)
    return res

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import flask, json
from lib.tools import my_md5, my_time, op_redis

server = flask.Flask(__name__)  # 把当前文件当作一个服务

@server.route(‘/reg‘, methods=[‘post‘])
def reg():
    username = flask.request.values.get(‘username‘) # 获取输入值
    pwd  = flask.request.values.get(‘password‘)
    k = ‘user:‘ + username
    v = my_md5(pwd)
    # print(‘username..‘, username)
    if username and pwd:
        if op_redis(k):
            res = {‘msg‘: ‘用户已存在‘, ‘msg_code‘: 2001}
        else:
            op_redis(k, v)
            res = {‘msg‘: ‘注册成功！‘, ‘msg_code‘: 0}
    else:
        res = {‘msg‘: ‘必填字段未填，请查看接口文档！‘, ‘msg_code‘: 1001}
    return json.dumps(res, ensure_ascii=False)


@server.route(‘/login‘, methods=[‘post‘])
def login():
    username = flask.request.values.get(‘username‘) # 获取输入值
    pwd  = flask.request.values.get(‘password‘)
    pwd1 = my_md5(pwd)
    k = ‘user:‘ + username
    v = {‘session_id‘: my_md5(pwd + my_time()), ‘login_time‘: my_time()}
    print(‘username..‘, username)
    if username and pwd:
        if op_redis(k) == pwd1:
            k = ‘login:‘ + username
            if op_redis(k):     # 如果 key 已存在，即已登陆状态
                res = op_redis(k).replace(‘登陆成功‘, ‘已登陆‘)
            else:               # key 不存在，未登陆
                v[‘status‘] = ‘登陆成功‘
                op_redis(k, v, 3600)
                res = op_redis(k)
        else:
            res = "{‘msg‘: ‘账号或密码错误！‘, ‘msg_code‘: 3001}"

    else:
        res = "{‘msg‘: ‘必填字段未填，请查看接口文档！‘, ‘msg_code‘: 1001}"
    return json.dumps(res, ensure_ascii=False)

#!/usr/bin/env python
# -*- coding: utf-8 -*-

from lib.interface import server
from config.setting import SERVER_PORT

server.run(
    host=‘0.0.0.0‘,
    port= SERVER_PORT,
    debug=True
)