day7

操作mysql补充：将查询到表的数据写入excel

import pymysql,xlwt

coon=pymysql.connect(host='xxx.24.3.40',user='jxz',password='123456',db='jxz',charset='utf8')     #连接数据库

cur=coon.cursor()

# cur=coon.cursor(cursor=pymysql.cursors.DictCursor)      #指定游标类型，返回的数据是字典

cur.execute('select * from app_student limit 10;')     #执行sql语句

# print(cur.description)       #返回表的信息，类型是二元数组(元素1是字段，其余指的是该字段长度、是否必填等约束信息）

fileds=[ filed[0] for filed in cur.description]      #获取到表所有的字段

data=list(cur.fetchall())      #将二元数组强制转换成list

data.insert(0,fileds)      #增加表字段

book=xlwt.Workbook()

sheet=book.add_sheet('sheet1')

 

# 写数据

# 1、一个一个单元格写，先写第一行的每个单元格

# 2、写每一行的时候，行号是不变的，列在变

# 3、嵌套的循环

方法1：

line=0 #行号

for d in data: #外层循环控制行

    col=0 #列号

    for col_data in d: #内层控制列

        sheet.write(line,col,col_data)

        col+=1

    line+=1

方法2(高端)：

for index,line_data in enumerate(data):

    for index2,col_data in enumerate(line_data):

        sheet.write(index,index2,col_data)

 

book.save('user.xls')

cur.close()

coon.close()

 

 

sys模块

import sys

print(sys.argv)      #在命令行运行python文件时(python xx.py).它会把传入的参数都放到这个argv里面；返回的是一个list;

在Python中，通过命令运行如下Python文件：

command=sys.argv

if len(command)>1:

    excle=command[1]

    print('用例已经运行完成',excle)

else:

    print('运行这个python文件的时候，需要传入一个用例名称，例如:'

  'python run_case.py case.xls')

导入模块

导入模块的顺序：

1、先在当前目录下找这个python文件

2、再去python的环境变量里面找这个python文件

导入模块的实质：

1.把这个模块的代码，从头到尾执行一次

print(sys.path)    #查看Python的环境变量

例1：在当前目录，导入当前目录的其他python文件

有一个b.py文件如下：

name= 'haha'

def coon_db():

print('this is a func..')

print('这是最后一段代码')

if __name__ == '__main__': #下面代码为测试代码，在其他Python文件导入这个模块，它不会执行

coon_db('b文件')       #测试代码

在a文件中导入b文件-方法1：

import b   #在a文件里导入b文件

b.coon_db()  #调用b文件里的my函数

print(b.name) #调用b文件里的name变量

在a文件中导入b文件-方法2：

from b import coon_db,name #只导入b文件里的coon_db函数和变量name

coon_db()

print(name)

 

例2：在当前目录，导入其他目录的python文件

1.把要导入模块的目录加入到环境变量里面

import sys

sys.path.append(r'E:spz ools')  #加入环境变量，加在最后

sys.path.insert(0,r'E:spz ools')   #加入环境变量:加入在指定位置

print(sys.path)    #查看环境变量里面的所有目录

 

操作mongodb

import pymongo

client=pymongo.MongoClient(host='xxx.24.3.40',port=27017) #连接

db=client['ht_spz']#选择数据库，如果这个数据库不存在的话，会帮你创建

collection=db['stu_info']#选择一个集合，相当于mysql里面的表；如果集合不存在，会帮你创建

增：

collection.insert({'name':'nhy','sex':'男'}) #往集合里面插入数据

collection.insert({'url':'http://www.baidu.com','title':'baidu.com','addr':'西二旗'})

查：

方法1：

for d in collection.find({'title':'baidu.com'}):

    print(d)

方法2：

print(list(collection.find({'title':'baidu.com'}))) #查找集合里面，指定数据

print(list(collection.find())) #查找集合里面，所有数据

删：

collection.delete_one({'title':'baidu.com'}) #如果有多条的话，只会删除一条

collection.delete_many({'title':'baidu.com'}) #如果有多条的话，会删除多条

改：

collection.update({'name':'nhy'},{'name':'nhy','addr':'亦庄'})

 

接口开发

1.更好的测试

2.mock(模拟接口)

3.给别人提供数据

import flask

