python脚本 环境准备

现在的公司用 Python 做 Web 开发，入职到现在为止（三个月），算是入门了 Python Web 开发

但是 Python 本身是脚本语言，我还从来没有体会到脚本语言能给日常工作带来的便利

就在昨天下午，我开发一个功能，因为有一些 bug，需要不断的调试，而调试意味着要准备环境，做各种操作，比如删除数据库数据、删除文件、发起 Post 请求等

那时我就在想，这些重复工作，不是正好可以用 Python 这个脚本语言帮我完成吗？

于是就开动了，经过一番查阅资料之后，写出了代码如下，可以一键完成各种准备工作：

 # -*- coding=utf-8 -*-

 import os
 import pymysql
 import requests
 import json

 # 删除ok_file文件
 def delete_file() :
     ok_file = os.path.join("/demo/2019-07-15/test.ok")
     ok_file = ok_file.replace('\\', '/')
     if os.path.exists(ok_file):
         os.remove(ok_file)

 # 删除数据库数据
 def delete_data():
     # 创建连接
     conn = pymysql.connect(host='127.0.0.1', port=3306, user='root', passwd='root', db='test')
     # 第二种创建连接方式
     # 连接配置信息
     # config = {
     #     'host': '127.0.0.1',
     #     'port': 3306,
     #     'user': 'root',
     #     'password': '123',
     #     'db': 't1',
     #     'charset': 'utf8mb4',
     #     'cursorclass': pymysql.cursors.DictCursor,
     # }
     # 创建连接 （2）
     # connection = pymysql.connect(**config)
     # 创建游标
     cursor = conn.cursor()
     # 执行SQL，并返回收影响行数
     effect_row = cursor.execute("delete from task_info")
     # 执行SQL，并返回受影响行数
     # effect_row = cursor.execute("update hosts set host = '1.1.1.2' where nid > %s", (1,))
     # 执行SQL，并返回受影响行数
     # effect_row = cursor.executemany("insert into hosts(host,color_id)values(%s,%s)", [("1.1.1.11",1),("1.1.1.11",2)])
     # 提交，不然无法保存新建或者修改的数据
     conn.commit()
     # 关闭游标
     cursor.close()
     # 关闭连接
     conn.close()

 # 参考： https://www.cnblogs.com/yourstars/p/8196054.html
 # 发起post请求
 def launch_post():
     url = 'http://127.0.0.1:8000/task/'
     body = {"demo_dirs": ["2019-07-15"]}
     headers = {'content-type': "application/json"}
     # print type(body)
     # print type(json.dumps(body))
     # 这里有个细节，如果body需要json形式的话，需要做处理
     # 可以是data = json.dumps(body)
     response = requests.post(url, data=json.dumps(body), headers=headers)
     # 也可以直接将data字段换成json字段，2.4.3版本之后支持
     # response  = requests.post(url, json = body, headers = headers)
     # 返回信息
     print(response.text)
     # 返回响应头
     print(response.status_code)

 delete_file() # 删除ok_file文件
 delete_data() # 删除数据库数据
 launch_post() # 发起post请求

参考：

https://www.cnblogs.com/jl-bai/p/6124088.html

https://www.cnblogs.com/yourstars/p/8196054.html