httpie + shell auto script

Author: caofanCPU

auto/ID.csv

@@ -0,0 +1 @@
+9290

auto/backup.txt

@@ -0,0 +1,13 @@
+- 创建分类(/api/v100/live/employee/upload/dibblingVideo/addCategory)
+"name=默认分类",
+"schoolId=NONE"
+
+- 登录
+"phone=18999999999",
+"smsAuthCode=123456",
+"loginType:=0",
+"pwd=ht123456.",
+
+
+- 删除分类(/api/v100/live/employee/upload/dibblingVideo/deleteCategory)
+"categoryId=NONE"

auto/batchHandler.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+import json
+import ssl
+import sys
+import subprocess
+import time
+import pandas as pd
+
+import requests
+
+# 屏蔽HTTPS证书校验, 忽略安全警告
+requests.packages.urllib3.disable_warnings()
+context = ssl._create_unverified_context()
+joiner = ' '
+id_key = 'NONE'
+# 登陆一次后是否服复用cookie
+hot_reload = True
+cmd = "http"
+no_ca = "--verify=no"
+httpie_allow_view = {
+    "-v": "显示请求详细信息",
+    "-h": "显示请求头",
+    "-b": "显示请求Body",
+    "-d": "下载文件",
+    "": "默认"
+}
+httpie_view = None
+try:
+    if len(sys.argv) > 1:
+        if httpie_allow_view.get(sys.argv[1]) is not None:
+            httpie_view = sys.argv[1]
+        else:
+            print("输入参数有误, 仅支持如下参数: -v显示请求详细信息|-h显示请求头|-b显示请求Body|-d下载文件")
+except Exception as e:
+    print(e)
+
+
+def httpie_cmd(id):
+    """
+    执行excuteUrl.json接口
+    :param id
+    :return:
+    """
+    with open("./excuteUrl.json", 'r') as request_data:
+        request_json = json.load(request_data)
+    url = request_json['url']
+    method = request_json['method']
+    request_headers = load_request_headers(request_json['headers'], hot_reload)
+    headers = joiner.join(request_headers)
+    body = joiner.join(request_json['body'])
+    body = body.replace(id_key, str(id))
+    httpie_params = [cmd, no_ca]
+    if httpie_view is not None:
+        httpie_params.append(httpie_view)
+    httpie_params.extend([method, url, headers, body])
+    httpie = joiner.join(httpie_params)
+    print(httpie)
+    print("当前ID: ", id)
+    # 延时执行
+    time.sleep(0.05)
+    subprocess.call(httpie, shell=True)
+
+
+def load_request_headers(headers, hot_reload):
+    """
+    加载请求header
+    :param headers:
+    :param hot_reload: 是否热加载, 复用已有Cookie
+    :return:
+    """
+    if hot_reload:
+        with open("./cookie.txt", "r") as f:
+            cookie = ''.join(f.readlines())
+    else:
+        cookie = auto_login()
+    headers.append(cookie)
+    return headers
+
+
+def load_data():
+    """
+    读取数据文件, 每行为一条数据
+    :return:
+    """
+    data = pd.read_csv("./ID.csv", header=-1)
+    data.columns = ['id']
+    return data['id']
+
+
+def auto_login():
+    """
+    自动登录, 获取登录Cookie, 写入文件, 在控制台打印
+    """
+    with open("./ssoLogin.json", 'r') as sso_login_request_data:
+        request_json = json.load(sso_login_request_data)
+    url = request_json['url']
+    method = request_json['method']
+    request_headers = {}
+    for item in request_json['headers']:
+        split = item.replace('=', '').split(':')
+        request_headers[split[0]] = split[1]
+    request_body = {}
+    for item in request_json['body']:
+        split = item.replace(':', '').split('=')
+        request_body[split[0]] = split[1]
+
+    request_headers = {"Content-Type": "application/json", "HT-app": "6"}
+    response = requests.request(method, url, headers=request_headers, json=request_body, timeout=3, verify=False)
+    response_headers = response.headers
+    cookie = "'Cookie:" + response_headers.get("set-Cookie") + "'"
+    with open("./cookie.txt", "w") as f:
+        f.write(cookie)
+    # JSON标准格式
+    response_body = json.dumps(response.json(), ensure_ascii=False, indent=4)
+    print(response_body)
+    return cookie
+
+
+def main():
+    # 首先登陆一次
+    auto_login()
+    for id in load_data():
+        httpie_cmd(id)
+
+
+if __name__ == '__main__':
+    main()

auto/cookie.txt

@@ -0,0 +1 @@
+'Cookie:_r=ihJwwY0cYz49sr6PMLsFz0BZpRyQxkFQVg2iP9QJYR2ggATr7qhOvPeiiuQ5ZE9t|32ziSuOPkzw=; Max-Age=2592000; Expires=Fri, 08-May-2020 10:04:29 GMT; Domain=testa.huitong.com; Path=/; HttpOnly, _a=HUQD4RRBySY9sr6PMLsFz0BZpRyQxkFQVg2iP9QJYR2ggATr7qhOvCgr4ayK7PjY55U6X+1Po30u3aMceLb+8kFgGfrroYoW|6tzip/GZSQY=; Max-Age=7200; Expires=Wed, 08-Apr-2020 12:04:29 GMT; Domain=emp.testa.huitong.com; Path=/; HttpOnly, _a=HUQD4RRBySY9sr6PMLsFz0BZpRyQxkFQVg2iP9QJYR2ggATr7qhOvCgr4ayK7PjY55U6X+1Po30u3aMceLb+8kFgGfrroYoW|6tzip/GZSQY=; Max-Age=7200; Expires=Wed, 08-Apr-2020 12:04:29 GMT; Domain=emp2.testa.huitong.com; Path=/; HttpOnly'

auto/excuteUrl.json

@@ -0,0 +1,11 @@
+{
+  "url": "https://live.testa.huitong.com/api/v100/live/employee/upload/dibblingVideo/deleteCategory",
+  "method": "POST",
+  "headers" : [
+      "Content-Type:application/json",
+      "HT-app:6"
+  ],
+  "body": [
+    "categoryId=NONE"
+  ]
+}