server=flask.Flask(__name__) #启动服务,把当前这个Python文件当做一个服务

@server.route('/login',methods=['get','post'])

def login():

    return 'hahahaha'

server.run(port=8989,debug=True)#启动服务;加上debug=True修改代码之后会自动重启

 

定义1个可以传参的接口(登录接口)：

import flask

import json

server=flask.Flask(__name__) #启动服务,把当前这个Python文件当做一个服务

@server.route('/login',methods=['get','post'])

def login():

    #入参是在?后面传参/k-v类型传参:

    username=flask.request.values.get('username') #括号里的username是从客户端发过来的数据

    pwd=flask.request.values.get('password')

    #入参json类型传参：

    json_data=flask.request.json   #获取所有数据

    json_data = flask.request.json.get('username') #获取某个key/value

    #入参有cookie:

    cookie=flask.request.cookies.get('xxx')  #获取cookie

    #入参有headers:

    headers=flask.request.headers.get('xxxx')  #获取headers

    if username == 'nhy' and pwd == '123456':

        res = {'code': 1000, 'msg': '登录成功', 'sign': 'sdfsdf23vsrqsdf'}

    else:

        res = {'code': 2000, 'msg': '账号/密码错误'}

    return json.dumps(res, ensure_ascii=False)

server.run(port=8989,debug=True)#启动服务;加上debug=True修改代码之后会自动重启

 

定义1个注册的接口，并把数据存到mysql里面：

import flask,pymysql

import hashlib

import json

server=flask.Flask(__name__) #启动服务,把当前这个Python文件当做一个服务

 

def my_db(sql,port=3306,charset='utf8'):

ip, user, passwd, db = 'xxx.24.3.40','jxz','123456','jxz'

coon = pymysql.connect(host=ip,user=user,

password=passwd,db=db,

port=port,charset=charset,autocommit=True)

cur = coon.cursor()

sql=sql.strip()

cur.execute(sql)

sql_start = sql[:6].lower()#取sql的开头,转成小写

if sql_start.startswith('select') or sql_start.startswith('show'):

data = cur.fetchall()

else:

data = 'ok'

cur.close()

coon.close()

return data

 

def my_md5(s):

m = hashlib.md5(s.encode())

return m.hexdigest()

 

@server.route('/register',methods=['post'])

def reg():

username = flask.request.values.get("username")

passwd = flask.request.values.get("passwd")

cpasswd = flask.request.values.get("cpasswd")

if username and passwd and cpasswd:

sql1 = 'select * from nhy where name="%s";'%username

sql_res = my_db(sql1)

if sql_res:

res = {'code':2001,'msg':'该用户已经被注册'}

else:

if passwd==cpasswd:#

md5_pwd = my_md5(passwd)

sql2 = 'insert into nhy (name,pwd) value ("%s","%s");'%(username,md5_pwd)

my_db(sql2)

res = {'code':1000,'msg':'注册成功'}

else:

res = {'code':2003,'msg':'两次输入的密码不一致'}

else:

res = {'code':2002,'msg':'必填参数未填，请查看接口文档！'}

return json.dumps(res,ensure_ascii=False)

server.run(port=8989,debug=True) #启动服务

 

程序分目录

1.新建主目录:my_api

2.新建二级目录：bin(运行程序的入口),conf(存放配置文件),lib(存放程序的主逻辑),logs(写日志)

3.新建二级普通文本文件：readme.txt(程序的说明)

4.在conf下新建setting.py,文件如下：

MYSQL_INFO = {

'host':'xxx.24.3.40',

'user':'jxz',

'password':'123456',

'port':3306,

'db':'jxz',

'charset':'utf8',

'autocommit':True

}#数据库配置信息

 

REDIS_INFO ={ }

 

SERVER_PORT = 8989

5.在lib下新建tools.py,文件如下：

import pymysql,hashlib

from conf.setting import MYSQL_INFO

def my_db(sql):

coon = pymysql.connect(**MYSQL_INFO)

#这里用2个星号，就会把字典里面的k和v变成 k=v，只适用于字典

cur = coon.cursor(cursor=pymysql.cursors.DictCursor)

sql=sql.strip()

cur.execute(sql)

sql_start = sql[:6].lower()#取sql的开头,转成小写

