python基础学习6-mongodb、sys、接口开发、操作excel

1       mysql补充

cur = conn.cursor(cursor=pymysql.cursors.DictCursor)    #直接获取的数据转换为字典格式的

cur.description                   #直接获取的描述信息

fileds = [filed[0] for filed in cur.description]        #列表生成式，获取到第一行所有的字段

import pymysql,xlwt
conn = pymysql.connect(host='1.1.1.1',user='jxz',password='123456',db='jxz',port=3306,charset='utf8')
# cur = conn.cursor(cursor=pymysql.cursors.DictCursor) #直接获取的数据转换为字典格式的
cur = conn.cursor() #写excel用默认的二维数组即可，因为用字典的话顺序不固定，数据会写错
cur.execute('select * from app_student limit 10')
print(cur.description) #表的信息
fileds = [filed[0] for filed in cur.description] #列表生成式，获取到所有的字段
print(fileds)
# print(cur.fetchall())
data = list(cur.fetchall())
data .insert(0,fileds) #insert只是插入指定位置的数据
print(data)
book = xlwt.Workbook()
sheet = book.add_sheet('sheet1')
#方法一：
# for index,line_data in enumerate(data):
# for index2,col_data in enumerate(line_data):
# sheet.write(index,index2,col_data)
#一个一个单元格写，先写第一行的每个单元格
#写每一行的时候，行号是不变的，列在变

#方法二：此方法需要定义一个行和一个列去控制
line = 0 #行号

for d in data: #控制行
col = 0 # 列号
for col_data in d: #控制列
sheet.write(line,col,col_data)
col+=1
line+=1

book.save('ex.xls')

cur.close()
conn.close()

import pymysql,xlwt
def mydb(sql):
conn = pymysql.connect(host='1.1.1.1',user='jxz',password='123456',port=3306,db='jxz',charset='utf8',autocommit=True)
cur = conn.cursor()
cur.execute(sql)
fileds = [filed[0] for filed in cur.description] #列表生成式，获取到第一行表头数据信息
data = list(cur.fetchall()) #将获取的数据直接转换成元组
data.insert(0,fileds) #将第一行获取到的列表插入到data列表的第1个位置
return data
print(mydb('select * from app_student limit 10'))

def write_excel(excel_name):
book = xlwt.Workbook
sheet = book.add_sheet('sheet1')
a = 'select * from app_student'
for index,line_data in enumerate(mydb(sql=a)):
for index2,col_data in enumerate(line_data):
sheet.write(index,index2,col_data)
book.save(excel_name)

2       sys模块

import sys

print(sys.argv)     #把运行python文件的时候，传入的参数都放到argv里面

if len(command)>1:
excel = command[1]
print('用例已经运行',excel)
else:
print('运行python文件的时候，需要传入一个用例名称，例如：python run_case.py case.xls')

导入模块的顺序（重点）

1、从当前目录下找python文件

2、如果当前目录下没有，则去python的环境变量（此处不是操作系统那设置的环境变量）里面找这个python文件，print(sys.path) #查看python自己的环境变量

导入模块的实质（重点）

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

name = 'hahaha'

def conn_db(name):

print('this is a funcation',name)

print(__name__)

if __name__ == '__main__':      #在其他python文件中导入该模块时，它不会执行if __name__ == '__main__'下面的代码

conn_db('b文件')

sys.path.append(r'D:pythonscriptday7 ools')              #如果创建的python文件不在环境变量中，可以通过此方法临时设置环境变量使用

3       mongodb

import pymongo

client = pymongo.MongoClient(host='118.24.3.40',port=27017)

db = client['zhouyifan']        #选择数据库，如果不存在直接创建

collection = db['stu_info']

db.zhouyifan.insert({'name':'test1','sex':'男','age':18})

db['stu_info'].insert({'url':'http://www.jd.com','title':'jd.com','addr':'beijing'})   #插入

print(list(db['stu_info'].find({'title':'百度'})))        #查找，需要转换成list才能打印出来

for d in db['stu_info'].find({'title':'百度'}):     #使用循环打印出来的是字典

     print(d)

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

collection.delete_many()            #会删除多条

print(collection)

collection.update({'url':'http://www.jd.com','title':'www.jd.com'},{'url':'http://www.jd.com','title':'www.jd.com','addr':'亦庄'})    #更改

4       接口开发

接口开发的作用：

1、  知道接口是怎么开发的

2、  避免别人操作你的数据库

