-
Notifications
You must be signed in to change notification settings - Fork 76
Expand file tree
/
Copy pathafdian.py
More file actions
166 lines (156 loc) · 6.49 KB
/
afdian.py
File metadata and controls
166 lines (156 loc) · 6.49 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
from collections import namedtuple
import os
import requests
import time
import hashlib
import json
from PIL import Image, ImageDraw, ImageFont
from io import BytesIO
class User(namedtuple('User', ['name', 'avatar', 'all_sum_amount', 'remark'])):
pass
class Order(namedtuple('Order', ['remark'])):
pass
manual_sponsors = [
User('等疾风', 'https://i0.hdslb.com/bfs/face/b658b5ca52f41e53321d04f978be6784ca6f8687.jpg', '1000.00', '小彭老师加油，希望给个赞助位'),
User('只喝白开水', 'https://i2.hdslb.com/bfs/face/821b88a24c1319d1fb51b3854884e2f829855c75.jpg', '100.00', '确实快了 30 倍[赞]'),
User('包乾', '', '26.90', ''),
User('柿柿如意', '', '20.00', '请小彭老师喝奶茶'),
User('Starry', '', '100.00', '小彭老师加油！'),
User('阿哲', '', '100.00', '从小彭老师视频里学到太多'),
User('Eureka', '', '20.00', '希望我能赚的多一点，之后发电也就多一点'),
User('孙斌', '', '200.00', '06班孙斌，望越来越好'),
User('nullptr', 'https://i0.hdslb.com/bfs/face/effa1ec9bb0f5d09ed415da75129aca9d16092ac.jpg', '23.30', '小彭老师千古，伟大无需多言'),
User('Fred Song', '', '25.00', '小彭老师早点休息'),
User('**振', '', '20.00', ''),
User('**伟', '', '20.00', ''),
User('**枫', '', '26.90', ''),
User('相欢', '', '100.00', ''),
User('**辉', '', '30.00', '支持一下小彭老师，嘿嘿（求匿名）'),
User('**峰', '', '26.90', '感谢小鹏老师在技术上无私奉献，加油！'),
User('**卿', '', '26.60', ''),
User('**飞', '', '26.90', '黑心老板太可恶，资助老师吃饭�9�'),
User('**宇', '', '26.90', '给小彭老师点赞'),
User('**逸', '', '26.90', '绵薄之力，希望小彭老师早日度过难关'),
User('**帆', '', '10.00', '祝小彭老师生活顺利'),
User('**蓝', '', '500.00', '小彭老师加油�0�5�0�5'),
User('*洋', '', '26.90', ''),
User('**豪', '', '66.00', ''),
User('**楠', '', '26.90', '小彭老师加油！凭你的才华没问题的'),
User('*锷', '', '100.00', ''),
User('**博', '', '30.00', '小彭老师加油，你这么优秀肯定能找到好工作'),
User('**运', '', '30.00', '小彭老师加油！B站视频太棒啦！'),
User('windy小助手', 'https://i0.hdslb.com/bfs/face/d5d323e4063cad911bd722292fbf67dc6a23493b.jpg', '500.00', ''),
User('**康', '', '40.00', '感谢小彭老师的指导'),
User('*坤', '', '40.00', ''),
User('**峰', '', '20.00', ''),
]
def afd_query(which, **params):
user_id = '6256dedc1af911eebf8152540025c377'
token = os.environ['AFDIAN_TOKEN']
ts = int(time.time())
params = json.dumps(params)
sign = hashlib.md5(f'{token}params{params}ts{ts}user_id{user_id}'.encode('utf-8')).hexdigest()
res = requests.get(f'https://afdian.com/api/open/{which}', params={
'user_id': user_id,
'params': params,
'ts': ts,
'sign': sign,
}).json()
assert res['ec'] == 200, res
return res['data']
def afd_paged_query(which):
i = 1
while True:
page = afd_query(which, page=i)
n = page['total_page']
for item in page['list']:
yield item
if i >= n:
break
i += 1
def afd_query_orders():
order_lut = {}
for order in afd_paged_query('query-order'):
print(f'{order['user_id']}: {order['remark']}')
order_lut[order['user_id']] = Order(order['remark'])
return order_lut
def afd_query_sponsors():
res = []
order_lut = afd_query_orders()
for user in afd_paged_query('query-sponsor'):
user_id = user['user']['user_id']
last_order = order_lut.get(user_id, None)
if last_order:
remark = last_order.remark
else:
remark = ''
user_obj = User(user['user']['name'],
user['user']['avatar'],
user['all_sum_amount'],
remark)
res.append(user_obj)
return res
def afd_gen_thank_list():
sponsors = afd_query_sponsors()
sponsors += manual_sponsors
max_x = 30
max_y = 30
stride_x = 450
stride_y = 120
limit_y = stride_y * 10
max_max_y = max_y
for user in sponsors:
max_y += stride_y
if max_y + 10 >= limit_y:
max_max_y = max(max_max_y, max_y)
max_y = 30
max_x += stride_x
max_max_y = max(max_max_y, max_y)
max_max_x = max_x + stride_x
max_y += 10
img = Image.new('RGB', (max_max_x, max_max_y), color='#19242e')
x = 30
y = 30
total = 0
for user in sponsors:
draw = ImageDraw.Draw(img)
font = ImageFont.truetype('/usr/share/fonts/noto-cjk/NotoSansCJK-Medium.ttc', size=20)
font_small = ImageFont.truetype('/usr/share/fonts/noto-cjk/NotoSansCJK-Medium.ttc', size=13)
# font_small = ImageFont.truetype('/usr/share/fonts/Unifont/Unifont.otf', size=13)
if user.avatar:
avatar = Image.open(BytesIO(requests.get(user.avatar).content))
else:
wx_path = f'/home/bate/下载/wx-{user.name.replace('/', '|')}.png'
if os.path.exists(wx_path):
avatar = Image.open(wx_path)
else:
this_dir = os.path.dirname(os.path.abspath(__file__))
avatar = Image.open(os.path.join(this_dir, '../docs/img/favicon.ico'))
avatar = avatar.resize((90, 90))
img.paste(avatar, (x, y))
draw.text((x + 100, y), f'{user.name}', fill='white', font=font)
draw.text((x + 100, y + 30), f'￥{user.all_sum_amount}', fill='#aaaaaa', font=font)
remark = user.remark
if remark:
remark = remark.rstrip('。').replace('.。', '，')
# chunk remark into 40 char per line:
chunk_size = int((stride_x - 100) / font_small.size)
if len(remark) > chunk_size:
remark_lines = [remark[i:i+chunk_size] for i in range(0, len(remark), chunk_size)]
remark = '\n'.join(remark_lines)
draw.text((x + 100, y + 60), f'{remark}', fill='#779977', font=font_small)
print(f'{user.name} ￥{user.all_sum_amount} {remark}')
total += float(user.all_sum_amount)
y += stride_y
if y + 10 >= limit_y:
y = 30
x += stride_x
print(total)
return img
def main():
img = afd_gen_thank_list()
file = 'docs/img/thanks.png'
img.save(file)
os.system(f'display {file}')
if __name__ == '__main__':
main()