-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathutility.py
126 lines (118 loc) · 5.15 KB
/
utility.py
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
from discord.ext import commands
import discord
import config
import sys
import asyncio
import random
async def check_yes_no(bot,ctx,message,timeout=300)-> bool:
await message.add_reaction("✅")
await message.add_reaction("❌")
def reaction_check(reaction_, user_):
is_author=user_==ctx.author
are_same_messages = reaction_.message.channel == message.channel and reaction_.message.id == message.id
return are_same_messages and is_author
emoji = await bot.wait_for('reaction_add', check=reaction_check, timeout=timeout)
if emoji[0].emoji=="✅":
return True
if emoji[0].emoji=="❌":
return False
return None
async def check_yes_no_cancel(bot,ctx,message,timeout=300) -> int:
await message.add_reaction("✅")
await message.add_reaction("♻")
await message.add_reaction("❌")
def reaction_check(reaction_, user_):
is_author=user_==ctx.author
are_same_messages = reaction_.message.channel == message.channel and reaction_.message.id == message.id
return are_same_messages and is_author
emoji = await bot.wait_for('reaction_add', check=reaction_check, timeout=timeout)
if emoji[0].emoji=="✅":
return 1
if emoji[0].emoji=="♻":
return -1
if emoji[0].emoji=="❌":
return 0
return None
async def yes_no(bot,ctx,text,timeout=300) -> bool:
sent_msg=await ctx.reply(text)
await sent_msg.add_reaction("✅")
await sent_msg.add_reaction("❌")
def reaction_check(reaction_, user_):
is_author=user_==ctx.author
are_same_messages = reaction_.message.channel == sent_msg.channel and reaction_.message.id == sent_msg.id
return are_same_messages and is_author
emoji = await bot.wait_for('reaction_add', check=reaction_check, timeout=timeout)
if emoji[0].emoji=="✅":
return True
if emoji[0].emoji=="❌":
return False
return None
async def yes_no_cancel(bot,ctx,text,timeout=300) -> int:
sent_msg=await ctx.reply(text)
await sent_msg.add_reaction("✅")
await sent_msg.add_reaction("♻")
await sent_msg.add_reaction("❌")
def reaction_check(reaction_, user_):
is_author=user_==ctx.author
are_same_messages = reaction_.message.channel == sent_msg.channel and reaction_.message.id == sent_msg.id
return are_same_messages and is_author
emoji = await bot.wait_for('reaction_add', check=reaction_check, timeout=timeout)
if emoji[0].emoji=="✅":
return 1
if emoji[0].emoji=="♻":
return -1
if emoji[0].emoji=="❌":
return 0
return None
#そのmemberがrolesをすべて持っていたらtrue
def check_condition(member,roles,not_roles=None) -> bool:
if len(roles)==0:return False
for i in roles:
if i not in member.roles:return False
if type(not_roles)!=type(None):
for i in not_roles:
if i in member.roles:return False
return True
#そのギルド内のメンバーで、targetsリストにあるロールすべて持っているメンバーを返す
#targetsにメンバーが含まれている場合、そのメンバー＋指定ロールを持っている人となる
def get_targets(guild,targets,not_targets=None):
target_roles=[]
target_members=[]
target_mentions=[]
not_target_roles=[]
not_target_members=[]
not_target_mentions=[]
for target in targets:
id=target.strip("<!&@>")#argはroleもしくはmember がstr形式で送られてくるので、id(int)を抽出する
role_=guild.get_role(int(id))
member_=guild.get_member(int(id))
if type(role_) is not type(None):#指定がロールのとき
target_roles.append(role_)
target_mentions.append(role_.mention)
elif type(member_) is not type(None):#指定がメンバーのとき
target_members.append(member_)
target_mentions.append(member_.mention)
if type(not_targets)!=type(None):
for target in not_targets:
id=target.strip("<!&@>")#argはroleもしくはmember がstr形式で送られてくるので、id(int)を抽出する
role_=guild.get_role(int(id))
member_=guild.get_member(int(id))
if type(role_) is not type(None):#指定がロールのとき
not_target_roles.append(role_)
not_target_mentions.append(role_.mention)
elif type(member_) is not type(None):#指定がメンバーのとき
not_target_roles.append(member_)
not_target_mentions.append(member_.mention)
for member in guild.members:
if member in not_target_members:continue
if not check_condition(member,target_roles,not_target_roles): continue
if member in target_members:continue#すでに入ってたら２重に送信しないようスキップ
target_members.append(member)
return target_members,target_mentions,not_target_mentions
import requests
#https://qiita.com/mizunana/items/4afddc71f37df555078e
def download_img(url, file_name):
r = requests.get(url, stream=True)
if r.status_code == 200:
with open(file_name, 'wb') as f:
f.write(r.content)