3、  mock服务（即模拟一个服务，保证测试流程能够顺利进行）

import sys
print(sys.path) #查看python自己的环境变量

python环境变量设置如下：

方法一：

快速将整个目录设置python的环境变量（如下方法只能在pycharm中好用，如果将代码拷贝给其他人使用，则设置的环境变量是不生效的）：

 

第二种设置环境变量的方法如下：

import sys,os

base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))     #环境变量设置必须在导入自己写的模块之前，自动获取当前文件向上两级文件夹的路径

sys.path.insert(0,base_path)        #将动态获取到的环境变量插入到最前面，这样导入模块更快

import sys,os
base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) #__file__是当前这个python文件的绝对路径,
# os.path.abspath(__file__)通过abspath方法将获取的路径分隔符进行转换
#再通过两个dirname获取到父目录
print(base_path)
sys.path.insert(0,base_path)

下面是从自己写的模块导入接口模块：

from lib.interface import server

from conf.setting import SERVER_PORT

server.run(host='0.0.0.0',port=SERVER_PORT,debug=True)      #设置host=0.0.0.0就可以在同一局域网内其他电脑能访问了，如果不指定host，则只能本机访问

name = 'hahaha'
def conn_db(name):
print('this is a funcation',name)

print('__name__',__name__)
if __name__ == '__main__': #在其他python文件中导入该模块时，它不会执行if __name__ == '__main__'下面的代码
conn_db('b文件')

接口开发示例：

import flask
import json
import pymysql
import hashlib
server = flask.Flask(__name__) #把当前的python文件当做一个服务，__name__表示main

def my_db(sql,port=3306,charset='utf8'):
ip, user, passwd, db = '1.1.1.1','jxz','123456','jxz'
conn = pymysql.connect(host=ip,user=user,
password=passwd,db=db,
port=port,charset=charset,autocommit=True)
cur = conn.cursor()
sql=sql.strip()
cur.execute(sql)
sqlstart = sql[:6].lower() #取sql的开头6位，转换为小写
if sqlstart.startswith('select') or sqlstart.startswith('insert'): #判断是selec或show的语句获取对应结果
data = cur.fetchall()
else: #加else是为了下面的return不报错
data = 'ok'
cur.close()
conn.close()
return data

def my_md5(s):
m = hashlib.md5(s.encode())
return m.hexdigest()

@server.route('/login',methods=['get','post'])
def login():
username = flask.request.values.get('username') #是从客户端发送过来的数据
pwd = flask.request.values.get('password')
# flask.request.cookies.get('ssss') #获取cookie
# flask.request.headers.get('tttt') #获取headers
# json_data = flask.request.json #获取入参是json类型的数据
if username=='test1'and pwd=='123456':
res = {"code":1000,"msg":"登录成功","sing":"dfsdfsdfsdfsdf"}
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')
pwd = flask.request.values.get('password')
cpwd = flask.request.values.get('cpassword')

if username and pwd and cpwd:
sql1 = 'select * from nhy where name="%s";' %username
sql_res = my_db(sql1)
if sql_res:
res = {"code":2001,"msg":"该用户已注册"}
else:
if pwd==cpwd:
md_pwd = my_md5(pwd)
sql2 = 'insert into nhy (name,pwd) value("%s","%s");' %(username,md_pwd)
sql_res = my_db(sql2)
res = {"code": 1000, "msg": "注册成功"}
else:
res = {"code": 2002, "msg": "两次密码输入不相同"}
else:
res = {"code":2002,"msg":"必填参数未填，请查看接口文档！"}
return json.dumps(res, ensure_ascii=False)

server.run(port=8989,debug=True) #启动服务，此项需要注意，启动该服务之后如果修改脚本再次运行直接重启即可，不能再次点击右键运行，否则会报接口冲突，因为原有运行的服务未停止

5       修改excel

import xlrd

from xlutils import copy        #修改excel需要导入该模块

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

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

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

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

sheet.write(1,1,'xiaobai')

new_book.save('ex.xls')

6       操作excel

#xlwt 只能写excel

#xlrd 只能读excel

import xlrd

book = xlrd.open_workbook('ex.xls')

sheet = book.sheet_by_index(0)      #通过sheet索引获取sheet

# sheet = book.sheet_by_name('sheet1')      #通过sheet名称获取sheet

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

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

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

print(sheet.row_values(0))      #获取某一行的数据

print(sheet.nrows)      #获取excel表中总共有多少行