forked from 521xueweihan/HelloGitHub
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathgithub_bot.py
More file actions
180 lines (154 loc) · 5.24 KB
/
Copy pathgithub_bot.py
File metadata and controls
180 lines (154 loc) · 5.24 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
#!/usr/bin/env python
# -*- coding:utf-8 -*-
#
# Author : XueWeiHan
# E-mail : [email protected]
# Date : 16/8/30 下午10:43
# Desc : Github Bot
import os
import logging
import smtplib
import datetime
from email.mime.text import MIMEText
from email.header import Header
import requests
logging.basicConfig(
level=logging.WARNING,
filename=os.path.join(os.getcwd(), 'bot_log.txt'),
filemode='a',
format='%(name)s %(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s'
)
logger = logging.getLogger('Bot') # 设置log名称
API = {
'events': 'https://api.github.com/users/你的github用户名/received_events/public'
}
# github帐号
ACCOUNT = {
'username': '',
'password': ''
}
# 发送邮件，邮箱的信息
MAIL = {
'mail': '', # 发送邮件的邮箱地址
'username': '',
'password': '',
'host': 'smtp.qq.com',
'port': 465
}
# 接收邮件的邮箱地址
RECEIVERS = []
# 几天前
DAY = 1
# qq邮件服务文档：http://service.mail.qq.com/cgi-bin/help?id=28
CONTENT_FORMAT = """
<table border="2" align="center">
<tr>
<th>头像</th>
<th>用户名</th>
<th>项目名</th>
<th>starred日期</th>
<th>项目star数量</th>
</tr>
{starred_info}
</table>
"""
def get_data(page=1, per_page=100):
"""
从目标源获取数据
"""
args = '?page={page}&per_page={per_page}'.format(
page=page, per_page=per_page)
response = requests.get(API['events']+args,
auth=(ACCOUNT['username'], ACCOUNT['password']))
status_code = response.status_code
if status_code == 200:
resp_json = response.json()
return resp_json
else:
logging.error('请求api失败：', status_code)
return None
def check_condition(data):
"""
过滤条件
"""
create_time = datetime.datetime.strptime(
data['created_at'], "%Y-%m-%dT%H:%M:%SZ") + datetime.timedelta(hours=8)
date_condition = create_time >= (datetime.datetime.now()
- datetime.timedelta(days=DAY))
if (data['type'] == 'WatchEvent') and date_condition:
if data['payload']['action'] == 'started':
data['date_time'] = create_time.strftime("%Y-%m-%d %H:%M:%S")
return True
else:
return False
def analyze(json_data):
"""
分析获取的数据
:return 符合过滤条件的数据
"""
result_data = []
for fi_data in json_data:
if check_condition(fi_data):
result_data.append(fi_data)
return result_data
def make_content():
"""
生成发布邮件的内容
"""
json_data = get_data()
data = analyze(json_data)
content = []
for fi_data in data:
user = fi_data['actor']['login']
user_url = 'https://github.com/' + user
avatar_url = fi_data['actor']['avatar_url']
repo_name = fi_data['repo']['name']
repo_url = 'https://github.com/' + repo_name
date_time = fi_data['date_time']
try:
repo_stars = requests.get(fi_data['repo']['url'], timeout=2).json()
if repo_stars:
repo_stars = repo_stars['stargazers_count']
else:
repo_stars = '未知数'
except Exception as e:
repo_stars = '未知数'
logger.warning(u'获取：{} 项目星数失败——{}'.format(repo_name, e))
starred_info = """<tr>
<td><img src={avatar_url} width=32px></img></td>
<td><a href={user_url}>{user}</a></td>
<td><a href={repo_url}>{repo_name}</a></td>
<td>{date_time}</td>
<td>{repo_stars}</td>
</tr>
""".format(user=user, repo_name=repo_name,
repo_url=repo_url, user_url=user_url,
avatar_url=avatar_url, repo_stars=repo_stars,
date_time=date_time)
content.append(starred_info)
return content
def send_email(receivers, email_content):
"""
发送邮件
"""
sender = MAIL['mail'] # 发送邮件的邮箱
receivers = receivers # 接收邮件的邮箱，可设置多个
# 三个参数：第一个为文本内容，第二个 html 设置文本格式，第三个 utf-8 设置编码
message = MIMEText(
CONTENT_FORMAT.format(starred_info=''.join(email_content)),
'html', 'utf-8'
)
message['From'] = Header(u'Github机器人', 'utf-8')
message['To'] = Header(u'削微寒', 'utf-8')
subject = u'今日Github热点' # 设置邮件主题
message['Subject'] = Header(subject, 'utf-8')
try:
smtp_obj = smtplib.SMTP_SSL() # qq邮箱要求是https连接，所以需要用SMTP_SSL
smtp_obj.connect(MAIL['host'], MAIL['port']) # 设置SMTP地址和端口号
smtp_obj.login(MAIL['username'], MAIL['password'])
smtp_obj.sendmail(sender, receivers, message.as_string())
except smtplib.SMTPException as e:
logger.error(u"无法发送邮件: {}".format(e))
if __name__ == '__main__':
content = make_content()
send_email(RECEIVERS, content)