Python项目：扇贝网小组查卡助手
扇贝网是一个非常棒的英语学习网站，大家还可以加入一些小组，一起交流学习、共同进步。但是，小组管理起来非常辛苦，尤其是在0点前踢出不打卡的成员，因此考虑利用程序来实现小组查卡自动化。
登录
| 操作 | 扇贝网登录 | |
| URL | http://www.shanbay.com/accounts/login/ | |
| 方式 | POST | |
| 数据 | csrfmiddlewaretoken | CSRF令牌 |
| username | 用户名 | |
| password | 密码 | |
CSRF令牌存在于Cookie中，我们需要先以GET方式访问该URL，就能取到CSRF令牌了。
# -*- coding: utf-8 -*- import requests class Shanbay(): def __init__(self, username, password): self.request = requests.Session() self.username = username self.password = password def login(self): url = 'http://www.shanbay.com/accounts/login/' r = self.request.get(url) csrftoken = r.cookies['csrftoken'] data = { 'csrfmiddlewaretoken': csrftoken, 'username': self.username, 'password': self.password, } return self.request.post(url, data=data).ok
成员管理
如果我们获取小组管理后台所有组员的信息，比较费时间。考虑实际需求，不妨仅获取当天未打卡的组员的信息，这样能大大提高查卡效率。
踢人需要data-id，这个在小组管理后台页面就能获取到。但是，如果我们想发站内短信，就需要username，而username在小组管理后台页面里是没有的，这个需要查看个人打卡日记。
从个人打卡日记不仅能看到username，还能看到该贝友入组后最近已连续有多少天未打卡（这往往也是组规限定的内容）等等。
| 操作 | 踢人 | |
| URL | http://www.shanbay.com/api/v1/team/member/ | |
| 方式 | PUT | |
| 数据 | action | 动作（'dispel'） |
| ids | data-id | |
# -*- coding: utf-8 -*- from bs4 import BeautifulSoup from Journal import Journal import re class Domain(): def __init__(self, shanbay): self.shanbay = shanbay self.request = shanbay.request def get_not_checked_members(self): ''' data_id : 踢人时需要data_id role : 身份标识 nickname : 昵称 user_id : 发短信时需要user_id username : 用户名 points : 贡献值 days : 组龄 rate : 打卡率 checked_yesterday: 昨天是否打卡 checked : 今天是否打卡 off_dyas : 入组后最近连续未打卡天数 ''' members = [] for page in range(1, 48): html = self.request.get('http://www.shanbay.com/team/manage/?t=checkin_today&page=%d' % page).text soup = BeautifulSoup(html, 'html5lib') for member in soup.find_all('tr', class_='member'): checked = member.find_all(class_='checked')[1].find('span').text.strip() == '已打卡' if checked: break days = int(member.find(class_='days').text) user_id = re.findall('\d+', member.find(class_='user').find('a')['href'])[0] user = Journal(shanbay=self.shanbay, user_id=user_id) checked_yesterday = member.find_all(class_='checked')[0].find('span').text.strip() == '已打卡' if checked_yesterday: off_days = 1 else: off_days = user.get_off_days(days) data = { 'data_id':member['data-id'], 'role':member['role'], 'nickname':member.find(class_='user').find('a').text, 'user_id':user_id, 'username':user.get_username(), 'points':int(member.find(class_='points').text), 'days':days, 'rate':float(member.find(class_='rate').find('span').text[:-2]), 'checked_yesterday':checked_yesterday, 'checked':checked, 'off_dyas':off_days } members.append(data) else: continue break return members def dismiss(self, data_ids): url = 'http://www.shanbay.com/api/v1/team/member/' data = { 'action': 'dispel', } data['ids'] = ','.join(map(str, data_ids)) r = self.request.put(url, data=data) return r.json()['msg'] == "SUCCESS"
（这里用到了Python跳出两层循环的技巧*^_^*）
打卡日记
通过打卡日记，我们可以获得一些基本信息，例如：用户名、最近连续未打卡天数等。
# -*- coding: utf-8 -*- from bs4 import BeautifulSoup import re import datetime import time class Journal(): def __init__(self, shanbay, user_id): self.shanbay = shanbay self.request = shanbay.request self.user_id = user_id self.soup = self.__get_journal_soup() def __get_journal_soup(self): html = self.request.get('http://www.shanbay.com/checkin/user/%s/' % self.user_id).text return BeautifulSoup(html) def get_username(self): return re.findall(u'(\w+)\s*的日记', self.soup.find_all(class_='page-header')[0].find('h2').text)[0] def get_off_days(self, days=0): pass
站内短信
| 操作 | 发送站内短信 | |
| URL | http://www.shanbay.com/api/v1/message/ | |
| 方式 | POST | |
| 数据 | recipient | 收件人（username） |
| subject | 标题 | |
| body | 内容 | |
| csrfmiddlewaretoken | CSRF令牌 | |
# -*- coding: utf-8 -*- class Message(): def __init__(self, shanbay): self.shanbay = shanbay self.request = shanbay.request def send_msg(self,recipient, subject, body): url = 'http://www.shanbay.com/api/v1/message/' data = { 'recipient': recipient, 'subject': subject, 'body': body, 'csrfmiddlewaretoken': self.request.cookies['csrftoken'] } return self.request.post(url, data=data).ok
小组管理
| 操作 | 设定加组条件 | |
| URL | http://www.shanbay.com/team/setqualification/{team_id} | |
| 方式 | POST | |
| 数据 | value | 天数 |
| kind | 类型 | |
| condition | 条件 | |
| team | 小组id | |
| csrfmiddlewaretoken | CSRF令牌 | |
若需要在小组发帖或回帖，需要forum_id而不是小组id，而forum_id可以通过小组主页找到。
| 操作 | 发帖 | |
| URL | http://www.shanbay.com/api/v1/forum/{forum_id}/thread/ | |
| 方式 | post | |
| 数据 | title | 标题 |
| body | 内容 | |
| csrfmiddlewaretoken | CSRF令牌 | |
| 操作 | 回帖 | |
| URL | http://www.shanbay.com/api/v1/forum/thread/{post_id}/post/ | |
| 方式 | POST | |
| 数据 | body | 内容 |
| csrfmiddlewaretoken | CSRF令牌 | |
# -*- coding: utf-8 -*- from bs4 import BeautifulSoup class Team(): def __init__(self, shanbay, team_id): self.shanbay = shanbay self.request = shanbay.request self.team_id = team_id self.forum_id = self.__get_forum_id() def set_join_limit(self, days, kind=2, condition='>='): url = 'http://www.shanbay.com/team/setqualification/%s' % self.team_id data = { 'value': days, 'kind': kind, 'condition': condition, 'team': self.team_id, 'csrfmiddlewaretoken': self.request.cookies['csrftoken'] } r = self.request.post(url, data=data) return 'http://www.shanbay.com/referral/invite/?kind=team' == r.url def __get_forum_id(self): html = self.request.get('http://www.shanbay.com/team/detail/%s/' % str(self.teamId)).text soup = BeautifulSoup(html) return soup.find(id='forum_id')['value'] def new_post(self, title, content): url = 'http://www.shanbay.com/api/v1/forum/%s/thread/' % self.forum_id data = { 'title': title, 'body': content, 'csrfmiddlewaretoken': self.request.cookies['csrftoken'] } return self.request.post(url, data=data).json() def reply_post(self, post_id, content): url = 'http://www.shanbay.com/api/v1/forum/thread/%s/post/' % post_id data = { 'body': content, 'csrfmiddlewaretoken': self.request.cookies.get('csrftoken') } return self.request.post(url, data=data).json()
