一个简单的公司自动订餐脚本实现

发表于 2016-01

前言

因为食堂难吃，公司推出了集体外卖订餐服务，员工需要每天去指定网页提交订餐。

由于经常忘了订餐，便诞生了下面这个脚本。

该系统功能是工作日每天定时订餐，并发送邮件告知用户订餐结果。

登陆过程分析

打开 http://cdcfan/ 便可见到登陆页面，登陆时只需输入自己英文名即可。

cdcfan 登陆页面

这里点击“进入”会发送 GET 请求，然后获得员工的各种信息，比如 ID，部门编号等。

cdcfan 登陆 GET

登陆结果如下

cdcfan 登陆结果

当用户点击“下单”，这时候才会发送 POST 参数，请求参数主要为之前 GET 请求获得的部门号和员工 id。

cdcfan 下单请求

请求会返回一个 json 数据。代表你是否成功订餐，还是已经订过餐。比如已经订过后，exceed_count 为 1，其余为 0.

cdcfan 下单结果

而页面则显示

cdcfan 吃胖了

回过头来看登陆过程，如图 2，每一步 GET 请求，都会附上一个隐含参数 '_=1451880513702'，参数值会逐次递增 1。这个值有什么用我还没研究出来，用 Burp Suite 代理修改了也依然能返回正确结果。不过实际上不需要去分析这个值是由服务器还是客户端产生，因为最关键的下单 POST 请求与这个毫无关系。

于是一切就变得简单了，实测这个系统其实并不依赖于会话机制，员工输入的姓名只是用来获得对应的员工 id 和部门号。那意味着，实现自动化订餐时，并不需要模拟登陆过程，只要向 /api/order-new POST 自己的 id 和部门号，就能完成订餐。

自动化订餐

首先定义数据库表，

create table cdcuser
{
    id integer primary key,
    psid varchar,
    depcode varchar,
    email varchar    
}

然后只需要遍历取出的每个记录，分别发送 POST 请求即可。

for user in r:
    psid = user[2]
    depcode = user[3]
    email = user[4]

    post_data = {'order':'e-1', 'psid':psid, 'depcode':depcode}
    post_data_urlencode = urllib.urlencode(post_data)

    req_url = 'http://cdcfan/api/order-new'
    req = urllib2.Request(url = req_url, data = post_data_urlencode)

    res_data = urllib2.urlopen(req)
    res_json = res_data.read()
    print res_json

定时任务和发送邮件

得益于公司内部 MTA 的转发策咯，任意邮件域名不作检查，除恶意邮件外，其他直接被转发。这意味着可以自己搭建 MTA 发送邮件。

MTA 可以直接使用 Postfix，在 Debian 8 上默认配置即可工作。Python 有非常方便的 smtplib 模块用于连接 MTA 并发送邮件

server = smtplib.SMTP('localhost')
server.set_debuglevel(1)
server.sendmail(fromaddr, toaddrs, msg)
server.quit()

不过我偷懒，直接解析 json 结果后，调用系统的 mail 命令

res = json.loads(res_json)
if (res['succeed_count'] == 0):
    cmd = "echo 'ding can shi bai, nin ke neng yi jing ding guo le. DO NOT REPLY' | mail -s cdcfanauto " + email
    # print cmd
    os.system(cmd)
else:
    cmd = "echo 'ding can cheng gong. DO NOT REPLY' | mail -s cdcfanauto " + email
    os.system(cmd)

最后就是在每周工作日定时调用，这里直接使用 crontab 的定时任务

10 10 * * 1-5 python /root/cdcfan/cdc.py >/dev/null 2>&1

问题

实际测试时遇到了邮件不能发送的问题，原因我没有仔细研究。

（大概是 crontab 在定时时会向系统管理员发送邮件，而由于上述的 python /root/cdcfan/cdc.py >/dev/null 2>&1 其实是 crontab 的子进程？？？，查看了 Postfix 的 maillog 之后发现，这两个进程之间的发送邮件会相互干扰。）

（比如 Postfix 的默认配置下，邮件域名为 @debian.com。所以默认管理员邮箱会向这个邮件域名发送。但这个域名是不存在，作为发送方，可以伪造，因为协议并没有强制要求。但作为接收方，伪造的邮件域名必然会不可达，最终被退回。）

（不过按理说不应该啊，不知道为什么管理员邮件发送失败会造成我调用 mail 失败。我又按照网上说的把 crontab 的发送邮件关闭，现象依旧，maillog 显示邮件被“自己的 MTA”退回。）

http 服务器

由于之前所述的我不能解释的问题，我只能另辟蹊径。让 crontab 定时发送 GET 请求到本地的一个 http 服务器，本地 http 服务器收到请求后再执行上述的订餐任务。本地 http 服务器和 crontab 完全是两个不相关的进程，不存在相互干扰的问题。

crontab 修改为

10 10 * * 1-5 curl http://127.0.0.1:5000/cdc/ >/dev/null 2>&1

服务器可以用 Flask 简单搭一个

from flask import Flask
import cdc

app = Flask(__name__)

@app.route('/cdc/')
def submit():
    cdc.submitcdcfan()
    return "haha"

if __name__ == '__main__':
    app.run()

最终结果

成功收到订餐成功的提示邮件

cdcfan 邮箱结果

彩蛋

大概因为是安全公司的内部系统，所以在页面源码里看到了如下的注释

js 源码

lyyyuna 的小花园

动静中之动, by lyyyuna