if sql_start.startswith('select') or sql_start.startswith('show'):

data = cur.fetchall()

else:

data = 'ok'

cur.close()

coon.close()

return data

 

def my_md5(s):

m = hashlib.md5(s.encode())

return m.hexdigest()

 

def op_redis():

pass

#在pacharm中加环境变量的方法：选择要加入环境变量的目录，右键——Mark Directory as——Sources Root；

6.在lib下新建interface.py(写接口),文件如下：

import flask,json

from lib.tools import my_db,my_md5

 

server = flask.Flask(__name__)

 

@server.route('/login',methods=['get','post'])

def taotao():

username = flask.request.values.get('username')

pwd = flask.request.values.get('password')

if username=='nhy' and pwd=='123456':

res  = {'code':1000,'msg':'登录成功','sign':'sdfsdf23vsrqsdf'}

else:

res = {'code':2000,'msg':'账号/密码错误'}

return json.dumps(res,ensure_ascii=False)

 

@server.route('/register',methods=['post'])

def reg():

username = flask.request.values.get("username")

passwd = flask.request.values.get("passwd")

cpasswd = flask.request.values.get("cpasswd")

if username and passwd and cpasswd:

sql1 = 'select * from nhy where name="%s";'%username

sql_res = my_db(sql1)

if sql_res:

res = {'code':2001,'msg':'该用户已经被注册'}

else:

if passwd==cpasswd:#

md5_pwd = my_md5(passwd)

sql2 = 'insert into nhy (name,pwd) value ("%s","%s");'%(username,md5_pwd)

my_db(sql2)

res = {'code':1000,'msg':'注册成功'}

else:

res = {'code':2003,'msg':'两次输入的密码不一致'}

else:

res = {'code':2002,'msg':'必填参数未填，请查看接口文档！'}

return json.dumps(res,ensure_ascii=False)

7.在bin下新建start.py(运行程序入口),文件如下：

写法1：

from lib.interface import server

from conf.setting import SERVER_PORT

server.run(port=SERVER_PORT,debug=True)#不写host，只能在本地访问

#host写成0.0.0.0就可以同一个局域网里面所有人都能访问了，如：server.run(host='0.0.0.0',port=SERVER_PORT,debug=True)

 

写法2：(不在pacharm中运行，把这些文件拿到别处运行，运行前需手动加环境变量)

import sys

sys.path.insert(0,r'E:spzday7my_api')

from lib.interface import server

from conf.setting import SERVER_PORT

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

#一定要先加环境变量，再导入

 

写法3（动态加环境变量的方法）：

import sys,os

base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

sys.path.insert(0,base_path)

from lib.interface import server

from conf.setting import SERVER_PORT

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

#print(__file__)#不管什么时候，__file__都是当前这个python文件的绝对路径

#这里加入环境变量的，通过__file__获取到父目录的父目录

#用abs_paht是因为分隔符不正确

 

读excel

import xlrd   #只能读

book = xlrd.open_workbook('nhy.xls')    #打开excel

# print(book.nsheets)    #获取到excel里面总共有多少个sheet页

sheet = book.sheet_by_index(0)     #获取sheet页:根据索引

#book.sheet_by_name('sheet1')     #获取sheet页:根据名字

print(sheet.cell(0,0).value)    #指定行和列，获取某个单元格里面的内容

print(sheet.cell(1,0).value)

print(sheet.row_values(0))   #获取某一行的数据，0代表获取第1行的数据

print(sheet.row_values(1))   #获取某一行的数据,1代表获取第2行的数据

print(sheet.nrows)      #获取这个excel里面总共有多少行

print(sheet.col_values(0))   #获取某一列的数据,0代表获取第1列的数据

print(sheet.col_values(1))   #某一列的数据，1代表获取第2列的数据

print(sheet.ncols)    #获取这个excel总共有多少列

 

修改excel

1.修改excel要先复制一个excel，再修改

import xlrd

from xlutils import copy

book1 = xlrd.open_workbook('nhy.xls')   #打开原来的excel

new_book = copy.copy(book1)     #拷贝一个新的excel

sheet = new_book.get_sheet(0)    #获取第一个sheet页

sheet.write(1,3,'18')

sheet.write(1,1,'王艳会')

new_book.save('nhy.xls')   #保存excel