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